Searched refs:thunk_launch_order (Results 1 – 2 of 2) sorted by relevance
40 const std::vector<HloInstruction*>& thunk_launch_order);59 const std::vector<HloInstruction*>& thunk_launch_order) in GpuHloOrdering() argument64 absl::make_unique<HloInstructionSequence>(thunk_launch_order); in GpuHloOrdering()86 for (const HloInstruction* hlo : thunk_launch_order) { in GpuHloOrdering()
221 HloVec thunk_launch_order = RemoveHlo(schedule->ThunkLaunchOrder(), {x, y}); in TEST_F() local222 EXPECT_TRUE(thunk_launch_order == HloVec({dot1, dot2, add}) || in TEST_F()223 thunk_launch_order == HloVec({dot2, dot1, add})); in TEST_F()