Searched refs:tuple_data (Results 1 – 7 of 7) 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()
229 auto tuple_data = in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 146 auto tuple_data = Tuple(&xla_builder_, {data0, data1}); in TEST_F() local
147 auto gte = GetTupleElement(tuple_data, 1); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 249 StatusOr<XlaOp> GetTupleElementInternal(const Shape& shape, XlaOp tuple_data,
|
| D | mlir_hlo_builder.cc | 642 XlaOp tuple_data, in GetTupleElementInternal() argument
648 loc_, result_type, GetValue(tuple_data), in GetTupleElementInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1329 XlaOp XlaBuilder::GetTupleElement(XlaOp tuple_data, int64_t index) { in GetTupleElement() argument
1331 TF_ASSIGN_OR_RETURN(const Shape* tuple_shape, GetShapePtr(tuple_data)); in GetTupleElement()
1343 ShapeUtil::GetTupleElementShape(*tuple_shape, index), tuple_data, in GetTupleElement()
1349 XlaOp tuple_data, in GetTupleElementInternal() argument
1355 {tuple_data}); in GetTupleElementInternal()
4146 XlaOp GetTupleElement(const XlaOp tuple_data, int64_t index) { in GetTupleElement() argument
4147 return tuple_data.builder()->GetTupleElement(tuple_data, index); in GetTupleElement()
|
| D | xla_builder.h | 520 XlaOp GetTupleElement(XlaOp tuple_data, int64_t index);
522 XlaOp tuple_data,
1153 friend XlaOp GetTupleElement(XlaOp tuple_data, int64_t index);
1903 XlaOp GetTupleElement(XlaOp tuple_data, int64_t index);
|
| /external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| D | ops.pyi | 245 def GetTupleElement(tuple_data: XlaOp, index: int) -> XlaOp: ...
|