/aosp_15_r20/external/XNNPACK/src/ |
H A D | indirection.c | 26 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 D | operator-delete.c | 29 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 D | indirection.c | 21 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 D | conv-run.cc | 211 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 D | convolution.c | 772 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 D | max-pooling.c | 210 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 D | average-pooling.c | 272 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 D | deconvolution.c | 286 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 D | operator-run.c | 409 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 D | operator-delete.c | 20 free(op->indirection_buffer); in pytorch_qnnp_delete_operator()
|
H A D | deconv-run.cc | 187 .indirect_a = (const uint8_t**)deconvolution->indirection_buffer, in qnnpackDeConv()
|
/aosp_15_r20/external/XNNPACK/src/xnnpack/ |
H A D | indirection.h | 57 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 D | operator.h | 111 const void** indirection_buffer; member 146 const void** indirection_buffer; member
|
/aosp_15_r20/external/XNNPACK/src/operators/ |
H A D | resize-bilinear-nchw.c | 154 …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 D | unpooling-nhwc.c | 188 …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 D | resize-bilinear-nhwc.c | 237 …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 D | argmax-pooling-nhwc.c | 233 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 D | deconvolution-nhwc.c | 701 …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 D | max-pooling-nhwc.c | 288 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 D | average-pooling-nhwc.c | 734 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 D | convolution-nhwc.c | 1568 …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 D | operator.h | 85 const void** indirection_buffer; member
|
/aosp_15_r20/external/XNNPACK/bench/ |
H A D | f16-dwconv.cc | 104 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in f16_dwconv()
|
H A D | f16-igemm.cc | 99 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in f16_igemm()
|
H A D | f32-igemm.cc | 96 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()
|