Searched refs:while_loops (Results 1 – 3 of 3) sorted by relevance
73 std::vector<const HloInstruction*> while_loops; in GetWhileLoops() local76 while_loops.push_back(instruction); in GetWhileLoops()79 return while_loops; in GetWhileLoops()197 auto while_loops = GetWhileLoops(module->entry_computation()); in TEST_F() local198 EXPECT_EQ(1, while_loops.size()); in TEST_F()199 EXPECT_EQ(1, ShapeUtil::TupleElementCount(while_loops[0]->shape())); in TEST_F()307 auto while_loops = GetWhileLoops(module->entry_computation()); in TEST_F() local308 EXPECT_EQ(2, while_loops.size()); in TEST_F()309 EXPECT_EQ(1, ShapeUtil::TupleElementCount(while_loops[0]->shape())); in TEST_F()310 EXPECT_EQ(1, ShapeUtil::TupleElementCount(while_loops[1]->shape())); in TEST_F()[all …]
47 bool while_loops; member
4040 * Bug fix in graph validation related to `tf.while_loops`.