Method SameNode

tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc:3527–3573

PRECONDITION: Node input orders are assumed to be canonicalized, i.e. control inputs for all nodes as well as regular inputs for commutative nodes must be sorted.

//  Node input orders are assumed to be canonicalized, i.e. control inputs for
//  all nodes as well as regular inputs for commutative nodes must be sorted.
bool UniqueNodes::SameNode(const NodeDef& node1, const NodeDef& node2) const {
  if (node1.op() != node2.op()) {
    return false;
  }
  if (node1.device() != node2.device()) {
    return false;
  }
  if (node1.input_size() != node2.input_size()) {
    return false;
  }

  int node1_attr_size = node1.attr_size();
  int node2_attr_size = node2.attr_size();
  // exclude 'stream_id' attribute
  for (const auto& attr : node1.attr()) {
    if (IsExcludeAttribute(attr.first)) {
      node1_attr_size -= 1;
    }
  }
  for (const auto& attr : node2.attr()) {
    if (IsExcludeAttribute(attr.first)) {
      node2_attr_size -= 1;
    }
  }
  if (node1_attr_size != node2_attr_size) {
    return false;
  }

  // Compare inputs.
  auto it1 = node1.input().begin();
  auto it2 = node2.input().begin();
  for (; it1 != node1.input().end(); ++it1, ++it2) {
    if (*it1 != *it2) return false;
  }

  // Compare attributes.
  for (const auto& attr1 : node1.attr()) {
    // exclude 'stream_id' attribute
    if (IsExcludeAttribute(attr1.first))
      continue;
    auto it = node2.attr().find(attr1.first);
    if (it == node2.attr().end()) return false;
    if (!FastAreAttrValuesEqual(attr1.second, it->second)) return false;
  }

  return true;
}

bool UniqueNodes::IsExcludeAttribute(const std::string& attr_name) const {
  if (exclude_attrs_.find(attr_name) != exclude_attrs_.end())