Home
last modified time | relevance | path

Searched defs:node_view (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgraph_view.cc36 FaninView::FaninView(NodeView* node_view, int index) in FaninView()
40 FanoutView::FanoutView(NodeView* node_view, int index) in FanoutView()
120 Status GraphView::CheckAndAddFaninsInternal(NodeView* node_view) { in CheckAndAddFaninsInternal()
171 MutableFaninView::MutableFaninView(MutableNodeView* node_view, int index) in MutableFaninView()
175 MutableFanoutView::MutableFanoutView(MutableNodeView* node_view, int index) in MutableFanoutView()
544 MutableNodeView& node_view = nodes_[i]; in AddFaninsInternal() local
716 const MutableNodeView& node_view = nodes_[node_name_state.second]; in RemovedOrMissingNodeFanoutsWellFormed() local
956 MutableNodeView& node_view = nodes_[node_index]; in AddNewNodes() local
1015 MutableNodeView* node_view, int i) { in RemoveRegularFaninFanoutInternal()
1061 MutableNodeView* node_view, const SafeTensorId& fanin_id) { in AddRegularFaninInternal()
[all …]
H A Dgraph_view_internal.h65 NodeViewT* node_view() const { in node_view() function
495 auto* node_view = diff->graph_view->GetNode(diff->node_index); in AddOrUpdateRegularFanin() local
537 auto* node_view = diff->graph_view->GetNode(diff->node_index); in RemoveRegularFanin() local
616 auto* node_view = diff->graph_view->GetNode(diff->node_index); in RemoveAttribute() local
696 auto* node_view = diff->graph_view->GetNode(diff->node_index); in IsWellFormed() local
H A Dpattern_utils.cc58 const OpTypePattern& pattern, MutableNodeView* node_view, in DoesOpTypePatternMatch()
171 MutableNodeView* node_view, std::map<string, int>* matched_nodes_map, in GetMatchedNodes()
H A Dpattern_utils.h148 MutableNodeView* node_view = nullptr; member
208 auto node_view = graph_view_->GetNode(node_idx); in IsSafeNodesToRemove() local
H A Dgraph_view_internal_test.cc47 for (const auto& node_view : graph_view->GetNodes()) { in GetUpdatedNodeNames() local
H A Dgraph_view_test.cc253 const auto* node_view = graph_view.GetNode(node.name()); in TYPED_TEST() local
805 MutableNodeView* node_view = graph_view->GetNode(node_name); in CompareGraphViewWithGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc549 inline bool HasControlFaninOrFanout(const utils::MutableNodeView& node_view) { in HasControlFaninOrFanout()
555 inline bool HasAtMostOneFanoutAtPort0(const utils::MutableNodeView& node_view) { in HasAtMostOneFanoutAtPort0()
562 const utils::MutableNodeView& node_view) { in HasAtMostOneDataFanoutAtPort0()
578 const utils::MutableNodeView& node_view, in IsBiasSemanticAdd()
661 const auto* node_view = ctx.graph_view.GetNode(node_index); in FindContractionWithBias() local
704 const auto* node_view = ctx.graph_view.GetNode(node_index); in FindContractionWithBiasAndActivation() local
755 const auto* node_view = ctx.graph_view.GetNode(node_index); in FindConvWithSqueezeAndBias() local
813 const auto* node_view = ctx.graph_view.GetNode(node_index); in FindConv2DWithBatchNorm() local
867 const auto* node_view = ctx.graph_view.GetNode(node_index); in FindConv2DWithBatchNormAndActivation() local
943 const auto* node_view = ctx.graph_view.GetNode(node_index); in FindPadWithConv3D() local
[all …]
H A Dimplementation_selector.cc147 Status UpdateNodeDef(utils::MutableNodeView* node_view, const string& funcName, in UpdateNodeDef()
346 auto* node_view = graph_view.GetNode(k); in SelectDeviceIndex() local
H A Dgeneric_layout_optimizer.cc131 auto* node_view = context->graph_view->GetNode(i); in ExpandLayoutSensitiveOp() local
153 auto* node_view = context->graph_view->GetNode(i); in ExpandLayoutAgnosticOp() local