Common Definition

Definition

1
2
3
4
using Status = int32_t;
using ReshapeFunc = std::function< void(const Dims &oldShape, Dims &newShape)>;
using InferShapeFunc = std::function< Status(const SVector< TensorDesc > &inTensorDescs, SVector< TensorDesc > &outTensorDescs)>;
constexpr uint32_t MAX_DIM = 8;

Parameters

Constant

Description

Status

Type definition. A series of status values are used to indicate the return values in the ATB.

ReshapeFunc

Type definition, reshape function, used to change the shape of a tensor.

InferShapeFunc

Type definition. The output tensor information is derived from the input tensor information.

MAX_DIM

Variable, which is the definition of the maximum data dimension.