Home
last modified time | relevance | path

Searched refs:StreamNumberForHlo (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment_test.cc62 EXPECT_EQ(assignment->StreamNumberForHlo(*dot1), in TEST_F()
63 assignment->StreamNumberForHlo(*dot2)); in TEST_F()
83 EXPECT_NE(assignment->StreamNumberForHlo(*dot1), in TEST_F()
84 assignment->StreamNumberForHlo(*dot2)); in TEST_F()
128 EXPECT_NE(assignment->StreamNumberForHlo(*d10), in TEST_F()
129 assignment->StreamNumberForHlo(*d11)); in TEST_F()
131 EXPECT_NE(assignment->StreamNumberForHlo(*d20), in TEST_F()
132 assignment->StreamNumberForHlo(*d21)); in TEST_F()
133 EXPECT_NE(assignment->StreamNumberForHlo(*d20), in TEST_F()
134 assignment->StreamNumberForHlo(*d22)); in TEST_F()
[all …]
Dgpu_hlo_schedule_test.cc86 EXPECT_EQ(streams->StreamNumberForHlo(*dot1), in TEST_F()
87 streams->StreamNumberForHlo(*dot2)); in TEST_F()
146 EXPECT_EQ(streams->StreamNumberForHlo(*add1), in TEST_F()
147 streams->StreamNumberForHlo(*add2)); in TEST_F()
148 EXPECT_EQ(streams->StreamNumberForHlo(*add1), in TEST_F()
149 streams->StreamNumberForHlo(*add3)); in TEST_F()
216 EXPECT_NE(streams->StreamNumberForHlo(*dot1), in TEST_F()
217 streams->StreamNumberForHlo(*dot2)); in TEST_F()
296 EXPECT_NE(streams->StreamNumberForHlo(*d10), in TEST_F()
297 streams->StreamNumberForHlo(*d11)); in TEST_F()
[all …]
Dthunk_schedule.cc36 if (stream_assignment_->StreamNumberForHlo(operand) != in AddDependenciesOnTransitiveOperands()
37 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(&thunk))) { in AddDependenciesOnTransitiveOperands()
125 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(dst)); in RemoveRedundantDependencyEdges()
131 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(src)); in RemoveRedundantDependencyEdges()
Dstream_assignment.cc33 int StreamAssignment::StreamNumberForHlo(const HloInstruction& hlo) const { in StreamNumberForHlo() function in xla::gpu::StreamAssignment
94 stream_assignment.StreamNumberForHlo(*operand)); in ComputeStreamToAssign()
109 int stream_num = stream_assignment.StreamNumberForHlo(*seen_gemm); in ComputeStreamToAssign()
Dstream_assignment.h30 int StreamNumberForHlo(const HloInstruction& hlo) const;
Dthunk_schedule.h80 return stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(thunk)); in StreamNumberForThunk()
Dgpu_hlo_schedule.cc100 const int stream_no = stream_assignment.StreamNumberForHlo(*hlo); in GpuHloOrdering()