Home
last modified time | relevance | path

Searched refs:SynchronizeStream (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h334 static port::Status SynchronizeStream(GpuContext* context,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc652 return GpuDriver::SynchronizeStream(context_, AsGpuStreamValue(stream)); in BlockHostUntilDone()
Drocm_driver.cc860 /* static */ port::Status GpuDriver::SynchronizeStream(GpuContext* context, in SynchronizeStream() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc771 return GpuDriver::SynchronizeStream(context_, AsGpuStreamValue(stream)); in BlockHostUntilDone()
Dcuda_driver.cc1047 /* static */ port::Status GpuDriver::SynchronizeStream(GpuContext* context, in SynchronizeStream() function in stream_executor::gpu::GpuDriver