Searched refs:GetInvariantGTEsForWhileBody (Results 1 – 6 of 6) sorted by relevance
136 TEST_F(WhileUtilTest, GetInvariantGTEsForWhileBody) { in TEST_F() argument168 WhileUtil::GetInvariantGTEsForWhileBody(*while_body); in TEST_F()
92 static std::vector<HloInstruction*> GetInvariantGTEsForWhileBody(
65 WhileUtil::GetInvariantGTEsForWhileBody(*while_body); in TrySinkingConstantsIntoWhileLoop()
170 for (auto* instr : WhileUtil::GetInvariantGTEsForWhileBody(*while_body)) { in TryHoistingInvariantInstructionsFromWhileBody()
158 for (auto* instr : WhileUtil::GetInvariantGTEsForWhileBody(*while_body)) { in TryHoistingInvariantInstructionsFromWhileBody()
274 /*static*/ std::vector<HloInstruction*> WhileUtil::GetInvariantGTEsForWhileBody( in GetInvariantGTEsForWhileBody() function in xla::WhileUtil