(二)Graph类的实现

基本介绍

位置：tensorflow/core/graph/graph.h

class Edge;
class EdgeSetTest;
class Graph;
class GraphDef;
class Node;
class VersionDef;
class WhileContext;

class NeighborIter;  // Declared below
class NodeIter;      // Declared below
class NodeProperties;  // Defined in .cc

此部分主要涉及图结构里的节点、边、图等类的声明与定义。下面主要对class Node, class Edge, class Graph展开分析。

  

解析

(一)class Node:

其成员函数提供了一系列对节点进行操作的接口，如节点的邻节点检索、入/出节点检索、节点属性查询等。需要主要的一些细节是，class Graph是节点类的友元所以可以访问其内部的所有成员变量，而节点的属性由std::shared_ptr<NodeProperties> props_描述。

不同种类的类节点：

  // A set of mutually exclusive classes for different kinds of nodes,
  // class_ is initialized in the Node::Initialize routine based on the
  // node's type_string().
  enum NodeClass {
    NC_UNINITIALIZED,
    NC_SWITCH,
    NC_MERGE,
    NC_ENTER,
    NC_EXIT,
    NC_NEXT_ITERATION,
    NC_LOOP_COND,
    NC_CONTROL_TRIGGER,
    NC_SEND,
    NC_HOST_SEND,
    NC_RECV,
    NC_HOST_RECV,
    NC_CONSTANT,
    NC_VARIABLE,
    NC_IDENTITY,
    NC_GET_SESSION_HANDLE,
    NC_GET_SESSION_TENSOR,
    NC_DELETE_SESSION_TENSOR,
    NC_METADATA,
    NC_OTHER  // Not a special kind of node
  };

(二)class Edge:

其程成员函数提供了对边源节点与目标节点访问的借口，同时class EdgeSetTest和class Graph是其友元类，并且当前class Edge用户自己不能实例化。

除此之外，需要额外补充的是关于类class GraphEdgesIterable,该类实现对图对象中的边进行迭代器遍历，并保证跳过null的项。同时，其内部class const_iterator类的构造揭示了迭代器类的实现细节。

(三)class Graph:

 图结构包含很多不同的ops，其最主要的成员变量为std::vector<Node*> nodes_和std::vector<Edge*> edges_，其成员函数包含了对这些节点与边的添加、复制、删除、更新、查找、校验等操作。

Graph对象可使用以下两种方式构造，二者构造的图包含单源节点和单汇节点：

 //可容纳registry中的ops
 1 Graph::Graph(const OpRegistryInterface* ops)
    : ops_(ops, FunctionDefLibrary()),
      versions_(new VersionDef),
      arena_(8 << 10 /* 8kB */) {
  versions_->set_producer(TF_GRAPH_DEF_VERSION);
  versions_->set_min_consumer(TF_GRAPH_DEF_VERSION_MIN_CONSUMER);

  // Initialize the name interning table for assigned_device_name.
  device_names_.push_back("");
  DCHECK_EQ(0, InternDeviceName(""));

  // Source and sink have no endpoints, just control edges.
  NodeDef def;
  def.set_name("_SOURCE");
  def.set_op("NoOp");
  Status status;
  Node* source = AddNode(def, &status);
  TF_CHECK_OK(status);
  CHECK_EQ(source->id(), kSourceId);

  def.set_name("_SINK");
  Node* sink = AddNode(def, &status);
  TF_CHECK_OK(status);
  CHECK_EQ(sink->id(), kSinkId);

  AddControlEdge(source, sink);
}


 // 可容纳flib_def中的ops
Graph::Graph(const FunctionLibraryDefinition& flib_def)
    : Graph(flib_def.default_registry()) {
  // Need a new-enough consumer to support the functions we add to the graph.
  if (flib_def.ToProto().function_size() > 0 &&
      versions_->min_consumer() < 12) {
    versions_->set_min_consumer(12);
  }
  Status s = ops_.AddLibrary(flib_def);
  CHECK(s.ok()) << s.error_message();
}

当前文件里，也包含节点、邻节点的类迭代器，用于方便对其进行操作。