Searched refs:tuple_data (Results 1 – 6 of 6) 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/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 136 auto tuple_data = Tuple(&xla_builder_, {data0, data1}); in TEST_F() local
137 auto gte = GetTupleElement(tuple_data, 1); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 238 StatusOr<XlaOp> GetTupleElementInternal(const Shape& shape, XlaOp tuple_data,
|
| D | mlir_hlo_builder.cc | 498 XlaOp tuple_data, in GetTupleElementInternal() argument
504 loc_, result_type, GetValue(tuple_data), in GetTupleElementInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1256 XlaOp XlaBuilder::GetTupleElement(XlaOp tuple_data, int64 index) { in GetTupleElement() argument
1258 TF_ASSIGN_OR_RETURN(const Shape* tuple_shape, GetShapePtr(tuple_data)); in GetTupleElement()
1270 ShapeUtil::GetTupleElementShape(*tuple_shape, index), tuple_data, in GetTupleElement()
1276 XlaOp tuple_data, in GetTupleElementInternal() argument
1282 {tuple_data}); in GetTupleElementInternal()
4194 XlaOp GetTupleElement(const XlaOp tuple_data, int64 index) { in GetTupleElement() argument
4195 return tuple_data.builder()->GetTupleElement(tuple_data, index); in GetTupleElement()
|
| D | xla_builder.h | 519 XlaOp GetTupleElement(XlaOp tuple_data, int64 index);
521 XlaOp tuple_data,
1111 friend XlaOp GetTupleElement(XlaOp tuple_data, int64 index);
1808 XlaOp GetTupleElement(XlaOp tuple_data, int64 index);
|