SetTiling

Function

Registers the tiling function. The prototype of the tiling function is fixed and accepts a TilingContext as the input. You can obtain the input and output shape pointers 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