Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A 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
H A 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
H A 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
H A 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
H A Dconv-prepack.cc164 const uint32_t n_stride = in PrePackConvWeights() local
201 const uint32_t n_stride = in PrePackConvWeights() local
H A 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
H A 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
H A Dfc-prepack.cc33 const uint32_t n_stride = (output_channels + (nr - 1)) & -nr; in PackBMatrix() local
H A Dfully-connected.c71 const uint32_t n_stride = (output_channels + (nr - 1)) & -nr; in pytorch_qnnp_create_fully_connected_nc_q8() local
H A Ddeconvolution.c128 const uint32_t n_stride = (group_output_channels + (nr - 1)) & -nr; in pytorch_qnnp_create_deconvolution2d_nhwc_q8() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/pool3d/neon/
H A 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
H A 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
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dfully-connected-nc.c118 const size_t n_stride = round_up(output_channels, nr); in create_fully_connected_nc() local
H A Ddeconvolution-nhwc.c158 const uint32_t n_stride = round_up(group_output_channels, nr); in create_deconvolution2d_nhwc() local
H A Dconvolution-nhwc.c496 const size_t n_stride = round_up(group_output_channels, nr); in create_convolution2d_nhwc() local