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函数名保持一致。 |
返回值
无。
父主题: ge命名空间