跳到主要内容

IMPLEMT_COMMON_INFERFUNC

函数功能

封装算子的Common_InferShape函数。

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

函数原型

IMPLEMT_COMMON_INFERFUNC(func_name)

约束说明

参数说明

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

返回值