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