Searched refs:ExpandedTupleInput (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 113 struct ExpandedTupleInput { struct
129 const xla::ShapeTree<ExpandedTupleInput>& elements, argument
212 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal,
|
| D | xrt_state.cc | 340 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal, in ExpandTreeOfTuples()
351 [&](const xla::ShapeIndex& index, const ExpandedTupleInput& element) { in ExpandTreeOfTuples()
382 const xla::ShapeTree<ExpandedTupleInput>& elements, in MakeTuple()
409 [&](const xla::ShapeIndex& index, const ExpandedTupleInput& element) { in MakeTuple()
443 const ExpandedTupleInput& element) { in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 71 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleNode()
93 XRTTupleAllocation::ExpandedTupleInput& input = in ParseTupleNode()
123 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleTree()
124 xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>* tuple_shape_tree, in ParseTupleTree()
135 *tuple_shape_tree = xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>( in ParseTupleTree()
139 XRTTupleAllocation::ExpandedTupleInput* element) { in ParseTupleTree()
399 std::vector<XRTTupleAllocation::ExpandedTupleInput> input_vector( in Compute()
412 xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput> tuple_shape_tree; in Compute()
|