Searched refs:cuda_stream (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 294 ":cuda_stream",
354 ":cuda_stream",
402 ":cuda_stream",
457 ":cuda_stream",
564 ":cuda_stream",
573 name = "cuda_stream",
575 hdrs = if_cuda_is_configured(["cuda_stream.h"]),
594 ":cuda_stream",
622 ":cuda_stream",
|
| D | cuda_gpu_executor.cc | 704 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream() local
705 if (!cuda_stream->IsIdle()) { in DeallocateStream()
708 cuda_stream->Destroy(); in DeallocateStream()
|
| D | cuda_blas.cc | 3604 cudaStream_t cuda_stream = CUDAStream(stream); in DoBlasInternalImpl() local
3638 cuda_algo.workspace_size(), cuda_stream); in DoBlasInternalImpl()
3674 remainder_algo.workspace_size(), cuda_stream); in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_context.cc | 96 const cudaStream_t* cuda_stream = in Create() local
100 CusolverStatusToStatus(cusolverDnSetStream(handle, *cuda_stream))); in Create()
|
| D | BUILD | 662 "//tensorflow/stream_executor/cuda:cuda_stream",
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_stream.h | 76 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream() function
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | generate_box_proposals_op.cu.cc | 346 auto cuda_stream = GetGpuStream(context); in Compute() local
355 cuda_stream); in Compute()
364 f4_ptr, int_ptr, image_stride * num_anchors, cuda_stream)); in Compute()
407 cuda_stream)); in Compute()
|
| D | non_max_suppression_op.cu.cc | 381 auto cuda_stream = tensorflow::GetGpuStream(context); in CountIf() local
400 num_elements, op, cuda_stream)); in CountIf()
414 auto cuda_stream = GetGpuStream(context); in DoNMS() local
435 cuda_stream); in DoNMS()
473 cuda_stream); in DoNMS()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3943 "//tensorflow/stream_executor/cuda:cuda_stream",
|