Home
last modified time | relevance | path

Searched refs:pytorch_qnnp_params (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Daverage-pooling.cc29 for (size_t channels = pytorch_qnnp_params.q8avgpool.kr; in TEST()
30 channels <= 3 * pytorch_qnnp_params.q8avgpool.kr; in TEST()
32 for (size_t poolSize = 2; poolSize <= pytorch_qnnp_params.q8avgpool.mr; in TEST()
48 for (size_t channels = pytorch_qnnp_params.q8avgpool.kr; in TEST()
49 channels <= 3 * pytorch_qnnp_params.q8avgpool.kr; in TEST()
51 for (size_t poolSize = 3; poolSize <= pytorch_qnnp_params.q8avgpool.mr; in TEST()
70 for (size_t channels = pytorch_qnnp_params.q8avgpool.kr; in TEST()
71 channels <= 3 * pytorch_qnnp_params.q8avgpool.kr; in TEST()
73 for (size_t poolSize = 2; poolSize <= pytorch_qnnp_params.q8avgpool.mr; in TEST()
90 for (size_t channels = pytorch_qnnp_params.q8avgpool.kr; in TEST()
[all …]
H A Dmax-pooling.cc29 for (size_t channels = pytorch_qnnp_params.u8maxpool.kr; in TEST()
30 channels <= 3 * pytorch_qnnp_params.u8maxpool.kr; in TEST()
32 for (size_t poolSize = 2; poolSize <= pytorch_qnnp_params.u8maxpool.mr; in TEST()
48 for (size_t channels = pytorch_qnnp_params.u8maxpool.kr; in TEST()
49 channels <= 3 * pytorch_qnnp_params.u8maxpool.kr; in TEST()
51 for (size_t poolSize = 3; poolSize <= pytorch_qnnp_params.u8maxpool.mr; in TEST()
72 for (size_t channels = pytorch_qnnp_params.u8maxpool.kr; in TEST()
73 channels <= 3 * pytorch_qnnp_params.u8maxpool.kr; in TEST()
75 for (size_t poolSize = 2; poolSize <= pytorch_qnnp_params.u8maxpool.mr; in TEST()
92 for (size_t channels = pytorch_qnnp_params.u8maxpool.kr; in TEST()
[all …]
H A Dglobal-average-pooling.cc26 for (size_t channels = pytorch_qnnp_params.q8gavgpool.nr; in TEST()
27 channels <= 3 * pytorch_qnnp_params.q8gavgpool.nr; in TEST()
29 for (size_t width = 1; width <= pytorch_qnnp_params.q8gavgpool.mr; in TEST()
44 for (size_t channels = pytorch_qnnp_params.q8gavgpool.nr; in TEST()
45 channels <= 3 * pytorch_qnnp_params.q8gavgpool.nr; in TEST()
47 for (size_t width = 1; width <= pytorch_qnnp_params.q8gavgpool.mr; in TEST()
53 .inputStride(5 * pytorch_qnnp_params.q8gavgpool.nr) in TEST()
63 for (size_t channels = pytorch_qnnp_params.q8gavgpool.nr; in TEST()
64 channels <= 3 * pytorch_qnnp_params.q8gavgpool.nr; in TEST()
66 for (size_t width = 1; width <= pytorch_qnnp_params.q8gavgpool.mr; in TEST()
[all …]
H A Dconvolution.cc107 if (pytorch_qnnp_params.q8conv_xzp.kthreshold != SIZE_MAX) { in TEST()
111 .groupInputChannels(pytorch_qnnp_params.q8conv_xzp.kthreshold + 1) in TEST()
120 if (pytorch_qnnp_params.q8conv_xzp.kthreshold != SIZE_MAX) { in TEST()
124 .groupInputChannels(pytorch_qnnp_params.q8conv_xzp.kthreshold + 1) in TEST()
134 if (pytorch_qnnp_params.q8conv_xzp.kthreshold != SIZE_MAX) { in TEST()
138 .groupInputChannels(pytorch_qnnp_params.q8conv_xzp.kthreshold + 1) in TEST()
148 if (pytorch_qnnp_params.q8conv_xzp.kthreshold != SIZE_MAX) { in TEST()
152 .inputPixelStride(pytorch_qnnp_params.q8conv_xzp.kthreshold + 5) in TEST()
153 .groupInputChannels(pytorch_qnnp_params.q8conv_xzp.kthreshold + 1) in TEST()
162 if (pytorch_qnnp_params.q8conv_xzp.kthreshold != SIZE_MAX) { in TEST()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dinit.c45 struct pytorch_qnnp_parameters pytorch_qnnp_params = {.initialized = false}; variable
54 pytorch_qnnp_params.q8conv = (struct pytorch_q8conv_parameters){ in init()
62 pytorch_qnnp_params.q8gemm_sparse_c1x4 = (struct pytorch_q8gemm_sparse_parameters){ in init()
76 pytorch_qnnp_params.q8gemm_sparse_c8x1 = (struct pytorch_q8gemm_sparse_parameters){ in init()
91 pytorch_qnnp_params.q8conv_xzp = (struct pytorch_q8conv_xzp_parameters){ in init()
102 pytorch_qnnp_params.q8conv_xzp.kthreshold = 64; in init()
105 pytorch_qnnp_params.q8conv_xzp.kthreshold = 256; in init()
108 pytorch_qnnp_params.q8conv_xzp.kthreshold = 32; in init()
111 pytorch_qnnp_params.q8conv_xzp.kthreshold = 16; in init()
117 pytorch_qnnp_params.q8conv_xzp = (struct pytorch_q8conv_xzp_parameters){ in init()
[all …]
H A Doperator-run.c930 ? pytorch_qnnp_params.q8dw9.updw_per_channel in pytorch_qnnp_run_operator()
931 : pytorch_qnnp_params.q8dw9.updw, in pytorch_qnnp_run_operator()
958 ? pytorch_qnnp_params.q8dw25.mpdw_per_channel in pytorch_qnnp_run_operator()
959 : pytorch_qnnp_params.q8dw25.mpdw, in pytorch_qnnp_run_operator()
986 .multipass_ukernel = pytorch_qnnp_params.q8dw27.mpdw, in pytorch_qnnp_run_operator()
1006 const uint32_t mr = pytorch_qnnp_params.q8conv_xzp.mr; in pytorch_qnnp_run_operator()
1007 const uint32_t nr = pytorch_qnnp_params.q8conv_xzp.nr; in pytorch_qnnp_run_operator()
1008 const uint32_t kr = pytorch_qnnp_params.q8conv_xzp.kr; in pytorch_qnnp_run_operator()
1025 .ukernel = pytorch_qnnp_params.q8sum_rows.sum_rows, in pytorch_qnnp_run_operator()
1036 pytorch_qnnp_params.q8sum_rows.m); in pytorch_qnnp_run_operator()
[all …]
H A Dconv-run.cc413 const uint32_t cr = pytorch_qnnp_params.q8dw9.cr; in qnnpackConv()
439 ? pytorch_qnnp_params.q8dw9.updw_per_channel in qnnpackConv()
440 : pytorch_qnnp_params.q8dw9.updw, in qnnpackConv()
442 ? pytorch_qnnp_params.q8dw25.mpdw_per_channel in qnnpackConv()
443 : pytorch_qnnp_params.q8dw25.mpdw, in qnnpackConv()
472 ? pytorch_qnnp_params.q8dw9.updw_per_channel in qnnpackConv()
473 : pytorch_qnnp_params.q8dw9.updw, in qnnpackConv()
475 ? pytorch_qnnp_params.q8dw25.mpdw_per_channel in qnnpackConv()
476 : pytorch_qnnp_params.q8dw25.mpdw, in qnnpackConv()
505 .multipass_ukernel = pytorch_qnnp_params.q8dw27.mpdw, in qnnpackConv()
[all …]
H A Daverage-pooling.c51 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
147 const uint32_t kr = pytorch_qnnp_params.q8avgpool.kr; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
148 const uint32_t mr = pytorch_qnnp_params.q8avgpool.mr; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
149 const uint32_t qr = pytorch_qnnp_params.q8avgpool.qr; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
177 if (channels >= pytorch_qnnp_params.q8avgpool.kr) { in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
214 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
264 const uint32_t mr = pytorch_qnnp_params.q8avgpool.mr; in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
H A Dconv-prepack.cc32 const uint32_t cr = pytorch_qnnp_params.q8dw9.cr; in PrePackConvWeights()
161 const uint32_t nr = pytorch_qnnp_params.q8conv_xzp.nr; in PrePackConvWeights()
162 const uint32_t kr = pytorch_qnnp_params.q8conv_xzp.kr; in PrePackConvWeights()
163 const uint32_t sr = pytorch_qnnp_params.q8conv_xzp.kc; in PrePackConvWeights()
199 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in PrePackConvWeights()
200 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in PrePackConvWeights()
H A Dconvolution.c74 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_convolution_ndhwc_q8()
215 group_input_channels >= pytorch_qnnp_params.q8conv_xzp.kthreshold in pytorch_qnnp_create_convolution_ndhwc_q8()
227 const uint32_t cr = pytorch_qnnp_params.q8dw9.cr; in pytorch_qnnp_create_convolution_ndhwc_q8()
372 const uint32_t nr = pytorch_qnnp_params.q8conv_xzp.nr; in pytorch_qnnp_create_convolution_ndhwc_q8()
373 const uint32_t kr = pytorch_qnnp_params.q8conv_xzp.kr; in pytorch_qnnp_create_convolution_ndhwc_q8()
374 const uint32_t sr = pytorch_qnnp_params.q8conv_xzp.kc; in pytorch_qnnp_create_convolution_ndhwc_q8()
411 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in pytorch_qnnp_create_convolution_ndhwc_q8()
412 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in pytorch_qnnp_create_convolution_ndhwc_q8()
693 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_convolution_ndhwc_q8()
767 const size_t output_tile_size = pytorch_qnnp_params.q8conv.mr; in pytorch_qnnp_setup_convolution_ndhwc_q8()
H A Dfully-connected.c40 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_fully_connected_nc_q8()
68 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in pytorch_qnnp_create_fully_connected_nc_q8()
69 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in pytorch_qnnp_create_fully_connected_nc_q8()
137 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_fully_connected_nc_q8()
H A Ddeconvolution.c65 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
125 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
126 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
230 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
281 const size_t output_tile_size = pytorch_qnnp_params.q8conv.mr; in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
H A Dfc-dynamic-run.cc75 const uint32_t mr = pytorch_qnnp_params.q8conv.mr; in qnnpackLinearDynamic()
76 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in qnnpackLinearDynamic()
77 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in qnnpackLinearDynamic()
100 .ukernel = pytorch_qnnp_params.q8conv.gemm_dq, in qnnpackLinearDynamic()
H A Dfc-run.cc75 const uint32_t mr = pytorch_qnnp_params.q8conv.mr; in qnnpackLinear()
76 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in qnnpackLinear()
77 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in qnnpackLinear()
98 .ukernel = pytorch_qnnp_params.q8conv.gemm, in qnnpackLinear()
H A Ddeconv-run.cc116 const uint32_t mr = pytorch_qnnp_params.q8conv.mr; in qnnpackDeConv()
117 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in qnnpackDeConv()
118 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in qnnpackDeConv()
192 .ukernel = pytorch_qnnp_params.q8conv.conv, in qnnpackDeConv()
H A Dfully-connected-sparse.c45 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
76 if (pytorch_qnnp_params.q8gemm_sparse_c8x1.packA == NULL) { in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
146 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_fully_connected_sparse_dq_nc_q8()
H A Dmax-pooling.c54 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
153 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
203 const uint32_t mr = pytorch_qnnp_params.u8maxpool.mr; in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
H A Dfc-prepack.cc30 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in PackBMatrix()
31 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in PackBMatrix()
H A Dchannel-shuffle.c28 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_channel_shuffle_nc_x8()
83 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_channel_shuffle_nc_x8()
H A Dclamp.c28 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_clamp_nc_u8()
84 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_clamp_nc_u8()
H A Dfc-unpack.cc24 const uint32_t nr = pytorch_qnnp_params.q8conv.nr; in unpackWeights()
25 const uint32_t kr = pytorch_qnnp_params.q8conv.kr; in unpackWeights()
H A Dsoftargmax.c29 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_softargmax_nc_q8()
121 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_softargmax_nc_q8()
H A Dadd.c36 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_add_nc_q8()
140 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_add_nc_q8()
H A Dglobal-average-pooling.c34 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_global_average_pooling_nwc_q8()
122 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_global_average_pooling_nwc_q8()
H A Dsigmoid.c32 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_create_sigmoid_nc_q8()
141 if (!pytorch_qnnp_params.initialized) { in pytorch_qnnp_setup_sigmoid_nc_q8()

12