HomeSort by relevance Sort by last modified time
    Searched refs:StreamAssignment (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
stream_assignment.h 27 class StreamAssignment {
32 // `hlo` needs to outlive this StreamAssignment object.
41 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module);
stream_assignment.cc 28 bool StreamAssignment::HasStreamAssigned(const HloInstruction& hlo) const {
32 int StreamAssignment::StreamNumberForHlo(const HloInstruction& hlo) const {
36 void StreamAssignment::AssignStreamToHlo(const HloInstruction* hlo,
66 const HloInstruction& hlo, const StreamAssignment& stream_assignment,
122 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module) {
123 auto stream_assignment = absl::make_unique<StreamAssignment>();
gpu_hlo_schedule.h 44 const HloModule& module, const StreamAssignment& stream_assignment,
thunk_schedule.h 50 std::unique_ptr<StreamAssignment> stream_assignment,
63 // Delegates to StreamAssignment.
91 std::unique_ptr<StreamAssignment> stream_assignment_;
gpu_hlo_schedule.cc 39 const StreamAssignment& stream_assignment,
58 const HloModule* module, const StreamAssignment& stream_assignment,
193 const HloModule& module, const StreamAssignment& stream_assignment,
stream_assignment_test.cc 61 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module);
82 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module);
126 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module);
gpu_hlo_schedule_test.cc 42 const HloModule& module, const StreamAssignment& streams) {
85 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module);
145 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module);
215 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module);
294 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module);
thunk_schedule.cc 52 std::unique_ptr<StreamAssignment> stream_assignment,
nvptx_compiler.cc 659 std::unique_ptr<StreamAssignment> stream_assignment = AssignStreams(*module);
    [all...]

Completed in 283 milliseconds