Method SetNode

tensorflow/core/framework/dataset.h:606–606

Associates the given performance modeling `Node` with this iterator.

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

  std::vector<std::function<void()>> cleanup_fns_;
  model::Node* node_ = nullptr;  // Not owned.