Searched refs:gpu_stream (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 84 ":gpu_stream",
134 hdrs = if_gpu_is_configured(["gpu_stream.h"]),
143 name = "gpu_stream",
144 srcs = if_gpu_is_configured(["gpu_stream.cc"]),
145 hdrs = if_gpu_is_configured(["gpu_stream.h"]),
173 ":gpu_stream",
|
| D | gpu_stream.h | 68 GpuStreamHandle gpu_stream() const { in gpu_stream() function
74 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream()
|
| D | gpu_timer.cc | 76 parent_->gpu_context(), start_event_, stream->gpu_stream()); in Start()
85 parent_->gpu_context(), stop_event_, stream->gpu_stream()); in Stop()
|
| D | gpu_event.cc | 41 stream->gpu_stream()); in Record()
|
| D | gpu_stream.cc | 57 return AsGpuStream(stream)->gpu_stream(); in AsGpuStreamValue()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 120 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize() argument
131 stream_ = gpu_stream; in Reinitialize()
753 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize() argument
756 stream_device_.Reinitialize(context, gpu_stream, tf_gpu_id, base_allocator, in Reinitialize()
936 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice() local
938 concrete_device->Reinitialize(context, gpu_stream, tf_gpu_id_, allocator, in ReinitializeDevice()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | BUILD | 112 "//tensorflow/stream_executor/gpu:gpu_stream",
|
| D | rocm_gpu_executor.cc | 593 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 388 "//tensorflow/stream_executor/gpu:gpu_stream",
|
| D | cuda_gpu_executor.cc | 712 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()
|