Home
last modified time | relevance | path

Searched refs:fanin_node (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer.cc239 auto* fanin_node = regular_fanin_0.node_view(); in EraseCancellableNodes() local
242 if (fanin_node->node_index() < original_num_nodes) { in EraseCancellableNodes()
245 if (!IsCancellableNodePair(*node, *fanin_node)) { in EraseCancellableNodes()
248 const auto& fanin_to_forward = fanin_node->GetRegularFanin(0); in EraseCancellableNodes()
260 mutation->RemoveNode(fanin_node); in EraseCancellableNodes()
261 if (fanin_node->NumRegularFanins() > 1) { in EraseCancellableNodes()
262 mutation->RemoveNode(fanin_node->GetRegularFanin(1).node_view()); in EraseCancellableNodes()
Dgeneric_layout_optimizer_transposer.cc850 auto* fanin_node = fanin.node_view(); in TransposeNode() local
851 const auto* output_shape_attr = fanin_node->GetAttr(kAttrOutputShape); in TransposeNode()
853 VLOG(3) << "Cannot compute the shape of " << fanin_node->GetName() in TransposeNode()
859 VLOG(3) << fanin_node->GetName() << " is not a vector."; in TransposeNode()
1106 auto* fanin_node = fanin.node_view(); in IsAfterDstToSrcTransform() local
1107 queue.push_back(fanin_node); in IsAfterDstToSrcTransform()
1108 visited_nodes.insert(fanin_node); in IsAfterDstToSrcTransform()
1124 auto* fanin_node = fanin.node_view(); in IsAfterDstToSrcTransform() local
1125 if (visited_nodes.insert(fanin_node).second) { in IsAfterDstToSrcTransform()
1126 queue.push_back(fanin_node); in IsAfterDstToSrcTransform()
[all …]
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc905 NodeDef* fanin_node = GetNode(fanin.node()); in AddRegularFanin() local
906 TF_RETURN_IF_ERROR(CheckNodeExists(fanin.node(), fanin_node, error_status)); in AddRegularFanin()
908 AddFaninInternal(node, {fanin_node, fanin.index()}); in AddRegularFanin()
929 NodeDef* fanin_node = GetNode(fanin.node()); in AddRegularFaninByPort() local
930 TF_RETURN_IF_ERROR(CheckNodeExists(fanin.node(), fanin_node, error_status)); in AddRegularFaninByPort()
944 OutputPort fanin_port(fanin_node, fanin.index()); in AddRegularFaninByPort()
949 if (CanDedupControlWithRegularInput(*this, *fanin_node)) { in AddRegularFaninByPort()
950 RemoveControllingFaninInternal(node, fanin_node); in AddRegularFaninByPort()
1027 NodeDef* fanin_node = GetNode(fanin.node()); in AddControllingFanin() local
1028 TF_RETURN_IF_ERROR(CheckNodeExists(fanin.node(), fanin_node, error_status)); in AddControllingFanin()
[all …]
Dmutable_graph_view.h311 bool RemoveControllingFaninInternal(NodeDef* node, NodeDef* fanin_node);