下载
中文
注册

SetInferShape

函数功能

注册Shape推导函数。

函数原型

OpDef &SetInferShape(gert::OpImplRegisterV2::InferShapeKernelFunc func);

参数说明

参数

输入/输出

说明

func

输入

Shape推导函数。InferShapeKernelFunc类型定义如下:

using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *);

返回值说明

OpDef算子定义,OpDef请参考OpDef

约束说明