INFER_FORMAT_FUNC_REG

函数功能

注册算子的InferFormat实现。

GE会在整图的Shape与Dtype推导前后分别调用一次整图的InferFormat，过程中会分别调用各个算子的InferFormat函数。如果算子没有注册InferFormat函数，GE将使用默认的推导函数，即输出的Format等于输入的Format。

函数原型

#define INFER_FORMAT_FUNC_REG(op_name, x) \

__INFER_FORMAT_FUNC_REG_IMPL__(op_name, INFER_FORMAT_FUNC(op_name, x), __COUNTER__)

参数说明

返回值

无

约束和限制说明

无

客户是否可用

是

调用示例和相关API

INFER_FORMAT_FUNC_REG(Transpose, TransposeInferFormat);