Function CheckNode

tensorflow/core/grappler/mutable_graph_view_test.cc:69–86

}

void CheckNode(const MutableGraphView& graph, absl::string_view node_name,
               absl::string_view op, absl::string_view device,
               absl::Span<const std::pair<string, FDH::AttrValueWrapper>> attrs,
               absl::Span<const string> fanins,
               absl::Span<const string> fanouts) {
  NodeDef* node = graph.GetNode(node_name);
  ASSERT_NE(node, nullptr);
  EXPECT_EQ(node->op(), op);
  EXPECT_EQ(node->device(), device);
  EXPECT_EQ(node->attr_size(), attrs.size());
  for (const auto& attr : attrs) {
    auto it = node->attr().find(attr.first);
    ASSERT_NE(it, node->attr().end());
    EXPECT_TRUE(AreAttrValuesEqual(it->second, attr.second.proto));
  }
  CompareNodeFanins(graph, node, fanins);
  CompareNodeFanouts(graph, node, fanouts);
}

void CheckGraph(const MutableGraphView& mutable_graph) {
  GraphView immutable_graph(mutable_graph.graph());