注册Tiling函数。Tiling函数的原型是固定的,接受一个TilingContext作为输入,在此context上可以获取到输入、输出的Shape指针等内容。注册的Tiling函数由框架调用,调用时会传入TilingContext参数。
1
|
OpAICoreDef &SetTiling(gert::OpImplRegisterV2::TilingKernelFunc func) |
|
参数 |
输入/输出 |
说明 |
||
|---|---|---|---|---|
|
func |
输入 |
Tiling函数。TilingKernelFunc类型定义如下:
|
OpAICoreDef算子定义,OpAICoreDef请参考OpAICoreDef。
无