Home
last modified time | relevance | path

Searched refs:fanin_node_view (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc144 NodeView& fanin_node_view = nodes_[fanin_node_index]; in CheckAndAddFaninsInternal() local
147 fanin_node_view.controlled_fanouts_.emplace_back(this, node_index, in CheckAndAddFaninsInternal()
151 node_view->fanins_set_.emplace(fanin_node_view.node(), in CheckAndAddFaninsInternal()
156 fanin_node_view.regular_fanouts_by_port_.size(); in CheckAndAddFaninsInternal()
158 fanin_node_view.regular_fanouts_by_port_.resize(fanin_id.index() + 1); in CheckAndAddFaninsInternal()
160 fanin_node_view.regular_fanouts_by_port_[fanin_id.index()].emplace_back( in CheckAndAddFaninsInternal()
162 ++fanin_node_view.num_regular_fanouts_; in CheckAndAddFaninsInternal()
165 node_view->fanins_set_.emplace(fanin_node_view.node(), fanin_id.index()); in CheckAndAddFaninsInternal()
557 MutableNodeView& fanin_node_view = nodes_[fanin_node_index]; in AddFaninsInternal() local
561 fanin_node_view.controlled_fanouts_.emplace_back( in AddFaninsInternal()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc382 auto* fanin_node_view = fanin_port.node_view(); in UpdateFaninEdgesWithOp() local
390 dst_port, fanin_node_view, dst_node)); in UpdateFaninEdgesWithOp()
607 const auto* fanin_node_view = regular_fanin.node_view(); in IsFaninPortDimsNIfConst() local
608 if (!IsConstant(*fanin_node_view->node())) { in IsFaninPortDimsNIfConst()
612 const auto* value_attr = fanin_node_view->GetAttr(kAttrValue); in IsFaninPortDimsNIfConst()