Help Center/ Atlas 300 Application (Model 3000)/ TE Custom Operator Development Guide/ Function Reference for Operator Plug-In Development/ WeightFormats
Updated on 2022-03-13 GMT+08:00
View PDF
Share

WeightFormats

Function

Sets the data formats supported by the operator weight.

Syntax

OpRegistrationData& WeightFormats(

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

Parameter Description

Parameter

Input/Output

Description

weight_formats

Input

Data formats supported by the weight:

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;