昇腾社区首页
中文
注册
开发者
下载

IMPLEMT_COMMON_INFERFUNC

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件

/include/graph/operator_reg.h

功能说明

封装算子的Common_InferShape函数。

IMPLEMT_INFERFUNC的区别是,此函数自动生成的一个类型为Operator类的对象op,可直接调用Operator接口进行InferShape的实现。若InferShape方法具有通用性,可被多个算子的原型实现调用,可选择此接口实现。

函数原型

1
IMPLEMT_COMMON_INFERFUNC(func_name)

参数说明

参数名

输入/输出

描述

func_name

输入

InferShape函数名,用户自定义。

返回值说明

约束说明