Home
last modified time | relevance | path

Searched refs:graph_view_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dpattern_utils.h181 SubGraphMatcher(MutableGraphView* graph_view) : graph_view_(graph_view){}; in SubGraphMatcher()
190 MutableGraphView* graph_view_;
205 auto node_view = graph_view_->GetNode(node_idx); in HasRemoveNodeExternalDependents()
Dgraph_view.cc37 : NodeIndexAndPortIndex(node_view->graph_view_, node_view->node_index_, in FaninView()
41 : NodeIndexAndPortIndex(node_view->graph_view_, node_view->node_index_, in FanoutView()
45 return &graph_view_->graph()->node(node_index_); in node()
49 if (fanin.index() < Graph::kControlSlot || graph_view_ != fanin.graph_view_) { in HasFanin()
53 {&graph_view_->graph_->node(fanin.node_index_), fanin.index()}); in HasFanin()
58 graph_view_ != fanout.graph_view_) { in HasFanout()
73 return graph_view_->missing_fanin_; in GetMissingFanin()
77 return graph_view_->missing_fanout_; in GetMissingFanout()
172 : NodeIndexAndPortIndex(node_view->graph_view_, node_view->node_index_, in MutableFaninView()
176 : NodeIndexAndPortIndex(node_view->graph_view_, node_view->node_index_, in MutableFanoutView()
[all …]
Dgraph_view_internal.h46 : graph_view_(nullptr), in NodeIndexAndPortIndex()
50 : graph_view_(graph_view), in NodeIndexAndPortIndex()
56 node_index_ == other.node_index_ && graph_view_ == other.graph_view_;
66 if (graph_view_ == nullptr) { in node_view()
69 return graph_view_->GetNode(node_index_); in node_view()
79 GraphViewT* graph_view_;
131 : graph_view_(graph_view), in NodeViewInternal()
136 : graph_view_(nullptr), node_index_(kMissingIndex), attrs_(AttrSlice()) {} in NodeViewInternal()
144 return node_index_ == other.node_index_ && graph_view_ == other.graph_view_;
252 GraphViewT* graph_view_; variable
Dgraph_view.h363 MutableGraphView* graph_view_ = nullptr; variable
395 if (from_node->graph_view_ == to_node->graph_view_) { in TopologicalDependency()
396 graph_view_ = from_node->graph_view_; in TopologicalDependency()
403 MutableGraphView* graph_view_ = nullptr;
Dpattern_utils.cc87 graph_view_->GetNode(child_node_index); in DoesOpTypePatternMatch()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc117 if (cmd == kCmds[1] && !graph_view_) { in BuildView()
118 graph_view_.reset(new TFGraph(ckpt_reader_.get())); in BuildView()
120 graph_view_->AddNode(it->second.get()); in BuildView()
122 graph_view_->Build(); in BuildView()
162 graph_view_->Show(prefix, nopts); in ShowGraphNode()
166 return graph_view_->Show(prefix, opts); in ShowGraphNode()
234 graph_view_.reset(nullptr); in AddGraph()
Dtfprof_stats.h108 std::unique_ptr<TFGraph> graph_view_; variable
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc937 graph_view_(graph), in AutoMixedPrecisionImpl()
1003 MutableGraphView graph_view_; member in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl
1732 const auto& fanout = graph_view_.GetFanout(output_port); in ForceColorMatchOnRecurrentEdges()
1834 graph_view_.GetRegularFanin(node_input); in NodeImplicitlyReadsNonResourceVariable()
1859 const auto& fanout = graph_view_.GetFanout(src); in MakeCastsAllowIfAllOutputsAllow()
1922 auto fanout = graph_view_.GetFanout(src); in ChangeTypeAttrsAndAddCasts()
1943 added_cast_node = graph_view_.AddNode( in ChangeTypeAttrsAndAddCasts()
1950 TF_RETURN_IF_ERROR(graph_view_.UpdateRegularFaninByPort( in ChangeTypeAttrsAndAddCasts()
Dfunction_optimizer.cc263 graph_view_(&graph) {} in FunctionOptimizerContext()
281 const GraphView& graph_view() const { return graph_view_; } in graph_view()
380 GraphView graph_view_; member in tensorflow::grappler::__anondd39c0ba0111::FunctionOptimizerContext