Searched refs:nodes_to_preserve_ (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 64 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in SafeToRemoveIdentity()
106 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in SafeToConvertToNoOp()
282 nodes_to_preserve_.find(node_name) == nodes_to_preserve_.end()) { in OptimizeNode()
449 nodes_to_preserve_.find(node_name) == nodes_to_preserve_.end()) { in OptimizeNode()
745 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
|
| D | common_subgraph_elimination.cc | 152 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end()) { in CanDedup()
274 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
|
| D | layout_optimizer.h | 54 std::unordered_set<string> nodes_to_preserve_; variable
|
| D | common_subgraph_elimination.h | 67 std::unordered_set<string> nodes_to_preserve_; variable
|
| D | dependency_optimizer.h | 79 std::unordered_set<string> nodes_to_preserve_; variable
|
| D | arithmetic_optimizer.h | 135 std::unordered_set<string> nodes_to_preserve_; variable
|
| D | scoped_allocator_optimizer.h | 114 std::unordered_set<string> nodes_to_preserve_; variable
|
| D | layout_optimizer.cc | 411 nodes_to_preserve_(nodes_to_preserve), in GraphProcessor()
472 const std::unordered_set<string>& nodes_to_preserve_; member in tensorflow::grappler::__anon5a8ea6d90111::GraphProcessor
580 return nodes_to_preserve_.find(node_->name()) != nodes_to_preserve_.end(); in MustPreserve()
2068 virtual_placer_, nodes_to_preserve_, in Expand()
2116 virtual_placer_, nodes_to_preserve_, in Expand()
2247 config, nodes_to_preserve_, output, in Tune()
2272 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
|
| D | constant_folding.cc | 1045 if (nodes_to_preserve_.find(node.name()) != nodes_to_preserve_.end() && in MaybeFoldable()
1598 nodes_to_preserve_.find(node->name()) == nodes_to_preserve_.end()) { in FoldNode()
1661 nodes_to_preserve_.find(node->name()) != nodes_to_preserve_.end()) { in FoldGraph()
3119 nodes_to_preserve_.find(ctx->op_child->name()) != in PrepareConstantPushDown()
3120 nodes_to_preserve_.end() || in PrepareConstantPushDown()
3477 nodes_to_preserve_.find(conv_node->name()) != nodes_to_preserve_.end()) { in MulConvPushDown()
3789 nodes_to_preserve_.find(node->name()) != nodes_to_preserve_.end() || in MergeConcat()
3981 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
|
| D | constant_folding.h | 336 std::unordered_set<string> nodes_to_preserve_; variable
|
| D | auto_mixed_precision.cc | 933 nodes_to_preserve_(nodes_to_preserve), in AutoMixedPrecisionImpl()
999 std::unordered_set<string> nodes_to_preserve_; member in tensorflow::grappler::__anonab6217530111::AutoMixedPrecisionImpl
1115 return nodes_to_preserve_.count(node.name()); in MustPreserve()
|
| D | scoped_allocator_optimizer.cc | 917 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
|
| D | arithmetic_optimizer.cc | 3671 const GraphOptimizerContext ctx(&nodes_to_preserve_, optimized_graph_, in SimplifyArithmeticOps()
3791 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
|