Home
last modified time | relevance | path

Searched refs:indirection_buffer (Results 1 – 17 of 17) sorted by relevance

/external/XNNPACK/src/
Dindirection.c23 const void** indirection_buffer = op->indirection_buffer; in xnn_indirection_init_conv2d() local
60 indirection_buffer[index] = (const void*) in xnn_indirection_init_conv2d()
63 indirection_buffer[index] = zero; in xnn_indirection_init_conv2d()
70 indirection_buffer[index] = zero; in xnn_indirection_init_conv2d()
83 const void** indirection_buffer = op->indirection_buffer; in xnn_indirection_init_deconv2d() local
123indirection_buffer[index] = (const void*) ((uintptr_t) input + (input_y * input_width + input_x) *… in xnn_indirection_init_deconv2d()
125 indirection_buffer[index] = zero; in xnn_indirection_init_deconv2d()
138 const void** indirection_buffer = op->indirection_buffer; in xnn_indirection_init_subconv2d() local
162 subconvolution_params->indirection_buffer = indirection_buffer; in xnn_indirection_init_subconv2d()
184 *indirection_buffer++ = in xnn_indirection_init_subconv2d()
[all …]
Doperator-delete.c29 xnn_release_memory(op->indirection_buffer); in xnn_delete_operator()
Doperator-run.c298 …(const void**) ((uintptr_t) subconvolution_params->indirection_buffer + slice_y * subconvolution_p… in xnn_compute_grouped_subconv2d()
336 …(const void**) ((uintptr_t) subconvolution_params->indirection_buffer + slice_y * subconvolution_p… in xnn_compute_subconv2d()
/external/XNNPACK/src/operators/
Dresize-bilinear-nhwc.c153 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(resize_op->indirection_buff… in xnn_setup_resize_bilinear2d_nhwc_f32() local
154 if (indirection_buffer == NULL) { in xnn_setup_resize_bilinear2d_nhwc_f32()
160 resize_op->indirection_buffer = indirection_buffer; in xnn_setup_resize_bilinear2d_nhwc_f32()
184 input, resize_op->indirection_buffer, resize_op->packed_weights, in xnn_setup_resize_bilinear2d_nhwc_f32()
198 .indirect_input = resize_op->indirection_buffer, in xnn_setup_resize_bilinear2d_nhwc_f32()
Dresize-bilinear-nchw.c153 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(resize_op->indirection_buff… in xnn_setup_resize_bilinear2d_nchw_f32() local
154 if (indirection_buffer == NULL) { in xnn_setup_resize_bilinear2d_nchw_f32()
160 resize_op->indirection_buffer = indirection_buffer; in xnn_setup_resize_bilinear2d_nchw_f32()
184 input, resize_op->indirection_buffer, resize_op->packed_weights, in xnn_setup_resize_bilinear2d_nchw_f32()
199 .indirect_input = resize_op->indirection_buffer, in xnn_setup_resize_bilinear2d_nchw_f32()
Dunpooling-nhwc.c195 …void** indirection_buffer = (void**) xnn_reallocate_memory(unpooling_op->indirection_buffer, indir… in xnn_setup_unpooling2d_nhwc_x32() local
196 if (indirection_buffer == NULL) { in xnn_setup_unpooling2d_nhwc_x32()
202 unpooling_op->indirection_buffer = (const void**) indirection_buffer; in xnn_setup_unpooling2d_nhwc_x32()
215 .indirect_output = indirection_buffer, in xnn_setup_unpooling2d_nhwc_x32()
Dargmax-pooling-nhwc.c233 const void** indirection_buffer = in xnn_setup_argmax_pooling2d_nhwc_f32() local
234 …(const void**) xnn_reallocate_memory(argmax_pooling_op->indirection_buffer, indirection_buffer_siz… in xnn_setup_argmax_pooling2d_nhwc_f32()
235 if (indirection_buffer == NULL) { in xnn_setup_argmax_pooling2d_nhwc_f32()
241 argmax_pooling_op->indirection_buffer = indirection_buffer; in xnn_setup_argmax_pooling2d_nhwc_f32()
260 .indirect_input = argmax_pooling_op->indirection_buffer, in xnn_setup_argmax_pooling2d_nhwc_f32()
Dmax-pooling-nhwc.c275 const void** indirection_buffer = 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()
277 if (indirection_buffer == NULL) { in setup_max_pooling2d_nhwc()
281 max_pooling_op->indirection_buffer = indirection_buffer; in setup_max_pooling2d_nhwc()
299 .indirect_input = max_pooling_op->indirection_buffer, in setup_max_pooling2d_nhwc()
Ddeconvolution-nhwc.c504 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(deconvolution_op->indirecti… in setup_conv_path() local
505 if (indirection_buffer == NULL) { in setup_conv_path()
511 deconvolution_op->indirection_buffer = indirection_buffer; in setup_conv_path()
529 .indirect_a = deconvolution_op->indirection_buffer, in setup_conv_path()
659 const void** indirection_buffer = in setup_subconv2d_path() local
660 …(const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size… in setup_subconv2d_path()
661 if (indirection_buffer == NULL) { in setup_subconv2d_path()
667 deconvolution_op->indirection_buffer = indirection_buffer; in setup_subconv2d_path()
Dconvolution-nhwc.c954 …const void** indirection_buffer = (const void**) xnn_reallocate_memory((void*) convolution_op->ind… in setup_convolution2d_nhwc() local
955 if (indirection_buffer == NULL) { in setup_convolution2d_nhwc()
961 convolution_op->indirection_buffer = indirection_buffer; in setup_convolution2d_nhwc()
977 .indirect_a = convolution_op->indirection_buffer, in setup_convolution2d_nhwc()
1098 const void** indirection_buffer = in setup_convolution2d_nhwc() local
1099 … (const void**) xnn_reallocate_memory(convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1100 if (indirection_buffer == NULL) { in setup_convolution2d_nhwc()
1105 convolution_op->indirection_buffer = indirection_buffer; in setup_convolution2d_nhwc()
1116 .indirect_input = convolution_op->indirection_buffer, in setup_convolution2d_nhwc()
Daverage-pooling-nhwc.c523 const void** indirection_buffer = in setup_average_pooling2d() local
524 …(const void**) xnn_reallocate_memory(average_pooling_op->indirection_buffer, indirection_buffer_si… in setup_average_pooling2d()
525 if (indirection_buffer == NULL) { in setup_average_pooling2d()
530 average_pooling_op->indirection_buffer = indirection_buffer; in setup_average_pooling2d()
578 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
607 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
/external/XNNPACK/src/xnnpack/
Dindirection.h56 const void** indirection_buffer,
68 const void** indirection_buffer,
Doperator.h182 const void** indirection_buffer; member
219 const void** indirection_buffer; member
/external/XNNPACK/bench/
Df16-igemm.cc98 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in IGEMMBenchmark()
Df16-dwconv.cc101 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in DWConvBenchmark()
Df32-dwconv.cc93 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in DWConvBenchmark()
Df32-igemm.cc92 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in IGEMMBenchmark()