Method GetNodeByName

tensorflow/core/common_runtime/placer_test.cc:315–319

Returns the node in "graph" with the given name. REQUIRES: "graph" was produced by the most recent call to BuildGraph.

  //
  // REQUIRES: "graph" was produced by the most recent call to BuildGraph.
  Node* GetNodeByName(const Graph& graph, const string& name) {
    const auto search = nodes_by_name_.find(name);
    CHECK(search != nodes_by_name_.end()) << "Unknown node name: " << name;
    return graph.FindNodeId(search->second);
  }

 protected:
  std::vector<std::unique_ptr<Device>> local_devices_;