Method SetStream

dali/python/backend_impl.cc:2815–2818

  }

  void SetStream(py::object cuda_stream) {
    set_output_order(AccessOrderFromPythonStreamObj(cuda_stream));
    py_stream_ = cuda_stream;  // keep the stream python object alive
  }

 private:
  std::shared_ptr<ThreadPool> shared_thread_pool_;