Home
last modified time | relevance | path

Searched defs:cuda_stream (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dgpu_stream.h76 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream() function
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc277 auto cuda_stream = tensorflow::GetGpuStream(context); in CountIf() local
321 auto cuda_stream = GetGpuStream(context); in DoNMS() local
Dgenerate_box_proposals_op.cu.cc346 auto cuda_stream = GetGpuStream(context); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_gpu_executor.cc750 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream() local