SetInferShape
功能说明
使用图模式时，需要调用该接口注册Shape推导函数。
函数原型
Text
OpDef &SetInferShape(gert::OpImplRegisterV2::InferShapeKernelFunc func)
参数说明
参数 | 输入/输出 | 说明 |
|---|---|---|
func | 输入 | Shape推导函数。InferShapeKernelFunc类型定义如下： using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *); |
返回值说明
OpDef算子定义，OpDef请参考OpDef。
约束说明
无