Searched refs:WaitStreamOnEvent (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 323 static bool WaitStreamOnEvent(GpuContext* context, GpuStreamHandle stream,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 593 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()
639 return GpuDriver::WaitStreamOnEvent(context_, AsGpuStreamValue(dependent), in CreateStreamDependency()
|
| D | rocm_driver.cc | 834 /* static */ bool GpuDriver::WaitStreamOnEvent(GpuContext* context, in WaitStreamOnEvent() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 712 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()
758 return GpuDriver::WaitStreamOnEvent(context_, AsGpuStreamValue(dependent), in CreateStreamDependency()
|
| D | cuda_driver.cc | 1023 /* static */ bool GpuDriver::WaitStreamOnEvent(GpuContext* context, in WaitStreamOnEvent() function in stream_executor::gpu::GpuDriver
|