Home
last modified time | relevance | path

Searched refs:while0 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc2088 auto while0 = builder.AddInstruction( in TEST_F() local
2096 HloInstruction::CreateGetTupleElement(data_shape_, while0, 2)); in TEST_F()
2108 assignment->GetUniqueSlice(while0, {0}).ConsumeValueOrDie()); in TEST_F()
2111 assignment->GetUniqueSlice(while0, {1}).ConsumeValueOrDie()); in TEST_F()
2113 EXPECT_EQ(assignment->GetUniqueSlice(while0, {2}).ConsumeValueOrDie(), in TEST_F()
2176 const HloInstruction* while0 = while1->operand(0)->operand(0); in TEST_F() local
2177 ASSERT_EQ(while0->opcode(), HloOpcode::kWhile); in TEST_F()
2184 assignment->GetUniqueSlice(while0, {})); in TEST_F()
2243 const HloInstruction* while0 = while1->operand(0)->operand(0); in TEST_F() local
2244 ASSERT_EQ(while0->opcode(), HloOpcode::kWhile); in TEST_F()
[all …]
Dbfloat16_propagation_test.cc894 auto while0 = builder.AddInstruction( in TEST_F() local
902 HloInstruction::CreateGetTupleElement(shape, while0, 0)), in TEST_F()
904 HloInstruction::CreateGetTupleElement(shape, while0, 1)))); in TEST_F()
Dmemory_space_assignment_test.cc3754 auto while0 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P() local
3758 schedule.set_sequence(computation, {param, gte0, gte1, tuple, while0}); in TEST_P()
3780 EXPECT_THAT(while0, op::ShapeWithLayout(t_s32_f32v1_in_default_mem)); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc233 HloInstruction* while0 = builder.AddInstruction( in CreateTestModule() local
236 HloInstruction::CreateGetTupleElement(s32_, while0, 0)); in CreateTestModule()
238 HloInstruction::CreateGetTupleElement(f32v1_, while0, 1)); in CreateTestModule()