更新时间:2021-10-26 GMT+08:00
分享

WeightFormats函数

函数功能

设置算子权重支持的数据格式。

函数原型

OpRegistrationData& WeightFormats(

const std::initializer_list<domi::tagDomiTensorFormat>& weight_formats);

参数说明

参数

输入/输出

说明

weight_formats

输入

权重支持的数据格式列表:

typedef enum tagDomiTensorFormat

{

DOMI_TENSOR_NCHW = 0, /**< NCHW */

DOMI_TENSOR_NHWC, /**< NHWC */

DOMI_TENSOR_ND, /**< Nd Tensor */

DOMI_TENSOR_NC1HWC0, /**< NC1HWC0 */

DOMI_TENSOR_FRACTAL_Z, /**< FRACTAL_Z */

DOMI_TENSOR_NC1C0HWPAD,

DOMI_TENSOR_NHWC1C0,

DOMI_TENSOR_FSR_NCHW,

DOMI_TENSOR_FRACTAL_DECONV,

DOMI_TENSOR_BN_WEIGHT,

DOMI_TENSOR_CHWN, /*Android NN Depth CONV*/

DOMI_TENSOR_FILTER_HWCK, /* filter input tensor format */

DOMI_TENSOR_RESERVED

} domiTensorFormat_t;

分享:

    相关文档

    相关产品