Searched refs:nodes_to_simplify (Results 1 – 4 of 4) sorted by relevance
225 SetVector<int>* nodes_to_simplify, in OptimizeNode() argument253 nodes_to_simplify->PushBack(node_to_idx_[fanout]); in OptimizeNode()294 nodes_to_simplify->PushBack(node_to_idx_[old_input_node]); in OptimizeNode()299 nodes_to_simplify->PushBack(node_to_idx_[node]); in OptimizeNode()411 nodes_to_simplify->PushBack(node_to_idx_[input]); in OptimizeNode()420 nodes_to_simplify->PushBack(node_to_idx_[consumer]); in OptimizeNode()444 SetVector<int> nodes_to_simplify; in OptimizeDependencies() local450 nodes_to_simplify.PushBack(i); in OptimizeDependencies()453 while (!nodes_to_simplify.Empty()) { in OptimizeDependencies()454 int node_to_simplify = nodes_to_simplify.PopBack(); in OptimizeDependencies()[all …]
64 void OptimizeNode(int node_idx, SetVector<int>* nodes_to_simplify,
125 SetVector<NodeDef*>* nodes_to_simplify);
194 explicit ArithmeticOptimizerContext(SetVector<NodeDef*>* nodes_to_simplify) in ArithmeticOptimizerContext()195 : nodes_to_simplify(nodes_to_simplify) {} in ArithmeticOptimizerContext()196 SetVector<NodeDef*>* nodes_to_simplify; member215 ctx_ext_.nodes_to_simplify->PushBack(node); in AddToOptimizationQueue()3528 SetVector<NodeDef*> nodes_to_simplify; in SimplifyArithmeticOps() local3529 nodes_to_simplify.Reserve(optimized_graph_->node_size()); in SimplifyArithmeticOps()3531 nodes_to_simplify.PushBack(optimized_graph_->mutable_node(i)); in SimplifyArithmeticOps()3537 const ArithmeticOptimizerContext ctx_ext(&nodes_to_simplify); in SimplifyArithmeticOps()3600 while (!nodes_to_simplify.Empty()) { in SimplifyArithmeticOps()3602 NodeDef* node = nodes_to_simplify.PopBack(); in SimplifyArithmeticOps()[all …]