Searched refs:GpuStreamHandle (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 74 static bool CreateStream(GpuContext* context, GpuStreamHandle* stream,
81 static void DestroyStream(GpuContext* context, GpuStreamHandle* stream);
265 GpuStreamHandle stream, void** kernel_params, void** extra);
325 GpuStreamHandle stream);
334 GpuStreamHandle stream);
353 GpuStreamHandle stream);
356 GpuStreamHandle stream);
359 GpuStreamHandle stream);
370 typedef void (*StreamCallback)(GpuStreamHandle stream, GpuStatus status,
376 static bool AddStreamCallback(GpuContext* context, GpuStreamHandle stream,
[all …]
|
| D | gpu_stream.h | 70 GpuStreamHandle gpu_stream() const { in gpu_stream()
72 return const_cast<GpuStreamHandle>(gpu_stream_); in gpu_stream()
76 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream()
82 GpuStreamHandle gpu_stream_; // Wrapped CUDA stream handle.
94 GpuStreamHandle AsGpuStreamValue(Stream* stream);
|
| D | gpu_types.h | 47 using GpuStreamHandle = hipStream_t; variable
65 using GpuStreamHandle = CUstream;
|
| D | gpu_stream.cc | 56 GpuStreamHandle AsGpuStreamValue(Stream* stream) { in AsGpuStreamValue()
|
| D | gpu_executor.h | 258 static void InternalHostCallback(GpuStreamHandle stream, GpuStatus status,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | custom_call_test.cc | 56 void Callback_IsInvoked(se::gpu::GpuStreamHandle /*stream*/, void** /*buffers*/, in Callback_IsInvoked() argument
79 void Callback_Memcpy(se::gpu::GpuStreamHandle stream, void** buffers, in Callback_Memcpy()
99 void Callback_Opaque(se::gpu::GpuStreamHandle /*stream*/, void** /*buffers*/, in Callback_Opaque() argument
112 void Callback_SubBuffers(se::gpu::GpuStreamHandle stream, void** buffers, in Callback_SubBuffers()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 402 unsigned int shared_mem_bytes, GpuStreamHandle stream, void** kernel_params, in LaunchKernel()
479 size_t uint32_count, GpuStreamHandle stream) { in AsynchronousMemsetUint8()
489 size_t uint32_count, GpuStreamHandle stream) { in AsynchronousMemsetUint32()
500 GpuStreamHandle stream, in AddStreamCallback()
561 GpuStreamHandle* stream, in CreateStream()
584 GpuStreamHandle* stream) { in DestroyStream()
726 GpuStreamHandle stream) { in RecordEvent()
783 GpuStreamHandle stream, in WaitStreamOnEvent()
809 GpuStreamHandle stream) { in SynchronizeStream()
820 GpuStreamHandle stream) { in IsStreamIdle()
[all …]
|
| D | rocm_gpu_executor.cc | 301 GpuStreamHandle hipstream = AsGpuStreamValue(stream); in Launch()
570 /* static */ void GpuExecutor::InternalHostCallback(GpuStreamHandle stream, in InternalHostCallback()
|