Home
last modified time | relevance | path

Searched refs:indirect_input (Results 1 – 11 of 11) sorted by relevance

/external/XNNPACK/test/
Davgpool-microkernel-tester.h218 …std::vector<const uint8_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_element…
220 input_offset() + indirect_input.size() * channels());
235 indirect_input[i] = input.data() + i * channels();
237 std::shuffle(indirect_input.begin(),
238 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
240 indirect_input[zero_index()] = zero.data();
270 const uint8_t* row = indirect_input[x * step() + p];
285 indirect_input.data(), input_offset() * sizeof(uint8_t), zero.data(),
320 …std::vector<const uint8_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_element…
322 input_offset() + indirect_input.size() * channels());
[all …]
Dargmaxpool-microkernel-tester.h152 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
164 indirect_input[i] = input.data() + i * channels() - input_offset();
166 std::shuffle(indirect_input.begin(),
167 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
172 float max_value = indirect_input[x * step()][c + input_offset()];
175 const float value = indirect_input[x * step() + p][c + input_offset()];
188 indirect_input.data(), input_offset() * sizeof(float), output.data(), index.data(),
200 indirect_input[x * step() + index_ref[x * channels() + c]][c + input_offset()],
201 indirect_input[x * step() + index[x * channels() + c]][c + input_offset()])
219 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
[all …]
Dmaxpool-microkernel-tester.h166 …std::vector<const uint8_t*> indirect_input((output_pixels() - 1) * step() + packed_pooling_element…
168 indirect_input.size() * channels());
179 indirect_input[i] = input.data() + i * channels() - input_offset();
181 std::shuffle(indirect_input.begin(),
182 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
200 max_value = std::max(max_value, indirect_input[x * step() + p][c + input_offset()]);
210 indirect_input.data(), input_offset() * sizeof(uint8_t), output.data(),
240 …std::vector<const float*> indirect_input((output_pixels() - 1) * step() + packed_pooling_elements(…
251 indirect_input[i] = input.data() + i * channels() - input_offset();
253 std::shuffle(indirect_input.begin(),
[all …]
/external/XNNPACK/src/
Doperator-run.c373 const void** indirect_input = in xnn_compute_dwconv_unipass() local
374 …(const void**) ((uintptr_t) context->indirect_input + output_y * context->indirect_input_height_st… in xnn_compute_dwconv_unipass()
381 indirect_input, context->packed_weights, output, in xnn_compute_dwconv_unipass()
464 const void** indirect_input = (const void**) ((uintptr_t) context->indirect_input + in xnn_compute_argmax_pooling_unipass() local
474 indirect_input, input_offset, output, index, in xnn_compute_argmax_pooling_unipass()
483 const void** indirect_input = (const void**) ((uintptr_t) context->indirect_input + in xnn_compute_argmax_pooling_multipass() local
496 …indirect_input, input_offset, multipass_accumulation_buffer, multipass_index_buffer, output, index, in xnn_compute_argmax_pooling_multipass()
505 const void** indirect_input = (const void**) ((uintptr_t) context->indirect_input + in xnn_compute_max_pooling() local
513 indirect_input, input_offset, output, in xnn_compute_max_pooling()
543 const void** indirect_input = in xnn_compute_average_pooling_unipass() local
[all …]
/external/XNNPACK/src/xnnpack/
Dcompute.h383 const void** indirect_input; member
490 const void** indirect_input; member
540 const void** indirect_input; member
574 const void** indirect_input; member
610 const void** indirect_input; member
704 const void** indirect_input; member
731 const void** indirect_input; member
/external/XNNPACK/src/operators/
Dresize-bilinear-nhwc.c198 .indirect_input = resize_op->indirection_buffer, in xnn_setup_resize_bilinear2d_nhwc_f32()
Dresize-bilinear-nchw.c199 .indirect_input = resize_op->indirection_buffer, in xnn_setup_resize_bilinear2d_nchw_f32()
Dargmax-pooling-nhwc.c260 .indirect_input = argmax_pooling_op->indirection_buffer, in xnn_setup_argmax_pooling2d_nhwc_f32()
Daverage-pooling-nhwc.c578 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
607 .indirect_input = average_pooling_op->indirection_buffer, in setup_average_pooling2d()
Dmax-pooling-nhwc.c299 .indirect_input = max_pooling_op->indirection_buffer, in setup_max_pooling2d_nhwc()
Dconvolution-nhwc.c1116 .indirect_input = convolution_op->indirection_buffer, in setup_convolution2d_nhwc()