Method AddCleanupFunction

tensorflow/core/framework/dataset.h:601–603

Registers a cleanup function to be called upon object destruction. Registered functions are invoked in the reserve order of registration.

  //
  // Registered functions are invoked in the reserve order of registration.
  void AddCleanupFunction(std::function<void()>&& cleanup_fn) {
    cleanup_fns_.push_back(std::move(cleanup_fn));
  }

  // Associates the given performance modeling `Node` with this iterator.
  void SetNode(std::shared_ptr<model::Node> node) { node_ = node.get(); }