Home
last modified time | relevance | path

Searched defs:indirection_buffer_size (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dconvolution.c769 const size_t indirection_buffer_size = in pytorch_qnnp_setup_convolution_ndhwc_q8() local
788 const size_t indirection_buffer_size = sizeof(void*) * batch_size * in pytorch_qnnp_setup_convolution_ndhwc_q8() local
H A Dmax-pooling.c206 const size_t indirection_buffer_size = sizeof(void*) * in pytorch_qnnp_setup_max_pooling2d_nhwc_u8() local
H A Daverage-pooling.c267 const size_t indirection_buffer_size = sizeof(void*) * in pytorch_qnnp_setup_average_pooling2d_nhwc_q8() local
H A Ddeconvolution.c283 const size_t indirection_buffer_size = in pytorch_qnnp_setup_deconvolution2d_nhwc_q8() local
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dresize-bilinear-nchw.c151 const size_t indirection_buffer_size = sizeof(void*) * (output_height * output_width * 4); in xnn_setup_resize_bilinear2d_nchw_f32() local
H A Dunpooling-nhwc.c187 …const size_t indirection_buffer_size = sizeof(void*) * (batch_size * input_height * input_width * … in xnn_setup_unpooling2d_nhwc_x32() local
H A Dargmax-pooling-nhwc.c231 const size_t indirection_buffer_size = sizeof(void*) * ((mr - 1) + output_height * step_height); in xnn_setup_argmax_pooling2d_nhwc_f32() local
H A Ddeconvolution-nhwc.c696 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_conv_path() local
923 const size_t indirection_buffer_size = sizeof(void*) * in setup_subconv2d_path() local
H A Dresize-bilinear-nhwc.c234 const size_t indirection_buffer_size = sizeof(void*) * (output_height * output_width * 4); in setup_resize_bilinear2d_nhwc() local
H A Dconvolution-nhwc.c1563 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_convolution2d_nhwc() local
1717 const size_t indirection_buffer_size = in setup_convolution2d_nhwc() local
H A Dmax-pooling-nhwc.c287 const size_t indirection_buffer_size = sizeof(void*) * ((mr - 1) + output_height * step_height); in setup_max_pooling2d_nhwc() local
H A Daverage-pooling-nhwc.c732 …const size_t indirection_buffer_size = sizeof(void*) * ((primary_tile - 1) + output_height * step_… in setup_average_pooling2d() local