Searched refs:RemoveHlo (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule_test.cc | 55 HloVec RemoveHlo(const HloVec& input, in RemoveHlo() function in xla::gpu::GpuHloScheduleTest
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()
|