Home
last modified time | relevance | path

Searched refs:output_locations (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc183 kernel_state.output_locations.push_back( in Initialize()
198 kernel_state.output_locations.resize(kernel_state.num_outputs); in Initialize()
201 kernel_state.output_locations[e->src_output()].push_back( in Initialize()
230 for (size_t j = 0; j < kernels_[i].output_locations.size(); ++j) { in Initialize()
231 for (size_t output_location : kernels_[i].output_locations[j]) { in Initialize()
371 for (size_t i = 0; i < kernel_state.output_locations.size(); ++i) { in Run()
372 Entry& input = inputs[kernel_state.output_locations[i]]; in Run()
425 const size_t num_destinations = kernel_state.output_locations[j].size(); in Run()
432 Entry& input = inputs[kernel_state.output_locations[j][k]]; in Run()
438 inputs[kernel_state.output_locations[j][num_destinations - 1]]; in Run()
[all …]
/external/mesa3d/src/mesa/state_tracker/
Dst_nir_builtins.c124 unsigned *output_locations, in st_nir_make_passthrough_shader() argument
151 snprintf(var_name, sizeof(var_name), "out_%u", output_locations[i]); in st_nir_make_passthrough_shader()
154 out->data.location = output_locations[i]; in st_nir_make_passthrough_shader()
Dst_nir.h74 unsigned *output_locations,