Searched refs:tuple_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | tuple_test.cc | 126 auto tuple_data = in XLA_TEST_F() local
129 GetTupleElement(tuple_data, 1); in XLA_TEST_F()
137 auto tuple_data = in XLA_TEST_F() local
141 GetTupleElement(tuple_data, 1); in XLA_TEST_F()
165 auto tuple_data = in XLA_TEST_F() local
168 auto vector_element = GetTupleElement(tuple_data, 0); in XLA_TEST_F()
169 auto matrix_element = GetTupleElement(tuple_data, 1); in XLA_TEST_F()
194 auto tuple_data = in XLA_TEST_F() local
198 {GetTupleElement(tuple_data, 1), GetTupleElement(tuple_data, 0)}); in XLA_TEST_F()
253 auto tuple_data = in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 571 LocalOp ComputationBuilder::GetTupleElement(const LocalOp& tuple_data, in GetTupleElement() argument
573 return xla::GetTupleElement(tuple_data.op(), index); in GetTupleElement()
|
| D | local_computation_builder.h | 286 LocalOp GetTupleElement(const LocalOp& tuple_data, int64 index);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 991 XlaOp XlaBuilder::GetTupleElement(const XlaOp& tuple_data, int64 index) { in GetTupleElement() argument
994 TF_ASSIGN_OR_RETURN(const Shape& tuple_shape, GetShape(tuple_data)); in GetTupleElement()
1006 {tuple_data}); in GetTupleElement()
2917 XlaOp GetTupleElement(const XlaOp& tuple_data, int64 index) { in GetTupleElement() argument
2918 return tuple_data.builder()->GetTupleElement(tuple_data, index); in GetTupleElement()
|
| D | xla_builder.h | 374 XlaOp GetTupleElement(const XlaOp& tuple_data, int64 index);
760 friend XlaOp GetTupleElement(const XlaOp& tuple_data, int64 index);
1262 XlaOp GetTupleElement(const XlaOp& tuple_data, int64 index);
|