Searched defs:cuda_stream (Results 1 – 10 of 10) sorted by relevance
37 cudaStream_t cuda_stream; in TEST() local184 cudaStream_t cuda_stream = streams[i]; in TEST() local272 const at::cuda::CUDAStream cuda_stream{wait_stream0}; in TEST() local292 const at::cuda::CUDAStream cuda_stream{wait_stream0}; in TEST() local305 cudaStream_t cuda_stream; in TEST() local
21 at::cuda::CUDAStream cuda_stream = c10::cuda::getCurrentCUDAStream(); in run() local28 at::cuda::CUDAStream cuda_stream) { in run_with_cuda_stream()
148 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in _calculate_moving_average() local199 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in _calc_moving_avg_qparams_helper() local
10 at::cuda::CUDAStream cuda_stream; member
579 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in GroupNormKernelImplInternal() local690 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in GroupNorm1dBackward() local817 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in GroupNormBackwardKernelImplInternal() local
792 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in LayerNormKernelImplInternal() local1159 cudaStream_t cuda_stream = at::cuda::getCurrentCUDAStream(); in LayerNormBackwardKernelImplInternal() local
76 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream() function
277 auto cuda_stream = tensorflow::GetGpuStream(context); in CountIf() local321 auto cuda_stream = GetGpuStream(context); in DoNMS() local
346 auto cuda_stream = GetGpuStream(context); in Compute() local
750 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream() local