1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | enum Format {
FORMAT_NCHW = 0, // NCHW
FORMAT_NHWC, // NHWC
FORMAT_ND, // Nd Tensor
FORMAT_NC1HWC0, // NC1HWC0
FORMAT_FRACTAL_Z, // FRACTAL_Z
FORMAT_NC1C0HWPAD = 5,
FORMAT_NHWC1C0,
FORMAT_FSR_NCHW,
FORMAT_FRACTAL_DECONV,
FORMAT_C1HWNC0,
FORMAT_FRACTAL_DECONV_TRANSPOSE = 10,
FORMAT_FRACTAL_DECONV_SP_STRIDE_TRANS,
FORMAT_NC1HWC0_C04, // NC1HWC0, C0 is 4
FORMAT_FRACTAL_Z_C04, // FRACZ, C0 is 4
FORMAT_CHWN,
FORMAT_FRACTAL_DECONV_SP_STRIDE8_TRANS = 15,
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,
FORMAT_C1HWNCoC0 = 25,
FORMAT_MD,
FORMAT_NDHWC,
FORMAT_FRACTAL_ZZ,
FORMAT_FRACTAL_NZ,
FORMAT_NCDHW = 30,
FORMAT_DHWCN, // 3D filter input tensor format
FORMAT_NDC1HWC0,
FORMAT_FRACTAL_Z_3D,
FORMAT_CN,
FORMAT_NC = 35,
FORMAT_DHWNC,
FORMAT_FRACTAL_Z_3D_TRANSPOSE, // 3D filter(transpose) input tensor format
FORMAT_FRACTAL_ZN_LSTM,
FORMAT_FRACTAL_Z_G,
FORMAT_RESERVED = 40,
FORMAT_ALL,
FORMAT_NULL,
FORMAT_ND_RNN_BIAS,
FORMAT_FRACTAL_ZN_RNN,
FORMAT_NYUV = 45,
FORMAT_NYUV_A,
FORMAT_NCL,
FORMAT_FRACTAL_Z_WINO,
FORMAT_C1HWC0,
// Add new formats definition here
FORMAT_END,
// FORMAT_MAX defines the max value of Format.
// Any Format should not exceed the value of FORMAT_MAX.
// ** Attention ** : FORMAT_MAX stands for the SPEC of enum Format and almost SHOULD NOT be used in code.
// If you want to judge the range of Format, you can use FORMAT_END.
FORMAT_MAX = 0xff
};
|
上述Format枚举值对应的序号取值,从0开始,依次递增。
IR构图不支持输入以下FORMAT:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | FORMAT_NC1HWC0
FORMAT_FRACTAL_Z
FORMAT_NC1C0HWPAD
FORMAT_NHWC1C0
FORMAT_FRACTAL_DECONV
FORMAT_C1HWNC0
FORMAT_FRACTAL_DECONV_TRANSPOSE
FORMAT_FRACTAL_DECONV_SP_STRIDE_TRANS
FORMAT_NC1HWC0_C04
FORMAT_FRACTAL_Z_C04
FORMAT_FRACTAL_DECONV_SP_STRIDE8_TRANS
FORMAT_NC1KHKWHWC0
FORMAT_C1HWNCoC0
FORMAT_FRACTAL_ZZ
FORMAT_FRACTAL_NZ
FORMAT_NDC1HWC0
FORMAT_FRACTAL_Z_3D
FORMAT_FRACTAL_Z_3D_TRANSPOSE
FORMAT_FRACTAL_ZN_LSTM
FORMAT_FRACTAL_Z_G
FORMAT_ND_RNN_BIAS
FORMAT_FRACTAL_ZN_RNN
FORMAT_NYUV
FORMAT_NYUV_A
|