Home
last modified time | relevance | path

Searched refs:input_coords (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc207 std::vector<InputCoords> input_coords; in GetComputationInputs() local
211 input_coords.emplace_back(arg.scalar<int64>()()); in GetComputationInputs()
217 input_coords.emplace_back(arg_vec(i)); in GetComputationInputs()
221 return std::move(input_coords); in GetComputationInputs()
256 const std::vector<InputCoords>& input_coords, in GetInputTupleAllocations() argument
260 if (input_coords.size() != num_input_shapes) { in GetInputTupleAllocations()
263 input_coords.size(), " vs. ", num_input_shapes); in GetInputTupleAllocations()
266 input_tuples.reserve(input_coords.size()); in GetInputTupleAllocations()
267 for (size_t i = 0; i < input_coords.size(); ++i) { in GetInputTupleAllocations()
269 working_set->LookupAndPin(backend, input_coords[i].handle)); in GetInputTupleAllocations()
[all …]
Dxrt_util.h78 const std::vector<InputCoords>& input_coords,
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc90 xla::Backend* backend, const std::vector<InputCoords>& input_coords, in GetInputTuples() argument
96 input_coords, working_set, backend, computation_layout.parameter_count(), in GetInputTuples()
398 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation() argument
405 input_coords, release_inputs)); in ExecuteComputation()
475 TF_ASSIGN_OR_RETURN(std::vector<InputCoords> input_coords, in DoWork()
490 input_coords, release_inputs, stream, rng_seed, in DoWork()
Dtpu_execute_op.cc337 TF_ASSIGN_OR_RETURN(std::vector<InputCoords> input_coords, in DoWork()
345 input_coords, &working_set, backend, executable.input_shapes_size(), in DoWork()