Lines Matching refs:CUDAStream

257 CUDAStream CUDAStreamForId(DeviceIndex device_index, StreamId stream_id) {  in CUDAStreamForId()
258   return CUDAStream(  in CUDAStreamForId()
259       CUDAStream::UNCHECKED,  in CUDAStreamForId()
269 cudaStream_t CUDAStream::stream() const {  in stream()
315 CUDAStream getStreamFromPool(const int priority, DeviceIndex device_index) {  in getStreamFromPool()
340 CUDAStream getStreamFromPool(const bool isHighPriority, DeviceIndex device) {  in getStreamFromPool()
346 CUDAStream getStreamFromExternal(  in getStreamFromExternal()
353 CUDAStream getDefaultCUDAStream(DeviceIndex device_index) {  in getDefaultCUDAStream()
363 CUDAStream getCurrentCUDAStream(DeviceIndex device_index) {  in getCurrentCUDAStream()
373 void setCurrentCUDAStream(CUDAStream stream) {  in setCurrentCUDAStream()
378 std::ostream& operator<<(std::ostream& stream, const CUDAStream& s) {  in operator <<()