Searched refs:FindNodeId (Results 1 – 15 of 15) sorted by relevance
67 const tensorflow::Node* node = graph.FindNodeId(node_id); in GetSubGraphIncomingEdges()81 const tensorflow::Node* node = graph.FindNodeId(node_id); in GetSubGraphOutgoingEdges()134 tensorflow::Node* node = graph->FindNodeId(node_id); in ConvertSubGraphToTensorRT()174 tensorflow::Node* node = graph->FindNodeId(node_id); in ConvertSubGraphToTensorRT()
1445 tensorflow::Node* node = graph.FindNodeId(node_id); in ConvertSubGraphToTensorRTNodeDef()1517 tensorflow::Node* node = graph.FindNodeId(node_id); in ConvertSubGraphToTensorRTNodeDef()
522 Node* FindNodeId(int id) const { return nodes_[id]; } in FindNodeId() function539 Node* source_node() const { return FindNodeId(kSourceId); } in source_node()540 Node* sink_node() const { return FindNodeId(kSinkId); } in sink_node()751 if (id_ >= graph_->num_node_ids() || graph_->FindNodeId(id_) != nullptr) {757 inline Node* NodeIter::operator*() { return graph_->FindNodeId(id_); }759 inline Node* NodeIter::operator->() { return graph_->FindNodeId(id_); }
895 const string& source = graph_.FindNodeId(Graph::kSourceId)->name(); in TEST_F()896 const string& sink = graph_.FindNodeId(Graph::kSinkId)->name(); in TEST_F()2428 const string& source = graph_.FindNodeId(Graph::kSourceId)->name(); in TEST_F()2429 const string& sink = graph_.FindNodeId(Graph::kSinkId)->name(); in TEST_F()
541 const Node* node = FindNodeId(id); in ToGraphDefSubRange()
314 Node* node = graph_.FindNodeId(id); in TEST_F()
608 Node* node = g->FindNodeId(i); in AddControlFlow()
259 auto* node = graph.FindNodeId(node_id); in DescribeCycle()480 Node* node_from = graph->FindNodeId(from); in RunImpl()496 Node* node_to = graph->FindNodeId(to); in RunImpl()
149 return graph_->FindNodeId(iter->second); in get_node_by_name()
171 Node* n1 = g1.FindNodeId(i); in TEST_F()
230 return graph.FindNodeId(search->second); in GetNodeByName()
132 Node* n = g.FindNodeId(it.first); in TEST_F()
137 tensorflow::Node* node = graph.FindNodeId(i); in SegmentGraph()
402 const Node* n = g.FindNodeId(i); in ValidateNoCycles()439 nodes_in_cycle.push_back(g.FindNodeId(i)->name()); in ValidateNoCycles()1865 Node* node = graph->graph.FindNodeId(*pos); in TF_GraphNextOperation()2011 auto* node = graph->graph.FindNodeId(i); in GraphImportGraphDefLocked()2332 Node* new_node = parent->graph.FindNodeId(i); in TF_FinishWhileHelper()2397 Node* n = g->graph.FindNodeId(i); in TF_AddGradients()2544 Node* const node = graph.FindNodeId(id); in ExtendSessionGraphHelper()
490 const Node* node = graph.FindNodeId(i); in BuildTensorShapeMapFromGraph()