Home
last modified time | relevance | path

Searched refs:tuple0 (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_domain_test.cc617 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 …]
Dtuple_simplifier_test.cc272 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()
Dbfloat16_propagation_test.cc283 HloInstruction* tuple0 = in TEST_F() local
286 builder.AddInstruction(HloInstruction::CreateTuple({tuple0, xpose})); in TEST_F()
294 tuple0->shape(), tuple1, 0)), in TEST_F()
560 HloInstruction* tuple0 = in TEST_F() local
565 tuple0->shape(), HloOpcode::kTupleSelect, pred, tuple0, tuple1)); in TEST_F()
817 HloInstruction* tuple0 = in TEST_F() local
825 HloInstruction::CreateParameter(0, tuple0->shape(), "cond0_param")); in TEST_F()
883 HloInstruction::CreateParameter(0, tuple0->shape(), "body_param")); in TEST_F()
895 HloInstruction::CreateWhile(tuple0->shape(), cond0, body, tuple0)); in TEST_F()
Dbuffer_assignment_test.cc2086 auto tuple0 = builder.AddInstruction( in TEST_F() local
2089 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2404 auto tuple0 = builder.AddInstruction( in TEST_F() local
2407 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2584 auto tuple0 = builder.AddInstruction( in TEST_F() local
2590 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F()
2658 auto tuple0 = builder.AddInstruction( in TEST_F() local
2661 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
Dlayout_assignment_test.cc258 auto tuple0 = builder.AddInstruction( in TEST_F() local
267 tuple0->shape(), HloOpcode::kTupleSelect, pred, tuple0, tuple1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc44 Shape tuple0 = ShapeUtil::MakeTupleShape({}); in TEST_F() local
48 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple0)); in TEST_F()
49 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple1)); in TEST_F()
50 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple2)); in TEST_F()
51 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple0)); in TEST_F()
52 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple0)); in TEST_F()