Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc64 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in SafeToRemoveIdentity()
96 nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in SafeToConvertToNoOp()
260 nodes_to_preserve_.find(node_name) == nodes_to_preserve_.end()) { in OptimizeNode()
426 nodes_to_preserve_.find(node_name) == nodes_to_preserve_.end()) { in OptimizeNode()
679 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dlayout_optimizer.h52 std::unordered_set<string> nodes_to_preserve_; variable
Ddependency_optimizer.h76 std::unordered_set<string> nodes_to_preserve_; variable
Darithmetic_optimizer.h131 std::unordered_set<string> nodes_to_preserve_; variable
Dscoped_allocator_optimizer.h100 std::unordered_set<string> nodes_to_preserve_; variable
Dlayout_optimizer.cc403 nodes_to_preserve_(nodes_to_preserve), in GraphProcessor()
464 const std::unordered_set<string>& nodes_to_preserve_; member in tensorflow::grappler::__anonbd4a17fb0111::GraphProcessor
568 return nodes_to_preserve_.find(node_->name()) != nodes_to_preserve_.end(); in MustPreserve()
2010 virtual_placer_, nodes_to_preserve_, in Expand()
2062 virtual_placer_, nodes_to_preserve_, in Expand()
2192 config, nodes_to_preserve_, output, in Tune()
2211 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dconstant_folding.cc862 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end() && in IsFoldable()
1426 nodes_to_preserve_.find(node->name()) == nodes_to_preserve_.end()) { in FoldNode()
1490 nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in FoldGraph()
2851 nodes_to_preserve_.find(op_child_node->name()) != in ConstantPushDown()
2852 nodes_to_preserve_.end() || in ConstantPushDown()
2929 nodes_to_preserve_.find(conv_node->name()) != nodes_to_preserve_.end()) { in MulConvPushDown()
3266 nodes_to_preserve_.find(node->name()) != nodes_to_preserve_.end() || in MergeConcat()
3428 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dconstant_folding.h250 std::unordered_set<string> nodes_to_preserve_; variable
Darithmetic_optimizer.cc3406 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in CanDedup()
3534 const GraphOptimizerContext ctx(&nodes_to_preserve_, optimized_graph_, in SimplifyArithmeticOps()
3653 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dscoped_allocator_optimizer.cc681 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()