Searched defs:cu_stream (Results 1 – 8 of 8) sorted by relevance
110 const auto& cu_stream = GetGpuStream(context); variable183 const auto& cu_stream = GetGpuStream(context); in GpuInclusivePrefixSum() local219 const auto& cu_stream = GetGpuStream(context); in GpuSegmentedReduce() local253 const auto& cu_stream = GetGpuStream(context); variable
328 const auto& cu_stream = GetGpuStream(c); in RadixSort() local361 const auto& cu_stream = GetGpuStream(c); in CountAndSortParts() local
213 const auto& cu_stream = GetGpuStream(context); in Compute() local
448 const auto& cu_stream = GetGpuStream(ctx); in LaunchSortKernel() local
240 const auto& cu_stream = GetGpuStream(ctx); in LSTMBlockCellFpropWithCUDA() local381 const auto& cu_stream = GetGpuStream(ctx); in LSTMBlockCellBpropWithCUDA() 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
215 CUstream cu_stream = stream ? AsGpuStreamValue(stream) : cudaStreamLegacy; in GetHandle() local225 CUstream cu_stream = AsGpuStreamValue(stream); in NotifyStreamDestroyed() local