TensorDesc类接口

函数原型

void Update(Shape shape, Format format = FORMAT_NCHW, DataType dt = DT_FLOAT);

功能说明

更新TensorDesc对象的shape、format、datatype属性。

参数说明

返回值

无。

异常处理

无。

约束说明

无。

数据类型说明

• Format

   enum Format {
          FORMAT_NCHW = 0,         /**< NCHW */
          FORMAT_NHWC,             /**< NHWC */
          FORMAT_ND,               /**< Nd Tensor */
          FORMAT_NC1HWC0,          /**< NC1HWC0 */
          FORMAT_FRACTAL_Z,        /**< FRACTAL_Z */
          FORMAT_NC1C0HWPAD,
          FORMAT_NHWC1C0,
          FORMAT_FSR_NCHW,
          FORMAT_FRACTAL_DECONV,
          FORMAT_C1HWNC0,
          FORMAT_FRACTAL_DECONV_TRANSPOSE,
          FORMAT_FRACTAL_DECONV_SP_STRIDE_TRANS,
          FORMAT_NC1HWC0_C04,  /**< NC1HWC0, C0 =4*/
          FORMAT_FRACTAL_Z_C04,/**< FRACZ格式，C0 =4 */
          FORMAT_CHWN,
          FORMAT_FRACTAL_DECONV_SP_STRIDE8_TRANS,
          FORMAT_HWCN,
          FORMAT_NC1KHKWHWC0, /** < KH,KW kernel h& kernel w maxpooling max output format*/
          FORMAT_BN_WEIGHT,
          FORMAT_FILTER_HWCK,     /* filter input tensor format */
          FORMAT_HASHTABLE_LOOKUP_LOOKUPS=20,
          FORMAT_HASHTABLE_LOOKUP_KEYS,
          FORMAT_HASHTABLE_LOOKUP_VALUE,
          FORMAT_HASHTABLE_LOOKUP_OUTPUT,
          FORMAT_HASHTABLE_LOOKUP_HITS=24,
          FORMAT_RESERVED
  
      };

• DataType

  enum DataType {
          DT_UNDEFINED = 16,   // Used to indicate a DataType field has not been set.
          DT_FLOAT = 0,           // float type
          DT_FLOAT16 = 1,         // fp16 type
          DT_INT8 = 2,            // int8 type
          DT_INT16 = 6,     // int16 type
          DT_UINT16 = 7,      // uint16 type
          DT_UINT8 = 4,           // uint8 type
          DT_INT32 = 3,           //
          DT_INT64 = 9,           // int64 type
          DT_UINT32 = 8,          // unsigned int32
          DT_UINT64 = 10,          // unsigned int64
          DT_BOOL = 12,            // bool type
          DT_DOUBLE = 11,          // double type
          DT_DUAL = 13,              /**< dual output type */
          DT_DUAL_SUB_INT8 = 14,    /**< dual output int8 type */
          DT_DUAL_SUB_UINT8 = 15,    /**< dual output uint8 type */
      };

函数原型

Shape GetShape() const;

功能说明

获取TensorDesc描述的shape。

参数说明

无。

返回值

异常处理

无。

约束说明

返回的shape为const，不可修改。

函数原型

Shape& MutableShape();

功能说明

获取TensorDesc中可修改的shape引用。

参数说明

无。

返回值

异常处理

无。

约束说明

无。

函数原型

void SetShape(Shape shape);

功能说明

设置TensorDesc的shape。

参数说明

返回值

无。

异常处理

无。

约束说明

无。

函数原型

Format GetFormat() const;

功能说明

获取TensorDesc所描述的Tensor的format信息。

参数说明

无。

返回值

异常处理

无。

约束说明

无。

函数原型

void SetFormat(Format format);

功能说明

设置TensorDesc所描述的Tensor的format信息。

参数说明

返回值

无。

异常处理

无。

约束说明

无。

函数原型

DataType GetDataType() const;

功能说明

获取TensorDesc所描述的Tensor的数据类型。

参数说明

无。

返回值

异常处理

无。

约束说明

无。

函数原型

void SetDataType(DataType dt);

功能说明

设置TensorDesc所描述的Tensor的数据类型。

参数说明

返回值

无。

异常处理

无。

约束说明

无。

函数原型

TensorDesc Clone() const;

功能说明

拷贝TensorDesc。

参数说明

无。

返回值

异常处理

无。

约束说明

无。

函数原型

graphStatus IsValid();

功能说明

判断tensor对象是否有效。

参数说明

无。

返回值

异常处理

无。

约束说明

无。