注册Shape推导函数。
OpDef &SetInferShape(gert::OpImplKernelRegistry::InferShapeKernelFunc func);
参数
输入/输出
说明
func
输入
Shape推导函数。InferShapeKernelFunc类型定义如下：
using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *);
OpDef算子定义，OpDef请参考OpDef类。
无