Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c26 const void** indirection_buffer = op->indirection_buffer; in xnn_indirection_init_conv2d() local
63 indirection_buffer[index] = (const void*) in xnn_indirection_init_conv2d()
66 indirection_buffer[index] = zero; in xnn_indirection_init_conv2d()
73 indirection_buffer[index] = zero; in xnn_indirection_init_conv2d()
86 const void** indirection_buffer = op->indirection_buffer; in xnn_indirection_init_deconv2d() local
126 …indirection_buffer[index] = (const void*) ((uintptr_t) input + (input_y * input_width + input_x) *… in xnn_indirection_init_deconv2d()
128 indirection_buffer[index] = zero; in xnn_indirection_init_deconv2d()
141 const void** indirection_buffer = op->indirection_buffer; in xnn_indirection_init_subconv2d() local
165 subconvolution_params->indirection_buffer = indirection_buffer; in xnn_indirection_init_subconv2d()
187 *indirection_buffer++ = in xnn_indirection_init_subconv2d()
[all …]
H A Doperator-delete.c29 xnn_release_memory(op->indirection_buffer); in xnn_delete_operator()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c21 const void** indirection_buffer = op->indirection_buffer; in pytorch_qnnp_indirection_init_conv3d() local
91 indirection_buffer[index] = (char*)input + in pytorch_qnnp_indirection_init_conv3d()
99 indirection_buffer[index] = zero; in pytorch_qnnp_indirection_init_conv3d()
112 indirection_buffer[index] = zero; in pytorch_qnnp_indirection_init_conv3d()
126 indirection_buffer[index] = zero; in pytorch_qnnp_indirection_init_conv3d()
227 const void** indirection_buffer = op->indirection_buffer; in pytorch_qnnp_indirection_init_dwconv() local
288 indirection_buffer[index] = (char*)input + in pytorch_qnnp_indirection_init_dwconv()
295 indirection_buffer[index] = zero; in pytorch_qnnp_indirection_init_dwconv()
310 indirection_buffer[index] = zero; in pytorch_qnnp_indirection_init_dwconv()
328 indirection_buffer[index] = zero; in pytorch_qnnp_indirection_init_dwconv()
[all …]
H A Dconv-run.cc211 const uint8_t** indirection_buffer; member
228 const uint8_t** indirection_buffer; member
251 context->indirection_buffer + in compute_dwconv2d_unipass()
276 context->indirection_buffer + in compute_dwconv2d_multiipass()
309 context->indirection_buffer + in compute_dwconv3d_multiipass()
424 .indirection_buffer = in qnnpackConv()
425 (const uint8_t**)convolution->indirection_buffer, in qnnpackConv()
457 .indirection_buffer = in qnnpackConv()
458 (const uint8_t**)convolution->indirection_buffer, in qnnpackConv()
490 .indirection_buffer = in qnnpackConv()
[all …]
H A Dconvolution.c772 const void** indirection_buffer = (const void**)realloc( in pytorch_qnnp_setup_convolution_ndhwc_q8() local
773 convolution->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
774 if (indirection_buffer == NULL) { in pytorch_qnnp_setup_convolution_ndhwc_q8()
780 convolution->indirection_buffer = indirection_buffer; in pytorch_qnnp_setup_convolution_ndhwc_q8()
791 const void** indirection_buffer = (const void**)realloc( in pytorch_qnnp_setup_convolution_ndhwc_q8() local
792 convolution->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_convolution_ndhwc_q8()
793 if (indirection_buffer == NULL) { in pytorch_qnnp_setup_convolution_ndhwc_q8()
799 convolution->indirection_buffer = indirection_buffer; in pytorch_qnnp_setup_convolution_ndhwc_q8()
H A Dmax-pooling.c210 const void** indirection_buffer = (const void**)realloc( 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()
212 if (indirection_buffer == NULL) { in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
218 max_pooling->indirection_buffer = indirection_buffer; in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
H A Daverage-pooling.c272 const void** indirection_buffer = (const void**)realloc( 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()
274 if (indirection_buffer == NULL) { in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
280 average_pooling->indirection_buffer = indirection_buffer; in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
H A Ddeconvolution.c286 const void** indirection_buffer = (const void**)realloc( in pytorch_qnnp_setup_deconvolution2d_nhwc_q8() local
287 deconvolution->indirection_buffer, indirection_buffer_size); in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
288 if (indirection_buffer == NULL) { in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
294 deconvolution->indirection_buffer = indirection_buffer; in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
H A Doperator-run.c409 const uint8_t** indirection_buffer; member
428 const uint8_t** indirection_buffer; member
451 context->indirection_buffer + in compute_dwconv2d_unipass()
477 context->indirection_buffer + in compute_dwconv2d_multiipass()
510 context->indirection_buffer + in compute_dwconv3d_multiipass()
917 .indirection_buffer = (const uint8_t**)op->indirection_buffer, in pytorch_qnnp_run_operator()
945 .indirection_buffer = (const uint8_t**)op->indirection_buffer, in pytorch_qnnp_run_operator()
973 .indirection_buffer = (const uint8_t**)op->indirection_buffer, in pytorch_qnnp_run_operator()
1307 .indirect_a = (const uint8_t**)op->indirection_buffer, in pytorch_qnnp_run_operator()
1350 .indirect_input = op->indirection_buffer, in pytorch_qnnp_run_operator()
[all …]
H A Doperator-delete.c20 free(op->indirection_buffer); in pytorch_qnnp_delete_operator()
H A Ddeconv-run.cc187 .indirect_a = (const uint8_t**)deconvolution->indirection_buffer, in qnnpackDeConv()
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dindirection.h57 const void** indirection_buffer,
69 const void** indirection_buffer,
81 const void** indirection_buffer,
93 const void** indirection_buffer,
105 const void** indirection_buffer,
117 const void** indirection_buffer,
H A Doperator.h111 const void** indirection_buffer; member
146 const void** indirection_buffer; member
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dresize-bilinear-nchw.c154 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(resize_op->indirection_buff… in xnn_setup_resize_bilinear2d_nchw_f32() local
155 if (indirection_buffer == NULL) { in xnn_setup_resize_bilinear2d_nchw_f32()
161 resize_op->indirection_buffer = indirection_buffer; in xnn_setup_resize_bilinear2d_nchw_f32()
185 input, resize_op->indirection_buffer, resize_op->packed_weights.pointer, in xnn_setup_resize_bilinear2d_nchw_f32()
202 .indirect_input = resize_op->indirection_buffer, in xnn_setup_resize_bilinear2d_nchw_f32()
H A Dunpooling-nhwc.c188 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(unpooling_op->indirection_b… in xnn_setup_unpooling2d_nhwc_x32() local
189 if (indirection_buffer == NULL) { in xnn_setup_unpooling2d_nhwc_x32()
195 unpooling_op->indirection_buffer = indirection_buffer; in xnn_setup_unpooling2d_nhwc_x32()
208 .indirect_output = indirection_buffer, in xnn_setup_unpooling2d_nhwc_x32()
H A Dresize-bilinear-nhwc.c237 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(resize_op->indirection_buff… in setup_resize_bilinear2d_nhwc() local
238 if (indirection_buffer == NULL) { in setup_resize_bilinear2d_nhwc()
244 resize_op->indirection_buffer = indirection_buffer; in setup_resize_bilinear2d_nhwc()
268 input, resize_op->indirection_buffer, resize_op->packed_weights.pointer, in setup_resize_bilinear2d_nhwc()
284 .indirect_input = resize_op->indirection_buffer, in setup_resize_bilinear2d_nhwc()
H A 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()
H A Ddeconvolution-nhwc.c701 …const void** indirection_buffer = (const void**) xnn_reallocate_memory(deconvolution_op->indirecti… in setup_conv_path() local
702 if (indirection_buffer == NULL) { in setup_conv_path()
708 deconvolution_op->indirection_buffer = indirection_buffer; in setup_conv_path()
727 .indirect_a = deconvolution_op->indirection_buffer, in setup_conv_path()
926 const void** indirection_buffer = in setup_subconv2d_path() local
927 …(const void**) xnn_reallocate_memory(deconvolution_op->indirection_buffer, indirection_buffer_size… in setup_subconv2d_path()
928 if (indirection_buffer == NULL) { in setup_subconv2d_path()
934 deconvolution_op->indirection_buffer = indirection_buffer; in setup_subconv2d_path()
H A Dmax-pooling-nhwc.c288 const void** indirection_buffer = 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()
290 if (indirection_buffer == NULL) { in setup_max_pooling2d_nhwc()
294 max_pooling_op->indirection_buffer = indirection_buffer; in setup_max_pooling2d_nhwc()
312 .indirect_input = max_pooling_op->indirection_buffer, in setup_max_pooling2d_nhwc()
H A Daverage-pooling-nhwc.c734 const void** indirection_buffer = in setup_average_pooling2d() local
735 …(const void**) xnn_reallocate_memory(average_pooling_op->indirection_buffer, indirection_buffer_si… in setup_average_pooling2d()
736 if (indirection_buffer == NULL) { in setup_average_pooling2d()
741 average_pooling_op->indirection_buffer = indirection_buffer; in setup_average_pooling2d()
782 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
811 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
H A Dconvolution-nhwc.c1568 …const void** indirection_buffer = (const void**) xnn_reallocate_memory((void*) convolution_op->ind… in setup_convolution2d_nhwc() local
1569 if (indirection_buffer == NULL) { in setup_convolution2d_nhwc()
1575 convolution_op->indirection_buffer = indirection_buffer; in setup_convolution2d_nhwc()
1592 .indirect_a = convolution_op->indirection_buffer, in setup_convolution2d_nhwc()
1720 const void** indirection_buffer = in setup_convolution2d_nhwc() local
1721 … (const void**) xnn_reallocate_memory(convolution_op->indirection_buffer, indirection_buffer_size); in setup_convolution2d_nhwc()
1722 if (indirection_buffer == NULL) { in setup_convolution2d_nhwc()
1727 convolution_op->indirection_buffer = indirection_buffer; in setup_convolution2d_nhwc()
1738 .indirect_input = convolution_op->indirection_buffer, in setup_convolution2d_nhwc()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Doperator.h85 const void** indirection_buffer; member
/aosp_15_r20/external/XNNPACK/bench/
H A Df16-dwconv.cc104 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in f16_dwconv()
H A Df16-igemm.cc99 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in f16_igemm()
H A Df32-igemm.cc96 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in f32_igemm()
226 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in f32_igemm()

12