ShapeInfo

【免费下载链接】asc-devkit 本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。 项目地址: https://gitcode.com/cann/asc-devkit

功能说明

ShapeInfo用来存放LocalTensor或GlobalTensor的shape信息。

函数原型

• ShapeInfo结构定义

  struct ShapeInfo {
  public:
      __aicore__ inline ShapeInfo();
      __aicore__ inline ShapeInfo(const uint8_t inputShapeDim, const uint32_t inputShape[],
          const uint8_t inputOriginalShapeDim, const uint32_t inputOriginalShape[], const DataFormat inputFormat);
      __aicore__ inline ShapeInfo(const uint8_t inputShapeDim, const uint32_t inputShape[], const DataFormat inputFormat);
      __aicore__ inline ShapeInfo(const uint8_t inputShapeDim, const uint32_t inputShape[]);
      uint8_t shapeDim;
      uint8_t originalShapeDim;
      uint32_t shape[K_MAX_DIM];
      uint32_t originalShape[K_MAX_DIM];
      DataFormat dataFormat;
  };
  

• 获取Shape中所有dim的累乘结果

  __aicore__ inline int GetShapeSize(const ShapeInfo& shapeInfo)
  

函数说明

表 1 ShapeInfo结构参数说明

enum class DataFormat : uint8_t {
    ND = 0,
    NZ,
    NCHW,
    NC1HWC0,
    NHWC,
};

表 2 GetShapeSize参数说明

【免费下载链接】asc-devkit 本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。 项目地址: https://gitcode.com/cann/asc-devkit