Searched defs:cuda_stream (Results 1 – 6 of 6) sorted by relevance
76 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream() function
96 const cudaStream_t* cuda_stream = in Create() local
381 auto cuda_stream = tensorflow::GetGpuStream(context); in CountIf() local414 auto cuda_stream = GetGpuStream(context); in DoNMS() local
346 auto cuda_stream = GetGpuStream(context); in Compute() local
704 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream() local
3604 cudaStream_t cuda_stream = CUDAStream(stream); in DoBlasInternalImpl() local