Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h317 static bool AddStreamCallback(GpuContext* context, GpuStreamHandle stream,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc567 return GpuDriver::AddStreamCallback(context_, AsGpuStreamValue(stream), in HostCallback()
Drocm_driver.cc558 /* static */ bool GpuDriver::AddStreamCallback(GpuContext* context, in AddStreamCallback() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc686 return GpuDriver::AddStreamCallback(context_, AsGpuStreamValue(stream), in HostCallback()
Dcuda_driver.cc717 /* static */ bool GpuDriver::AddStreamCallback(GpuContext* context, in AddStreamCallback() function in stream_executor::gpu::GpuDriver