注册Tiling函数。
1
OpAICoreDef &SetTiling(gert::OpImplRegisterV2::TilingKernelFunc func)
参数
输入/输出
说明
func
输入
Tiling函数。TilingKernelFunc类型定义如下：
using TilingKernelFunc = UINT32 (*)(TilingContext *);
OpAICoreDef算子定义，OpAICoreDef请参考OpAICoreDef。
无