Searched refs:ExpandedTupleInput (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 124 struct ExpandedTupleInput { struct
141 const xla::ShapeTree<ExpandedTupleInput>& elements, argument
268 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal,
|
| D | xrt_state.cc | 414 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal, in ExpandTreeOfTuples()
425 [&](const xla::ShapeIndex& index, const ExpandedTupleInput& element) { in ExpandTreeOfTuples()
456 const xla::ShapeTree<ExpandedTupleInput>& elements, in MakeTuple()
483 [&](const xla::ShapeIndex& index, const ExpandedTupleInput& element) { in MakeTuple()
517 const ExpandedTupleInput& element) { in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 75 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleNode()
97 XRTTupleAllocation::ExpandedTupleInput& input = in ParseTupleNode()
127 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleTree()
128 xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>* tuple_shape_tree, in ParseTupleTree()
139 *tuple_shape_tree = xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>( in ParseTupleTree()
143 XRTTupleAllocation::ExpandedTupleInput* element) { in ParseTupleTree()
442 std::vector<XRTTupleAllocation::ExpandedTupleInput> input_vector( in Compute()
447 xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput> tuple_shape_tree; in Compute()
|