| /external/XNNPACK/test/ |
| D | avgpool-microkernel-tester.h | 214 …std::vector<const uint16_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elemen… in Test()
216 input_offset() + indirect_input.size() * channels()); in Test()
227 indirect_input[i] = input.data() + i * channels(); in Test()
229 std::shuffle(indirect_input.begin(), in Test()
230 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng); in Test()
232 indirect_input[zero_index()] = zero.data(); in Test()
240 const uint16_t* row = indirect_input[x * step() + p]; in Test()
271 …reinterpret_cast<const void**>(indirect_input.data()), input_offset() * sizeof(uint16_t), zero.dat… in Test()
305 …std::vector<const uint16_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elemen… in Test()
307 input_offset() + indirect_input.size() * channels()); in Test()
[all …]
|
| D | argmaxpool-microkernel-tester.h | 151 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
163 indirect_input[i] = input.data() + i * channels() - input_offset();
165 std::shuffle(indirect_input.begin(),
166 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
171 float max_value = indirect_input[x * step()][c + input_offset()];
174 const float value = indirect_input[x * step() + p][c + input_offset()];
187 indirect_input.data(), input_offset() * sizeof(float), output.data(), index.data(),
199 indirect_input[x * step() + index_ref[x * channels() + c]][c + input_offset()],
200 indirect_input[x * step() + index[x * channels() + c]][c + input_offset()])
218 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
[all …]
|
| D | maxpool-microkernel-tester.h | 167 …std::vector<const int8_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements… in Test()
169 indirect_input.size() * channels()); in Test()
180 indirect_input[i] = input.data() + i * channels() - input_offset(); in Test()
182 std::shuffle(indirect_input.begin(), in Test()
183 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng); in Test()
194 max_value = std::max(max_value, indirect_input[x * step() + p][c + input_offset()]); in Test()
204 indirect_input.data(), input_offset() * sizeof(int8_t), output.data(), in Test()
239 …std::vector<const uint8_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_element… in Test()
241 indirect_input.size() * channels()); in Test()
252 indirect_input[i] = input.data() + i * channels() - input_offset(); in Test()
[all …]
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 645 const void** indirect_input = in xnn_compute_dwconv_unipass() local
646 …(const void**) ((uintptr_t) context->indirect_input + output_y * context->indirect_input_height_st… in xnn_compute_dwconv_unipass()
653 indirect_input, context->packed_weights, output, in xnn_compute_dwconv_unipass()
680 const void** indirect_input = (const void**) ((uintptr_t) context->indirect_input + in xnn_compute_argmax_pooling_unipass() local
690 indirect_input, input_offset, output, index, in xnn_compute_argmax_pooling_unipass()
699 const void** indirect_input = (const void**) ((uintptr_t) context->indirect_input + in xnn_compute_argmax_pooling_multipass() local
712 …indirect_input, input_offset, multipass_accumulation_buffer, multipass_index_buffer, output, index, in xnn_compute_argmax_pooling_multipass()
721 const void** indirect_input = (const void**) ((uintptr_t) context->indirect_input + in xnn_compute_max_pooling() local
729 indirect_input, input_offset, output, in xnn_compute_max_pooling()
759 const void** indirect_input = in xnn_compute_average_pooling_unipass() local
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 489 const void** indirect_input; member
549 const void** indirect_input; member
599 const void** indirect_input; member
633 const void** indirect_input; member
670 const void** indirect_input; member
765 const void** indirect_input; member
792 const void** indirect_input; member
|
| /external/ComputeLibrary/src/cpu/kernels/assembly/ |
| D | arm_gemm.hpp | 161 unsigned int nmulti, bool indirect_input, Activation act, const int maxthreads, in GemmArgs()
163 …ns(Ksections), _nbatches(nbatches), _nmulti(nmulti), _indirect_input(indirect_input), _act(act), _… in GemmArgs()
|
| /external/XNNPACK/src/operators/ |
| D | resize-bilinear-nchw.c | 202 .indirect_input = resize_op->indirection_buffer, in xnn_setup_resize_bilinear2d_nchw_f32()
|
| D | argmax-pooling-nhwc.c | 260 .indirect_input = argmax_pooling_op->indirection_buffer, in xnn_setup_argmax_pooling2d_nhwc_f32()
|
| D | resize-bilinear-nhwc.c | 284 .indirect_input = resize_op->indirection_buffer, in setup_resize_bilinear2d_nhwc()
|
| D | average-pooling-nhwc.c | 782 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
811 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
|
| D | max-pooling-nhwc.c | 312 .indirect_input = max_pooling_op->indirection_buffer, in setup_max_pooling2d_nhwc()
|
| D | convolution-nhwc.c | 1738 .indirect_input = convolution_op->indirection_buffer, in setup_convolution2d_nhwc()
|