Home
last modified time | relevance | path

Searched refs:indirection_buffer_size (Results 1 – 8 of 8) sorted by relevance

/external/XNNPACK/src/operators/
Dresize-bilinear-nhwc.c150 const size_t indirection_buffer_size = sizeof(void*) * (output_height * output_width * 4); in xnn_setup_resize_bilinear2d_nhwc_f32() local
153 …fer = (const void**) xnn_reallocate_memory(resize_op->indirection_buffer, indirection_buffer_size); in xnn_setup_resize_bilinear2d_nhwc_f32()
157 … indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_resize_bilinear_nhwc_f32)); in xnn_setup_resize_bilinear2d_nhwc_f32()
Dunpooling-nhwc.c194 …const size_t indirection_buffer_size = sizeof(void*) * (batch_size * input_height * input_width * … in xnn_setup_unpooling2d_nhwc_x32() local
195 …buffer = (void**) xnn_reallocate_memory(unpooling_op->indirection_buffer, indirection_buffer_size); in xnn_setup_unpooling2d_nhwc_x32()
199 indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_unpooling_nhwc_x32)); in xnn_setup_unpooling2d_nhwc_x32()
Dresize-bilinear-nchw.c150 const size_t indirection_buffer_size = sizeof(void*) * (output_height * output_width * 4); in xnn_setup_resize_bilinear2d_nchw_f32() local
153 …fer = (const void**) xnn_reallocate_memory(resize_op->indirection_buffer, indirection_buffer_size); in xnn_setup_resize_bilinear2d_nchw_f32()
157 … indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_resize_bilinear_nchw_f32)); in xnn_setup_resize_bilinear2d_nchw_f32()
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()
238 … indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_argmax_pooling_nhwc_f32)); in xnn_setup_argmax_pooling2d_nhwc_f32()
Dmax-pooling-nhwc.c274 const size_t indirection_buffer_size = sizeof(void*) * ((mr - 1) + output_height * step_height); in setup_max_pooling2d_nhwc() local
276 … (const void**) xnn_reallocate_memory(max_pooling_op->indirection_buffer, indirection_buffer_size); in setup_max_pooling2d_nhwc()
278 xnn_log_error("failed to allocate %zu bytes for indirection buffer", indirection_buffer_size); in setup_max_pooling2d_nhwc()
Ddeconvolution-nhwc.c499 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_conv_path() local
504 …const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size); in setup_conv_path()
508 indirection_buffer_size, xnn_operator_type_to_string(deconvolution_op->type)); in setup_conv_path()
656 const size_t indirection_buffer_size = sizeof(void*) * in setup_subconv2d_path() local
660 …const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size); in setup_subconv2d_path()
664 indirection_buffer_size, xnn_operator_type_to_string(deconvolution_op->type)); in setup_subconv2d_path()
Dconvolution-nhwc.c949 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_convolution2d_nhwc() local
954 …void**) xnn_reallocate_memory((void*) convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
958 indirection_buffer_size, xnn_operator_type_to_string(convolution_op->type)); in setup_convolution2d_nhwc()
1096 const size_t indirection_buffer_size = sizeof(void*) * output_height * step_height; in setup_convolution2d_nhwc() local
1099 … (const void**) xnn_reallocate_memory(convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1102 indirection_buffer_size, xnn_operator_type_to_string(convolution_op->type)); in setup_convolution2d_nhwc()
Daverage-pooling-nhwc.c521 … const size_t indirection_buffer_size = sizeof(void*) * ((mr - 1) + output_height * step_height); in setup_average_pooling2d() local
524 …nst void**) xnn_reallocate_memory(average_pooling_op->indirection_buffer, indirection_buffer_size); in setup_average_pooling2d()
527 indirection_buffer_size, xnn_operator_type_to_string(average_pooling_op->type)); in setup_average_pooling2d()