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