Searched refs:tuple0 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_domain_test.cc | 617 auto tuple0 = FindInstruction(module.get(), "tuple.0"); in TEST_F() local
618 tuple0->clear_sharding(); in TEST_F()
635 auto tuple0_users = tuple0->users(); in TEST_F()
636 auto computation = tuple0->parent(); in TEST_F()
638 HloInstruction::CreateUnary(tuple0->operand(1)->shape(), HloOpcode::kCopy, in TEST_F()
639 tuple0->mutable_operand(1))); in TEST_F()
640 TF_EXPECT_OK(tuple0->ReplaceOperandWith(1, copy0)); in TEST_F()
643 HloInstruction::CreateUnary(tuple0->shape(), HloOpcode::kCopy, tuple0)); in TEST_F()
649 ShapeUtil::GetTupleElementShape(tuple0->shape(), 1), tuple0, 1)); in TEST_F()
656 TF_EXPECT_OK(tuple0->ReplaceUseWith(user, tuple1)); in TEST_F()
[all …]
|
| D | bfloat16_propagation_test.cc | 282 HloInstruction* tuple0 = in TEST_F() local
285 builder.AddInstruction(HloInstruction::CreateTuple({tuple0, xpose})); in TEST_F()
293 tuple0->shape(), tuple1, 0)), in TEST_F()
530 HloInstruction* tuple0 = in TEST_F() local
535 tuple0->shape(), HloOpcode::kTupleSelect, pred, tuple0, tuple1)); in TEST_F()
787 HloInstruction* tuple0 = in TEST_F() local
795 HloInstruction::CreateParameter(0, tuple0->shape(), "cond0_param")); in TEST_F()
853 HloInstruction::CreateParameter(0, tuple0->shape(), "body_param")); in TEST_F()
865 HloInstruction::CreateWhile(tuple0->shape(), cond0, body, tuple0)); in TEST_F()
|
| D | tuple_simplifier_test.cc | 272 HloInstruction* tuple0 = in TEST_F() local
275 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple0, 0)); in TEST_F()
277 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple0, 1)); in TEST_F()
|
| D | buffer_assignment_test.cc | 1938 auto tuple0 = builder.AddInstruction( in TEST_F() local
1941 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2257 auto tuple0 = builder.AddInstruction( in TEST_F() local
2260 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2402 auto tuple0 = builder.AddInstruction( in TEST_F() local
2408 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F()
2477 auto tuple0 = builder.AddInstruction( in TEST_F() local
2480 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
|
| D | layout_assignment_test.cc | 258 auto tuple0 = builder.AddInstruction( in TEST_F() local
267 tuple0->shape(), HloOpcode::kTupleSelect, pred, tuple0, tuple1)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util_test.cc | 52 Shape tuple0 = ShapeUtil::MakeTupleShape({}); in TEST_F() local
56 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple0)); in TEST_F()
57 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple1)); in TEST_F()
58 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple2)); in TEST_F()
59 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple0)); in TEST_F()
60 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple0)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 490 tuple0, array1 = pieces
494 got = tuple0.to_py()
|