Searched refs:sub_tuple (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 288 XRTTupleAllocation* sub_tuple; in GetInputTupleAllocations() local
290 tuple.get(), input_coords[i].index, &sub_tuple, in GetInputTupleAllocations()
292 input_tuples.emplace_back(sub_tuple); in GetInputTupleAllocations()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 106 XRTTupleAllocation* sub_tuple; in GetChainedOpInputs() local
108 tuple.get(), {input.output_index() - 1}, &sub_tuple, in GetChainedOpInputs()
110 input_tuples.emplace_back(sub_tuple); in GetChainedOpInputs()
|
| D | xrt_execute_op.cc | 114 XRTTupleAllocation* sub_tuple; in GetChainedOpInputTuples() local
116 op_inputs[i].get(), {input.output_index() - 1}, &sub_tuple, in GetChainedOpInputTuples()
118 input_tuples.emplace_back(sub_tuple); in GetChainedOpInputTuples()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 496 Value sub_tuple; in ReplaceWithTupleResult() local
503 if (!sub_tuple) in ReplaceWithTupleResult()
504 sub_tuple = CreateSubTuple(builder, replacement, tuple_type.size()); in ReplaceWithTupleResult()
506 use.set(sub_tuple); in ReplaceWithTupleResult()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1564 auto sub_tuple = in TEST_F() local
1580 GetAllocation(*assignment, sub_tuple, /*index=*/{})); in TEST_F()
1587 GetTopLevelAllocation(*assignment, sub_tuple)); in TEST_F()
|