Searched refs:fetch_node (Results 1 – 9 of 9) sorted by relevance
159 Node* fetch_node; in FetchOutputs() local161 fetch_rewrites[i]->AddNode(g, {n, id.second}, &fetch_node)); in FetchOutputs()164 (*name_index)[fetch_node->name()] = fetch_node; in FetchOutputs()168 g->AddControlEdge(fetch_node, g->sink_node(), true); in FetchOutputs()169 out_fetch_nodes->push_back(fetch_node); in FetchOutputs()
123 Node* fetch_node = it->second; in AddRetvalNodes() local124 if (id.output_index() >= fetch_node->num_outputs()) { in AddRetvalNodes()127 fetch_node->num_outputs()); in AddRetvalNodes()134 .Input(fetch_node, id.output_index()) in AddRetvalNodes()135 .Attr("T", BaseType(fetch_node->output_type(id.output_index()))) in AddRetvalNodes()
90 string fetch_node = ""; in FromGraph() local93 fetch_node = node.input(0); in FromGraph()96 if (fetch_node.empty()) { in FromGraph()105 {fetch_node}, &outputs)); in FromGraph()
45 const NodeDef* fetch_node = nullptr) { in RunAndFetch() argument49 if (fetch_node) { in RunAndFetch()50 *graph.add_node() = *fetch_node; in RunAndFetch()
499 const NodeDef* fetch_node = nullptr) { in RunAndFetch() argument503 if (fetch_node) { in RunAndFetch()504 *graph.add_node() = *fetch_node; in RunAndFetch()
368 NodeDef** fetch_node) { in GetFetchNode() argument375 *fetch_node = graph.GetNode(item.fetch.at(0)); in GetFetchNode()
171 NodeDef** fetch_node);
306 for (const auto& fetch_node : cfg.fetch_nodes) { in GrapplerItemFromMetaGraphDef() local307 new_item->fetch.emplace_back(NodeName(fetch_node)); in GrapplerItemFromMetaGraphDef()
4001 const NodeDef* fetch_node = node_map_->GetNode(fetch); in RunOptimizationPass() local4002 if (fetch_node && NumOutputs(*fetch_node, graph_) == 1) { in RunOptimizationPass()4003 nodes_allowlist_.insert(fetch_node->name()); in RunOptimizationPass()