Searched refs:HloVec (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule_test.cc | 36 using HloVec = std::vector<HloInstruction*>; typedef in xla::gpu::GpuHloScheduleTest
55 HloVec RemoveHlo(const HloVec& input, in RemoveHlo()
57 HloVec result(input); in RemoveHlo()
92 HloVec({dot1, dot2})); in TEST_F()
154 HloVec({add1, add2, add3})); in TEST_F()
221 HloVec thunk_launch_order = RemoveHlo(schedule->ThunkLaunchOrder(), {x, y}); in TEST_F()
222 EXPECT_TRUE(thunk_launch_order == HloVec({dot1, dot2, add}) || in TEST_F()
223 thunk_launch_order == HloVec({dot2, dot1, add})); in TEST_F()
314 const HloVec all_params( in TEST_F()
316 const HloVec all_ops({d00, d10, d11, d20, d21, d22, d30, d31, d40}); in TEST_F()
|