Home
last modified time | relevance | path

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

/external/XNNPACK/src/operators/
Dunpooling-nhwc.c195 …const size_t indirection_buffer_size = sizeof(void*) * (batch_size * input_height * input_width * … in xnn_setup_unpooling2d_nhwc_x32() local
196 … = (const void**) xnn_reallocate_memory(unpooling_op->indirection_buffer, indirection_buffer_size); in xnn_setup_unpooling2d_nhwc_x32()
200 indirection_buffer_size, xnn_operator_type_to_string(xnn_operator_type_unpooling_nhwc_x32)); in xnn_setup_unpooling2d_nhwc_x32()
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()
158 … 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()
Dresize-bilinear-nhwc.c205 const size_t indirection_buffer_size = sizeof(void*) * (output_height * output_width * 4); in setup_resize_bilinear2d_nhwc() local
208 …fer = (const void**) xnn_reallocate_memory(resize_op->indirection_buffer, indirection_buffer_size); in setup_resize_bilinear2d_nhwc()
212 indirection_buffer_size, xnn_operator_type_to_string(resize_op->type)); in setup_resize_bilinear2d_nhwc()
Ddeconvolution-nhwc.c572 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_conv_path() local
577 …const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size); in setup_conv_path()
581 indirection_buffer_size, xnn_operator_type_to_string(deconvolution_op->type)); in setup_conv_path()
729 const size_t indirection_buffer_size = sizeof(void*) * in setup_subconv2d_path() local
733 …const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size); in setup_subconv2d_path()
737 indirection_buffer_size, xnn_operator_type_to_string(deconvolution_op->type)); in setup_subconv2d_path()
Dmax-pooling-nhwc.c276 const size_t indirection_buffer_size = sizeof(void*) * ((mr - 1) + output_height * step_height); in setup_max_pooling2d_nhwc() local
278 … (const void**) xnn_reallocate_memory(max_pooling_op->indirection_buffer, indirection_buffer_size); in setup_max_pooling2d_nhwc()
280 xnn_log_error("failed to allocate %zu bytes for indirection buffer", indirection_buffer_size); in setup_max_pooling2d_nhwc()
Dconvolution-nhwc.c1219 const size_t indirection_buffer_size = sizeof(void*) * kernel_size * tiled_output_size; in setup_convolution2d_nhwc() local
1224 …void**) xnn_reallocate_memory((void*) convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1228 indirection_buffer_size, xnn_operator_type_to_string(convolution_op->type)); in setup_convolution2d_nhwc()
1367 const size_t indirection_buffer_size = sizeof(void*) * output_height * step_height; in setup_convolution2d_nhwc() local
1370 … (const void**) xnn_reallocate_memory(convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1373 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*) * ((primary_tile - 1) + output_height * step_… 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()