Searched defs:cu_stream (Results 1 – 9 of 9) sorted by relevance
76 cudaStream_t* cu_stream = reinterpret_cast<cudaStream_t*>( in RunNcclCollective() local
68 cudaStream_t* cu_stream = reinterpret_cast<cudaStream_t*>( in RunNcclCollective() local
127 cudaStream_t* cu_stream = reinterpret_cast<cudaStream_t*>( in RunNcclCollective() local
240 const auto& cu_stream = GetGpuStream(ctx); in LSTMBlockCellFpropWithCUDA() local381 const auto& cu_stream = GetGpuStream(ctx); in LSTMBlockCellBpropWithCUDA() local
339 const auto& cu_stream = GetGpuStream(c); in RadixSort() local372 const auto& cu_stream = GetGpuStream(c); in CountAndSortParts() local
213 const auto& cu_stream = GetGpuStream(context); in Compute() local
705 const cudaStream_t* cu_stream = reinterpret_cast<const cudaStream_t*>( in LoopKernelLaunches() local
56 const auto& cu_stream = GetGpuStream(c); in operator ()() local
187 CUstream cu_stream = stream ? AsGpuStreamValue(stream) : cudaStreamLegacy; in GetHandle() local