Home
last modified time | relevance | path

Searched defs:n_stride (Results 1 – 15 of 15) sorted by relevance

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dconv-prepack.cc164 const uint32_t n_stride = in PrePackConvWeights() local
201 const uint32_t n_stride = in PrePackConvWeights() local
Dfc-run.cc10 size_t n_stride; member
34 const size_t n_stride = context->n_stride; in compute_q8gemm() local
79 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in qnnpackLinear() local
Dfc-dynamic-run.cc10 size_t n_stride; member
34 const size_t n_stride = context->n_stride; in compute_q8gemm_dq() local
79 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in qnnpackLinearDynamic() local
Dconv-run.cc16 size_t n_stride; member
42 const size_t n_stride = context->n_stride; in compute_q8gemm_xzp() local
71 size_t n_stride; member
93 const size_t n_stride = context->n_stride; in compute_q8gemm() local
123 size_t n_stride; member
148 const size_t n_stride = context->n_stride; in compute_q8conv() local
527 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in qnnpackConv() local
604 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in qnnpackConv() local
641 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in qnnpackConv() local
Ddeconv-run.cc35 size_t n_stride; member
61 const size_t n_stride = context->n_stride; in compute_q8conv() local
120 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in qnnpackDeConv() local
Dfc-prepack.cc33 const uint32_t n_stride = (output_channels + (nr - 1)) & -nr; in PackBMatrix() local
Dfully-connected.c71 const uint32_t n_stride = (output_channels + (nr - 1)) & -nr; in pytorch_qnnp_create_fully_connected_nc_q8() local
Dconvolution.c375 const uint32_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_create_convolution_ndhwc_q8() local
413 const uint32_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_create_convolution_ndhwc_q8() local
Doperator-run.c29 size_t n_stride; member
52 const size_t n_stride = context->n_stride; in compute_q8gemm() local
298 size_t n_stride; member
325 const size_t n_stride = context->n_stride; in compute_q8gemm_xzp() local
358 size_t n_stride; member
384 const size_t n_stride = context->n_stride; in compute_q8conv() local
1010 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_run_operator() local
1078 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_run_operator() local
1289 const size_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_run_operator() local
Ddeconvolution.c128 const uint32_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_create_deconvolution2d_nhwc_q8() local
/external/ComputeLibrary/src/cpu/kernels/pool3d/neon/
Dquantized.h76 const int n_stride = static_cast<int>(src->info()->strides_in_bytes()[4]); in avg_poolingMxNxD_q8_neon_ndhwc() local
257 const int n_stride = static_cast<int>(src->info()->strides_in_bytes()[4]); in max_poolingMxNxD_q8_neon_ndhwc() local
Dimpl.cpp66 const int n_stride = static_cast<int>(src->info()->strides_in_bytes()[4]); in max_poolingMxNxD_fp_neon_ndhwc() local
178 const int n_stride = static_cast<int>(src->info()->strides_in_bytes()[4]); in avg_poolingMxNxD_fp_neon_ndhwc() local
305 const int n_stride = static_cast<int>(src->info()->strides_in_bytes()[4]); in l2_poolingMxNxD_fp_neon_ndhwc() local
/external/XNNPACK/src/operators/
Dfully-connected-nc.c118 const size_t n_stride = round_up(output_channels, nr); in create_fully_connected_nc() local
Ddeconvolution-nhwc.c158 const uint32_t n_stride = round_up(group_output_channels, nr); in create_deconvolution2d_nhwc() local
Dconvolution-nhwc.c496 const size_t n_stride = round_up(group_output_channels, nr); in create_convolution2d_nhwc() local