跳到主要内容

COMMON_INFER_FUNC_REG

函数功能​

注册算子的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函数名保持一致。

返回值​

无