HomeSort by relevance Sort by last modified time
    Searched full:cuda_stream_ (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_stream.h 39 : parent_(parent), cuda_stream_(nullptr), completed_event_(nullptr) {}
44 void *CudaStreamHack() override { return cuda_stream_; }
46 return reinterpret_cast<void **>(&cuda_stream_);
70 DCHECK(cuda_stream_ != nullptr);
71 return const_cast<CUstream>(cuda_stream_);
78 CUstream cuda_stream_; // Wrapped CUDA stream handle. member in class:perftools::gputools::cuda::CUDAStream
cuda_stream.cc 27 if (!CUDADriver::CreateStream(parent_->cuda_context(), &cuda_stream_)) {
44 CUDADriver::DestroyStream(parent_->cuda_context(), &cuda_stream_);
48 return CUDADriver::IsStreamIdle(parent_->cuda_context(), cuda_stream_);
  /external/tensorflow/tensorflow/core/kernels/
cuda_solvers.cc 155 cuda_stream_ = *cu_stream_ptr;
157 auto it = handle_map->find(cuda_stream_);
159 LOG(INFO) << "Creating CudaSolver handles for stream " << cuda_stream_;
163 new CudaSolverHandles(cuda_stream_));
165 handle_map->insert(std::make_pair(cuda_stream_, std::move(new_handles)))
    [all...]
cuda_solvers.h 314 cudaStream_t cuda_stream_; member in class:tensorflow::CudaSolver

Completed in 222 milliseconds