| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 208 std::vector<InputCoords> input_coords; in GetComputationInputs() local
212 input_coords.emplace_back(arg.scalar<int64_t>()()); in GetComputationInputs()
218 input_coords.emplace_back(arg_vec(i)); in GetComputationInputs()
222 return std::move(input_coords); in GetComputationInputs()
258 const std::vector<InputCoords>& input_coords, in GetInputTupleAllocations() argument
263 if (input_coords.size() != num_input_shapes) { in GetInputTupleAllocations()
266 input_coords.size(), " vs. ", num_input_shapes); in GetInputTupleAllocations()
269 input_tuples.reserve(input_coords.size()); in GetInputTupleAllocations()
270 for (size_t i = 0; i < input_coords.size(); ++i) { in GetInputTupleAllocations()
272 working_set->LookupAndPin(backend, input_coords[i].handle, allocator)); in GetInputTupleAllocations()
[all …]
|
| D | xrt_util.h | 78 const std::vector<InputCoords>& input_coords,
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NESpaceToDepthLayerKernel.cpp | 129 Coordinates input_coords{ in_x, in_y, z, batch_id }; in run() local
130 memcpy(out.ptr(), _input->ptr_to_element(input_coords), element_size); in run()
148 Coordinates input_coords{ z, in_x, in_y, batch_id }; in run() local
149 memcpy(out.ptr(), _input->ptr_to_element(input_coords), element_size); in run()
|
| D | NESpaceToBatchLayerKernel.cpp | 193 Coordinates input_coords{ in_x, in_y, z, w }; in run() local
194 memcpy(out.ptr(), _input->ptr_to_element(input_coords), element_size); in run()
219 Coordinates input_coords{ z, in_x, in_y, w }; in run() local
220 memcpy(out.ptr(), _input->ptr_to_element(input_coords), element_size); in run()
|
| D | NETileKernel.cpp | 113 …Coordinates input_coords{ x % src_shape[0], y % src_shape[1], z % src_shape[2], w % src_shape[3] … in run() local
114 …memcpy(output_it.ptr(), _input->ptr_to_element(input_coords), _input->info()->dimension(0) * _inpu… in run()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Unstack.cpp | 78 … const Coordinates input_coords = expand_coordinates(id, axis, slice, unpacked_num_dimensions); in get_slice() local
79 … *reinterpret_cast<T *>(output(id)) = *reinterpret_cast<const T *>(input_tensor(input_coords)); in get_slice()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 90 xla::Backend* backend, const std::vector<InputCoords>& input_coords, in GetInputTuples() argument
96 input_coords, working_set, backend, computation_layout.parameter_count(), in GetInputTuples()
402 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation() argument
409 input_coords, release_inputs, device_ref->allocator())); in ExecuteComputation()
479 TF_ASSIGN_OR_RETURN(std::vector<InputCoords> input_coords, in DoWork()
494 input_coords, release_inputs, stream, rng_seed, in DoWork()
|
| D | tpu_execute_op.cc | 339 TF_ASSIGN_OR_RETURN(std::vector<InputCoords> input_coords, in DoWork()
347 input_coords, &working_set, backend, executable.input_shapes_size(), in DoWork()
|