d5b140e5创建于 2025年12月26日历史提交

COMMON_INFER_FUNC_REG

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

#include <graph/operator_reg.h>

注册算子的InferShape函数。

与INFER_FUNC_REG的区别是，此函数注册的InferShape函数入参为Operator基类而非子类，此接口支持多算子共用同一个InferShape函数。

COMMON_INFER_FUNC_REG(op_name, x)

该函数内部会自动调用COMMON_INFER_VERIFY_FUNC(x)，COMMON_INFER_VERIFY_FUNC(x)函数中的x为指向COMMON_INFER_FUNC_REG(op_name, x)中“x”的指针。

参数名	输入/输出	描述
op_name	输入	算子类型。
x	输入	InferShape函数名，和IMPLEMT_COMMON_INFERFUNC的InferShape函数名保持一致。

无

无