Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc249 NodeDef* node = optimized_graph_->mutable_node(node_idx); in OptimizeNode()
311 old_input, optimized_graph_, node_map_.get()); in OptimizeNode()
461 for (int i = 0; i < optimized_graph_->node_size(); ++i) { in CleanControlInputs()
462 DedupControlInputs(optimized_graph_->mutable_node(i)); in CleanControlInputs()
469 for (int i = 0; i < optimized_graph_->node_size(); ++i) { in OptimizeDependencies()
470 const NodeDef& node = optimized_graph_->node(i); in OptimizeDependencies()
487 << optimized_graph_->node_size() << " nodes."; in OptimizeDependencies()
488 EraseNodesFromGraph(nodes_to_delete, optimized_graph_); in OptimizeDependencies()
489 node_map_.reset(new NodeMap(optimized_graph_)); in OptimizeDependencies()
528 const int num_nodes = optimized_graph_->node_size(); in TransitiveReduction()
[all …]
Dloop_optimizer.cc62 : optimized_graph_(optimized_graph) {} in LoopInvariantNodeMotionOptimizer()
75 GraphDef* optimized_graph_; // Not owned. member in tensorflow::grappler::__anon3d4744950111::LoopInvariantNodeMotionOptimizer
135 const_node = optimized_graph_->add_node(); in HandleConst()
181 switch_output, optimized_graph_, node_map_.get()); in HandleConst()
228 NodeDef* new_enter = optimized_graph_->add_node(); in HandleInvariantNode()
382 node_map_.reset(new NodeMap(optimized_graph_)); in Optimize()
385 TF_RETURN_IF_ERROR(frame_view.InferFromGraph(*optimized_graph_)); in Optimize()
390 for (const NodeDef& node : optimized_graph_->node()) { in Optimize()
Ddependency_optimizer.h82 GraphDef* optimized_graph_; // Not owned. variable
Darithmetic_optimizer.h138 GraphDef* optimized_graph_ = nullptr; // Not owned. variable
Darithmetic_optimizer.cc3666 nodes_to_simplify.Reserve(optimized_graph_->node_size()); in SimplifyArithmeticOps()
3667 for (int i = 0; i < optimized_graph_->node_size(); ++i) { in SimplifyArithmeticOps()
3668 nodes_to_simplify.PushBack(optimized_graph_->mutable_node(i)); in SimplifyArithmeticOps()
3671 const GraphOptimizerContext ctx(&nodes_to_preserve_, optimized_graph_, in SimplifyArithmeticOps()
3794 optimized_graph_ = &optimized_item.graph; in Optimize()
3796 node_map_.reset(new NodeMap(optimized_graph_)); in Optimize()
3808 TF_RETURN_IF_ERROR(TopologicalSort(optimized_graph_)); in Optimize()
3825 optimized_graph->Swap(optimized_graph_); in Optimize()