Home
last modified time | relevance | path

Searched refs:GpuStreamHandle (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h77 static bool CreateStream(GpuContext* context, GpuStreamHandle* stream);
83 static void DestroyStream(GpuContext* context, GpuStreamHandle* stream);
208 GpuStreamHandle stream, void** kernel_params,
267 GpuStreamHandle stream);
275 GpuStreamHandle stream);
294 GpuStreamHandle stream);
297 GpuStreamHandle stream);
300 GpuStreamHandle stream);
311 typedef void (*StreamCallback)(GpuStreamHandle stream, GpuStatus status,
317 static bool AddStreamCallback(GpuContext* context, GpuStreamHandle stream,
[all …]
Dgpu_stream.h68 GpuStreamHandle gpu_stream() const { in gpu_stream()
70 return const_cast<GpuStreamHandle>(gpu_stream_); in gpu_stream()
74 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream()
80 GpuStreamHandle gpu_stream_; // Wrapped CUDA stream handle.
91 GpuStreamHandle AsGpuStreamValue(Stream* stream);
Dgpu_types.h47 using GpuStreamHandle = hipStream_t; variable
65 using GpuStreamHandle = CUstream;
Dgpu_stream.cc55 GpuStreamHandle AsGpuStreamValue(Stream* stream) { in AsGpuStreamValue()
Dgpu_executor.h260 static void InternalHostCallback(GpuStreamHandle stream, GpuStatus status,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc439 unsigned int shared_mem_bytes, GpuStreamHandle stream, void** kernel_params, in LaunchKernel()
529 GpuStreamHandle stream) { in AsynchronousMemsetUint8()
545 GpuStreamHandle stream) { in AsynchronousMemsetUint32()
559 GpuStreamHandle stream, in AddStreamCallback()
620 GpuStreamHandle* stream) { in CreateStream()
636 GpuStreamHandle* stream) { in DestroyStream()
778 GpuStreamHandle stream) { in RecordEvent()
835 GpuStreamHandle stream, in WaitStreamOnEvent()
861 GpuStreamHandle stream) { in SynchronizeStream()
877 GpuStreamHandle stream) { in IsStreamIdle()
[all …]
Drocm_gpu_executor.cc290 GpuStreamHandle hipstream = AsGpuStreamValue(stream); in Launch()
571 /* static */ void GpuExecutor::InternalHostCallback(GpuStreamHandle stream, in InternalHostCallback()