SetTiling

Function

Registers the tiling function. The prototype of the tiling function is definite. It accepts a TilingContext as the input. The input and output shape pointers can be obtained from the context. The registered tiling function is called by the framework, and the TilingContext parameter is passed during the call.

Prototype

      
           OpAICoreDef &SetTiling(gert::OpImplRegisterV2::TilingKernelFunc func)

Parameters

Parameter

Input/Output

Description

func

Input

Tiling function. The TilingKernelFunc type is defined as follows:

           
                using TilingKernelFunc = UINT32 (*)(TilingContext *);

Returns

OpAICoreDef operator definition. For details, see OpAICoreDef.

Constraints

None

Parent topic: OpAICoreDef