更新时间:2021-03-18 GMT+08:00
Update
函数原型
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 */ };
父主题: TensorDesc类接口