SetTiling
函数功能
注册Tiling函数。
函数原型
OpAICoreDef &SetTiling(gert::OpImplKernelRegistry::TilingKernelFunc func);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
func |
输入 |
Tiling函数。TilingKernelFunc类型定义如下: using TilingKernelFunc = UINT32 (*)(TilingContext *); |
返回值说明
OpAICoreDef算子定义,OpAICoreDef请参考OpAICoreDef类。
约束说明
无
父主题: OpAICoreDef类