Searched defs:cu_stream (Results 1 – 7 of 7) sorted by relevance
50 auto* cu_stream = ctx->eigen_device<GPUDevice>().stream(); in operator ()() local
147 const cudaStream_t& cu_stream = GetCudaStream(context); in Compute() local
329 const cudaStream_t& cu_stream = GetCudaStream(c); in RadixSort() local362 const cudaStream_t& cu_stream = GetCudaStream(c); in CountAndSortParts() local
286 cudaStream_t* cu_stream = reinterpret_cast<cudaStream_t*>( in DoAllReduce() local
235 const cudaStream_t& cu_stream = GetCudaStream(ctx); in LSTMBlockCellFpropWithCUDA() local374 const cudaStream_t& cu_stream = GetCudaStream(ctx); in LSTMBlockCellBpropWithCUDA() local
544 const cudaStream_t* cu_stream = reinterpret_cast<const cudaStream_t*>( in LoopKernelLaunches() local
187 CUstream cu_stream = stream ? AsGpuStreamValue(stream) : cudaStreamLegacy; in GetHandle() local