Searched refs:cuda_stream (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 197 ":cuda_stream",
240 ":cuda_stream",
280 ":cuda_stream",
322 ":cuda_stream",
371 ":cuda_stream",
380 name = "cuda_stream",
382 hdrs = if_cuda_is_configured(["cuda_stream.h"]),
401 ":cuda_stream",
419 ":cuda_stream",
|
| D | cuda_gpu_executor.cc | 732 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream() local
733 if (!cuda_stream->IsIdle()) { in DeallocateStream()
736 cuda_stream->Destroy(); in DeallocateStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_context.cc | 95 const cudaStream_t* cuda_stream = in Create() local
99 CusolverStatusToStatus(cusolverDnSetStream(handle, *cuda_stream))); in Create()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_stream.h | 74 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3918 "//tensorflow/stream_executor/cuda:cuda_stream",
|