tf::cudaStreamCreator struct

functor to create a cudaStream_t object

Public functions

auto operator()() const -> cudaStream_t
constructs a new cudaStream_t object using cudaStreamCreate
auto operator()(cudaStream_t stream) const -> cudaStream_t
returns the given cudaStream_t object