Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2846 NodeDef* op_child_node = left_child_is_constant ? right_child : left_child; in ConstantPushDown() local
2850 if (op_child_node->input_size() < 2 || in ConstantPushDown()
2851 nodes_to_preserve_.find(op_child_node->name()) != in ConstantPushDown()
2853 NumNonControlOutputs(*op_child_node, *node_map_) > 1) { in ConstantPushDown()
2858 NodeDef* left_leaf = node_map_->GetNode(op_child_node->input(0)); in ConstantPushDown()
2859 NodeDef* right_leaf = node_map_->GetNode(op_child_node->input(1)); in ConstantPushDown()
2868 const auto& child_output = node_map_->GetOutputs(op_child_node->name()); in ConstantPushDown()
2874 CHECK(MaybeRemoveControlInput(op_child_node->name(), const_child_node, in ConstantPushDown()
2876 string other_leaf_input = left_leaf_is_constant ? op_child_node->input(0) in ConstantPushDown()
2877 : op_child_node->input(1); in ConstantPushDown()
[all …]