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

INFER_FUNC_REG

产品支持情况

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

#include <graph/operator_reg.h>

注册算子的InferShape函数。

INFER_FUNC_REG(op_name, x)

该函数内部会自动调用INFER_VERIFY_FUNC(op_name, x)，INFER_VERIFY_FUNC函数中的op_name为算子的类型，x为指向INFER_FUNC_REG（op_name,x）中“x”的指针。

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

无

无