INFER_FORMAT_FUNC_REG

Applicability

Header File

#include <graph/operator_reg.h>

Function Usage

Registers the InferFormat function of an operator.

The GE issues a graph InferFormat call before and after shape and dtype inference, respectively. Between the two calls, the operator InferFormat function is called for each operator. If an operator has no corresponding registered InferFormat function, the GE uses the default inference function, which means that the output format is set to the input format.

Prototype

1
2

#define INFER_FORMAT_FUNC_REG(op_name, x) \
__INFER_FORMAT_FUNC_REG_IMPL__(op_name, INFER_FORMAT_FUNC(op_name, x), __COUNTER__)

Parameters

Returns

None

Restrictions

None

Call Examples and Related APIs

1

INFER_FORMAT_FUNC_REG(Transpose, TransposeInferFormat);