Searched refs:multipass_buffer (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| D | operator-run.c | 618 int32_t* multipass_buffer = in compute_average_pooling_multipass()
621 int32_t multipass_buffer[context->packed_channels]; in compute_average_pooling_multipass()
630 multipass_buffer, in compute_average_pooling_multipass()
637 _freea(multipass_buffer); in compute_average_pooling_multipass()
685 int32_t* multipass_buffer = in compute_global_average_pooling_multipass()
688 int32_t multipass_buffer[context->packed_channels]; in compute_global_average_pooling_multipass()
697 multipass_buffer, in compute_global_average_pooling_multipass()
702 _freea(multipass_buffer); in compute_global_average_pooling_multipass()
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 783 void* multipass_buffer = in xnn_compute_average_pooling_multipass() local
788 indirect_input, input_offset, context->zero, multipass_buffer, output, in xnn_compute_average_pooling_multipass()
826 …void* multipass_buffer = XNN_SIMD_ALLOCA(context->channels * sizeof(int32_t) + XNN_EXTRA_BYTES * s… in xnn_compute_pixelwise_average_pooling_multipass() local
830 indirect_input, input_offset, context->zero, pixelwise_buffer, multipass_buffer, output, in xnn_compute_pixelwise_average_pooling_multipass()
863 void* multipass_buffer = in xnn_compute_global_average_pooling_nwc_multipass() local
872 multipass_buffer, in xnn_compute_global_average_pooling_nwc_multipass()
|