Searched refs:main_stream (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc133 se::Stream* main_stream = run_options->stream(); local
136 main_stream->parent()->GetDeviceDescription().cuda_compute_capability(
150 HloExecutionProfiler profiler(do_profile, hlo_execution_profile, main_stream,
155 // Stream 0 indicates `main_stream` and substreams start from stream 1.
161 run_options->BorrowStream(main_stream->parent()->device_ordinal()));
173 (stream_no == 0 ? main_stream : sub_streams[stream_no - 1].get());
189 TF_RETURN_IF_ERROR(main_stream->BlockHostUntilDone());
199 auto finish_event = MakeUnique<se::Event>(main_stream->parent());
218 main_stream->ThenWaitFor(&sub_streams);
223 Status block_status = main_stream
[all...]

Completed in 190 milliseconds