Searched refs:nodes_to_simplify (Results 1 – 4 of 4) sorted by relevance
247 SetVector<int>* nodes_to_simplify, in OptimizeNode() argument275 nodes_to_simplify->PushBack(node_to_idx_[fanout]); in OptimizeNode()316 nodes_to_simplify->PushBack(node_to_idx_[old_input_node]); in OptimizeNode()322 nodes_to_simplify->PushBack(node_to_idx_[node]); in OptimizeNode()435 nodes_to_simplify->PushBack(node_to_idx_[input]); in OptimizeNode()443 nodes_to_simplify->PushBack(node_to_idx_[consumer]); in OptimizeNode()467 SetVector<int> nodes_to_simplify; in OptimizeDependencies() local473 nodes_to_simplify.PushBack(i); in OptimizeDependencies()476 while (!nodes_to_simplify.Empty()) { in OptimizeDependencies()477 int node_to_simplify = nodes_to_simplify.PopBack(); in OptimizeDependencies()[all …]
66 void OptimizeNode(int node_idx, SetVector<int>* nodes_to_simplify,
129 SetVector<NodeDef*>* nodes_to_simplify);
203 explicit ArithmeticOptimizerContext(SetVector<NodeDef*>* nodes_to_simplify) in ArithmeticOptimizerContext()204 : nodes_to_simplify(nodes_to_simplify) {} in ArithmeticOptimizerContext()205 SetVector<NodeDef*>* nodes_to_simplify; member224 ctx_ext_.nodes_to_simplify->PushBack(node); in AddToOptimizationQueue()3665 SetVector<NodeDef*> nodes_to_simplify; in SimplifyArithmeticOps() local3666 nodes_to_simplify.Reserve(optimized_graph_->node_size()); in SimplifyArithmeticOps()3668 nodes_to_simplify.PushBack(optimized_graph_->mutable_node(i)); in SimplifyArithmeticOps()3674 const ArithmeticOptimizerContext ctx_ext(&nodes_to_simplify); in SimplifyArithmeticOps()3742 while (!nodes_to_simplify.Empty()) { in SimplifyArithmeticOps()3744 NodeDef* node = nodes_to_simplify.PopBack(); in SimplifyArithmeticOps()[all …]