Home
last modified time | relevance | path

Searched refs:StreamNumberForHlo (Results 1 – 8 of 8) 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.hlo_instruction())) { in AddDependenciesOnTransitiveOperands()
119 stream_assignment_->StreamNumberForHlo(*dst->hlo_instruction()); in RemoveRedundantDependencyEdges()
125 stream_assignment_->StreamNumberForHlo(*src->hlo_instruction()); in RemoveRedundantDependencyEdges()
Dstream_assignment.cc32 int StreamAssignment::StreamNumberForHlo(const HloInstruction& hlo) const { in StreamNumberForHlo() function in xla::gpu::StreamAssignment
89 stream_assignment.StreamNumberForHlo(*operand)); in ComputeStreamToAssign()
104 int stream_num = stream_assignment.StreamNumberForHlo(*seen_gemm); in ComputeStreamToAssign()
Dthunk_schedule.h65 int StreamNumberForHlo(const HloInstruction& hlo) const { in StreamNumberForHlo() function
66 return stream_assignment_->StreamNumberForHlo(hlo); in StreamNumberForHlo()
Dstream_assignment.h30 int StreamNumberForHlo(const HloInstruction& hlo) const;
Dgpu_hlo_schedule.cc100 const int stream_no = stream_assignment.StreamNumberForHlo(*hlo); in GpuHloOrdering()
Dgpu_executable.cc127 thunk_schedule_->StreamNumberForHlo(*thunk->hlo_instruction()); in ExecuteThunks()