更新时间:2021-03-18 GMT+08:00
分享

Update

函数原型

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

功能说明

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

参数说明

参数名

输入/输出

类型

描述

shape

输入

Shape

需刷新的shape对象。

format

输入

Format

需刷新的format对象,默认取值FORMAT_NCHW。

关于Format数据类型的定义,请参见•Format

dt

输入

DataType

需刷新的datatype对象,默认取值DT_FLOAT。

关于DataType数据类型的定义,请参见•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 */
        };
分享:

    相关文档

    相关产品