tf::cudaStreamCreator struct

functor to create a cudaStream_t object

Public functions

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