Searched refs:input_tuples (Results 1 – 8 of 8) 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
166 RebuildOutputAliases(output_tuple_ptr, input_tuples, input_output_alias)); in AllocateOutputTuple()
176 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in AllocateOutputTensors() argument
181 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()
300 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in RunExecutable() argument
310 input_tuples, input_is_dynamic, release_inputs)); in RunExecutable()
372 output_tuple_ptr, input_tuples, in RunExecutable()
382 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in ExecuteComputation() argument
386 return RunExecutable(context, device_ref, executable, input_tuples, in ExecuteComputation()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 268 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetInputTupleAllocations() local
269 input_tuples.reserve(input_coords.size()); in GetInputTupleAllocations()
289 input_tuples.emplace_back(std::move(tuple)); in GetInputTupleAllocations()
295 input_tuples.emplace_back(sub_tuple); in GetInputTupleAllocations()
298 return std::move(input_tuples); in GetInputTupleAllocations()
303 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in RebuildOutputAliases() argument
308 TF_RET_CHECK(alias.parameter_number < input_tuples.size()); in RebuildOutputAliases()
309 return output_tuple->AliasBufferFrom(*input_tuples[alias.parameter_number], in RebuildOutputAliases()
317 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in GetArgumentsBuffers() argument
329 bool alias_outputs = release_inputs && input_tuples.size() == 1 && in GetArgumentsBuffers()
[all …]
|
| D | xrt_util.h | 86 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples,
91 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples,
|
| /external/cronet/stable/build/ |
| D | zip_helpers.py | 134 input_tuples = []
143 input_tuples.append(tup)
146 input_tuples.sort(key=lambda tup: tup[0])
153 for zip_path, fs_path in input_tuples:
|
| /external/cronet/tot/build/ |
| D | zip_helpers.py | 134 input_tuples = []
143 input_tuples.append(tup)
146 input_tuples.sort(key=lambda tup: tup[0])
153 for zip_path, fs_path in input_tuples:
|
| /external/angle/build/ |
| D | zip_helpers.py | 134 input_tuples = []
143 input_tuples.append(tup)
146 input_tuples.sort(key=lambda tup: tup[0])
153 for zip_path, fs_path in input_tuples:
|
| /external/libchrome/build/android/gyp/util/ |
| D | build_utils.py | 336 input_tuples = []
340 input_tuples.append(tup)
343 input_tuples.sort(key=lambda tup: tup[0])
345 for zip_path, fs_path in input_tuples:
|