Searched defs:CUDAStream (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
H A Dcuda_stream.h16 // Defines the CUDAStream type - the CUDA-specific implementation of the generic
36 class CUDAStream : public internal::StreamInterface { class in namespace:perftools::gputools::cuda
38 explicit CUDAStream(CUDAExecutor *parent) function in class:perftools::gputools::cuda::CUDAStream
42 ~CUDAStream() override {}
67 // Precond: this CUDAStream has been allocated (otherwise passing a nullptr
85 // Converts a Stream to the underlying CUDAStream implementation.
86 CUDAStream *AsCUDAStream(Stream *stream);
88 // Extracts a CUstream from a CUDAStream-backed Stream object.

Completed in 125 milliseconds