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 D | convolution.c | 769 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 D | max-pooling.c | 206 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 D | average-pooling.c | 267 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 D | deconvolution.c | 283 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 D | unpooling-nhwc.c | 187 …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 D | resize-bilinear-nchw.c | 151 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() 158 … indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_resize_bilinear_nchw_f32)); in xnn_setup_resize_bilinear2d_nchw_f32()
|
H A D | argmax-pooling-nhwc.c | 231 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() 238 … indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_argmax_pooling_nhwc_f32)); in xnn_setup_argmax_pooling2d_nhwc_f32()
|
H A D | resize-bilinear-nhwc.c | 234 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 D | deconvolution-nhwc.c | 696 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 D | max-pooling-nhwc.c | 287 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 D | convolution-nhwc.c | 1563 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 D | average-pooling-nhwc.c | 732 …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()
|