公共定义
定义
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;
说明
常量名称 |
参数值说明 |
|---|---|
Status |
类型定义，用一系列状态值表示加速库中的返回值。 |
ReshapeFunc |
类型定义，Reshape功能，改变Tensor的shape。 |
InferShapeFunc |
类型定义，根据输入tensor信息推导输出tensor信息。 |
MAX_DIM |
变量，数据最大维度定义。 |
父主题： atb/types.h