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.cc257 NodeDef* node = optimized_graph_->mutable_node(node_idx); in OptimizeNode()
319 old_input, optimized_graph_, node_map_.get()); in OptimizeNode()
469 for (int i = 0; i < optimized_graph_->node_size(); ++i) { in CleanControlInputs()
470 DedupControlInputs(optimized_graph_->mutable_node(i)); in CleanControlInputs()
477 for (int i = 0; i < optimized_graph_->node_size(); ++i) { in OptimizeDependencies()
478 const NodeDef& node = optimized_graph_->node(i); in OptimizeDependencies()
495 << optimized_graph_->node_size() << " nodes."; in OptimizeDependencies()
496 EraseNodesFromGraph(nodes_to_delete, optimized_graph_); in OptimizeDependencies()
497 node_map_.reset(new NodeMap(optimized_graph_)); in OptimizeDependencies()
536 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::__anon235c56440111::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.h79 GraphDef* optimized_graph_; // Not owned. variable
Darithmetic_optimizer.h139 GraphDef* optimized_graph_ = nullptr; // Not owned. variable
Darithmetic_optimizer.cc4291 nodes_to_simplify.Reserve(optimized_graph_->node_size()); in SimplifyArithmeticOps()
4292 for (int i = 0; i < optimized_graph_->node_size(); ++i) { in SimplifyArithmeticOps()
4293 nodes_to_simplify.PushBack(optimized_graph_->mutable_node(i)); in SimplifyArithmeticOps()
4296 const GraphOptimizerContext ctx(&nodes_to_preserve_, optimized_graph_, in SimplifyArithmeticOps()
4427 optimized_graph_ = &optimized_item.graph; in Optimize()
4429 node_map_.reset(new NodeMap(optimized_graph_)); in Optimize()
4441 TF_RETURN_IF_ERROR(TopologicalSort(optimized_graph_)); in Optimize()
4457 *optimized_graph = std::move(*optimized_graph_); in Optimize()