IMPLEMT_COMMON_INFERFUNC
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
头文件
/include/graph/operator_reg.h
功能说明
封装算子的Common_InferShape函数。
与IMPLEMT_INFERFUNC的区别是,此函数自动生成的一个类型为Operator类的对象op,可直接调用Operator接口进行InferShape的实现。若InferShape方法具有通用性,可被多个算子的原型实现调用,可选择此接口实现。
函数原型
1
|
IMPLEMT_COMMON_INFERFUNC(func_name) |
参数说明
|
参数名 |
输入/输出 |
描述 |
|---|---|---|
|
func_name |
输入 |
InferShape函数名,用户自定义。 |
返回值说明
无
约束说明
无
父主题: 图基础数据结构和接口