SetInferShapeRange
函数功能
注册ShapeRange推导函数。
函数原型
OpDef &SetInferShapeRange(gert::OpImplRegisterV2::InferShapeRangeKernelFunc func);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
func |
输入 |
ShapeRange推导函数。InferShapeRangeKernelFunc类型定义如下: using InferShapeRangeKernelFunc = UINT32 (*)(InferShapeRangeContext *); |
返回值说明
OpDef算子定义,OpDef请参考OpDef类。
约束说明
无
父主题: OpDef类