Searched refs:input_tuples (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 95 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetChainedOpInputs() local
96 input_tuples.reserve(op.inputs_size()); in GetChainedOpInputs()
104 input_tuples.push_back(tuple); in GetChainedOpInputs()
110 input_tuples.emplace_back(sub_tuple); in GetChainedOpInputs()
113 input_tuples.back()->on_host_shape())) { in GetChainedOpInputs()
121 return std::move(input_tuples); in GetChainedOpInputs()
139 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in AllocateOutputTuple() argument
165 RebuildOutputAliases(output_tuple_ptr, input_tuples, input_output_alias)); in AllocateOutputTuple()
175 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in AllocateOutputTensors() argument
180 AllocateOutputTuple(node_context, stream, input_tuples, in AllocateOutputTensors()
[all …]
|
| D | xrt_execute_op.cc | 104 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetChainedOpInputTuples() local
105 input_tuples.reserve(op.inputs_size()); in GetChainedOpInputTuples()
112 input_tuples.emplace_back(op_inputs[i]); in GetChainedOpInputTuples()
118 input_tuples.emplace_back(sub_tuple); in GetChainedOpInputTuples()
121 return input_tuples; in GetChainedOpInputTuples()
299 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in RunExecutable() argument
309 input_tuples, input_is_dynamic, release_inputs)); in RunExecutable()
368 output_tuple_ptr, input_tuples, in RunExecutable()
378 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in ExecuteComputation() argument
382 return RunExecutable(context, device_ref, executable, input_tuples, in ExecuteComputation()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 265 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetInputTupleAllocations() local
266 input_tuples.reserve(input_coords.size()); in GetInputTupleAllocations()
286 input_tuples.emplace_back(std::move(tuple)); in GetInputTupleAllocations()
292 input_tuples.emplace_back(sub_tuple); in GetInputTupleAllocations()
295 return std::move(input_tuples); in GetInputTupleAllocations()
300 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in RebuildOutputAliases() argument
305 TF_RET_CHECK(alias.parameter_number < input_tuples.size()); in RebuildOutputAliases()
306 return output_tuple->AliasBufferFrom(*input_tuples[alias.parameter_number], in RebuildOutputAliases()
314 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in GetArgumentsBuffers() argument
326 bool alias_outputs = release_inputs && input_tuples.size() == 1 && in GetArgumentsBuffers()
[all …]
|
| D | xrt_util.h | 85 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples,
90 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples,
|
| /external/libchrome/build/android/gyp/util/ |
| D | build_utils.py | 343 input_tuples = []
347 input_tuples.append(tup)
350 input_tuples.sort(key=lambda tup: tup[0])
352 for zip_path, fs_path in input_tuples:
|