Home
last modified time | relevance | path

Searched refs: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
773 convolution->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
777 indirection_buffer_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
788 const size_t indirection_buffer_size = sizeof(void*) * batch_size * in pytorch_qnnp_setup_convolution_ndhwc_q8() local
792 convolution->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
796 indirection_buffer_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
H A Dmax-pooling.c206 const size_t indirection_buffer_size = sizeof(void*) * in pytorch_qnnp_setup_max_pooling2d_nhwc_u8() local
211 max_pooling->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
215 indirection_buffer_size); in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
H A Daverage-pooling.c267 const size_t indirection_buffer_size = sizeof(void*) * in pytorch_qnnp_setup_average_pooling2d_nhwc_q8() local
273 average_pooling->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
277 indirection_buffer_size); in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
H A Ddeconvolution.c283 const size_t indirection_buffer_size = in pytorch_qnnp_setup_deconvolution2d_nhwc_q8() local
287 deconvolution->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
291 indirection_buffer_size); in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
/aosp_15_r20/external/XNNPACK/src/operators/
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
188 … = (const void**) xnn_reallocate_memory(unpooling_op->indirection_buffer, indirection_buffer_size); in xnn_setup_unpooling2d_nhwc_x32()
192 indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_unpooling_nhwc_x32)); in xnn_setup_unpooling2d_nhwc_x32()
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
154 …fer = (const void**) xnn_reallocate_memory(resize_op->indirection_buffer, indirection_buffer_size); in xnn_setup_resize_bilinear2d_nchw_f32()
158indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_resize_bilinear_nchw_f32)); in xnn_setup_resize_bilinear2d_nchw_f32()
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
234 …onst void**) xnn_reallocate_memory(argmax_pooling_op->indirection_buffer, indirection_buffer_size); in xnn_setup_argmax_pooling2d_nhwc_f32()
238indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_argmax_pooling_nhwc_f32)); in xnn_setup_argmax_pooling2d_nhwc_f32()
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
237 …fer = (const void**) xnn_reallocate_memory(resize_op->indirection_buffer, indirection_buffer_size); in setup_resize_bilinear2d_nhwc()
241 indirection_buffer_size, xnn_operator_type_to_string(resize_op->type)); in setup_resize_bilinear2d_nhwc()
H A Ddeconvolution-nhwc.c696 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_conv_path() local
701 …const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size); in setup_conv_path()
705 indirection_buffer_size, xnn_operator_type_to_string(deconvolution_op->type)); in setup_conv_path()
923 const size_t indirection_buffer_size = sizeof(void*) * in setup_subconv2d_path() local
927 …const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size); in setup_subconv2d_path()
931 indirection_buffer_size, xnn_operator_type_to_string(deconvolution_op->type)); in setup_subconv2d_path()
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
289 … (const void**) xnn_reallocate_memory(max_pooling_op->indirection_buffer, indirection_buffer_size); in setup_max_pooling2d_nhwc()
291 xnn_log_error("failed to allocate %zu bytes for indirection buffer", indirection_buffer_size); in setup_max_pooling2d_nhwc()
H A Dconvolution-nhwc.c1563 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_convolution2d_nhwc() local
1568 …void**) xnn_reallocate_memory((void*) convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1572 indirection_buffer_size, xnn_operator_type_to_string(convolution_op->type)); in setup_convolution2d_nhwc()
1717 const size_t indirection_buffer_size = in setup_convolution2d_nhwc() local
1721 … (const void**) xnn_reallocate_memory(convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1724 indirection_buffer_size, xnn_operator_type_to_string(convolution_op->type)); in setup_convolution2d_nhwc()
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
735 …nst void**) xnn_reallocate_memory(average_pooling_op->indirection_buffer, indirection_buffer_size); in setup_average_pooling2d()
738 indirection_buffer_size, xnn_operator_type_to_string(average_pooling_op->type)); in setup_average_pooling2d()