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