产品支持情况
产品
|
是否支持
|
|---|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
头文件
#include <graph/operator_reg.h>
功能说明
封装算子的inferFormat函数。
函数原型
IMPLEMT_INFERFORMAT_FUNC(op_name, func_name)
参数说明
参数名
|
输入/输出
|
描述
|
|---|
op_name
|
输入
|
算子类型。
|
func_name
|
输入
|
inferFormat函数名，用户自定义。
|
返回值说明
无
约束说明
无