Searched refs:ThunkLaunchOrder (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule.h | 49 const std::vector<HloInstruction*>& ThunkLaunchOrder() const { in ThunkLaunchOrder() function
|
| D | gpu_hlo_schedule_test.cc | 91 EXPECT_EQ(RemoveHlo(schedule->ThunkLaunchOrder(), {x, y, z}), in TEST_F()
153 EXPECT_EQ(RemoveHlo(schedule->ThunkLaunchOrder(), {x, y, z}), in TEST_F()
221 HloVec thunk_launch_order = RemoveHlo(schedule->ThunkLaunchOrder(), {x, y}); in TEST_F()
|
| D | gpu_compiler.cc | 613 absl::Span<HloInstruction* const> order = hlo_schedule->ThunkLaunchOrder(); in CompileModuleToLlvmIrImpl()
|