跳到主要内容

IMPLEMT_COMMON_INFERFUNC

函数功能​

封装算子的Common_InferShape函数。

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

函数原型​

IMPLEMT_COMMON_INFERFUNC(func_name)

约束说明​

无

参数说明​

参数名输入/输出描述
func_name输入InferShape函数名,开发者自定义。

返回值​

无