Searched refs:GetCudaStream (Results 1 – 10 of 10) sorted by relevance
67 const cudaStream_t& stream = GetCudaStream(context); in Compute()87 const cudaStream_t& stream = GetCudaStream(context); in Compute()
58 const cudaStream_t& stream = GetCudaStream(context); in Compute()
142 const cudaStream_t& cu_stream = GetCudaStream(ctx);269 const cudaStream_t& cu_stream = GetCudaStream(ctx);
69 const cudaStream_t& stream = GetCudaStream(context); in Compute()
329 const cudaStream_t& cu_stream = GetCudaStream(c); in RadixSort()362 const cudaStream_t& cu_stream = GetCudaStream(c); in CountAndSortParts()
438 const cudaStream_t& cu_stream = GetCudaStream(ctx);552 const cudaStream_t& cu_stream = GetCudaStream(context);
147 const cudaStream_t& cu_stream = GetCudaStream(context); in Compute()
857 const cudaStream_t& cu_stream = GetCudaStream(ctx);
235 const cudaStream_t& cu_stream = GetCudaStream(ctx); in LSTMBlockCellFpropWithCUDA()374 const cudaStream_t& cu_stream = GetCudaStream(ctx); in LSTMBlockCellBpropWithCUDA()
293 inline const cudaStream_t& GetCudaStream(OpKernelContext* context) { in GetCudaStream() function