IMPLEMT_COMMON_INFERFUNC
函数功能
封装算子的Common_InferShape函数。
与IMPLEMT_INFERFUNC的区别是,此函数自动生成的一个类型为Operator类的对象op,可直接调用Operator接口进行InferShape的实现。若InferShape方法具有通用性,可被多个算子的原型实现调用,可选择此接口实现。
函数原型
IMPLEMT_COMMON_INFERFUNC(func_name)
约束说明
无。
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
func_name |
输入 |
InferShape函数名,用户自定义。 |
返回值
无。
父主题: ge命名空间