Lines Matching refs:indirect_input

152 …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(…
235         indirect_input[i] = input.data() + i * channels() - input_offset();
237       std::shuffle(indirect_input.begin(),
238         indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
243           float max_value = indirect_input[x * step()][c + input_offset()];
246             const float value = indirect_input[x * step() + p][c + input_offset()];
259         indirect_input.data(), input_offset() * sizeof(float),
273               indirect_input[x * step() + index_ref[x * channels() + c]][c + input_offset()],
274               indirect_input[x * step() + index[x * channels() + c]][c + input_offset()])