Searched refs:while0 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1940 auto while0 = builder.AddInstruction( in TEST_F() local
1948 HloInstruction::CreateGetTupleElement(data_shape_, while0, 2)); in TEST_F()
1960 assignment->GetUniqueSlice(while0, {0}).ConsumeValueOrDie()); in TEST_F()
1963 assignment->GetUniqueSlice(while0, {1}).ConsumeValueOrDie()); in TEST_F()
1965 EXPECT_EQ(assignment->GetUniqueSlice(while0, {2}).ConsumeValueOrDie(), in TEST_F()
2028 const HloInstruction* while0 = while1->operand(0)->operand(0); in TEST_F() local
2029 ASSERT_EQ(while0->opcode(), HloOpcode::kWhile); in TEST_F()
2036 assignment->GetUniqueSlice(while0, {})); in TEST_F()
2095 const HloInstruction* while0 = while1->operand(0)->operand(0); in TEST_F() local
2096 ASSERT_EQ(while0->opcode(), HloOpcode::kWhile); in TEST_F()
[all …]
|
| D | bfloat16_propagation_test.cc | 864 auto while0 = builder.AddInstruction( in TEST_F() local
872 HloInstruction::CreateGetTupleElement(shape, while0, 0)), in TEST_F()
874 HloInstruction::CreateGetTupleElement(shape, while0, 1)))); in TEST_F()
|