}

tensorflow::StatusOr<const tensorflow::NodeDef*> TFPackage::GetGraphDefNode(
    std::string name) {
  const auto& iter = graph_def_nodes_by_name_.find(name);
  if (iter == graph_def_nodes_by_name_.end()) {
    return tensorflow::Status(absl::StatusCode::kInternal,
                              absl::StrCat("Failed to find node named ", name));
  }
  return iter->second;
}

  std::unique_ptr<tensorflow::BundleReader> variable_reader_;
  std::string variables_filepath_;
  bool has_checkpoint_;
  absl::flat_hash_map<std::string, const NodeDef*> graph_def_nodes_by_name_;
};

}  // namespace libexport
}  // namespace tensorflow