| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | shuffle_and_repeat_fusion.cc | 194 const auto nodes_to_preserve = item.NodesToPreserve(); in OptimizeAndCollectStats() local
195 if (nodes_to_preserve.find(shuffle_node.name()) == in OptimizeAndCollectStats()
196 nodes_to_preserve.end() && in OptimizeAndCollectStats()
197 nodes_to_preserve.find(repeat_node.name()) == nodes_to_preserve.end()) { in OptimizeAndCollectStats()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | graph_optimizer_stage.h | 49 GraphOptimizerContext(const std::unordered_set<string>* nodes_to_preserve, in GraphOptimizerContext()
54 : nodes_to_preserve(nodes_to_preserve), in GraphOptimizerContext()
61 const std::unordered_set<string>* nodes_to_preserve; member
|
| D | model_pruner.cc | 423 const std::unordered_set<string> nodes_to_preserve = item.NodesToPreserve(); in Optimize() local
430 if (!nodes_to_preserve.empty()) { in Optimize()
434 std::vector<string> terminal_nodes(nodes_to_preserve.begin(), in Optimize()
435 nodes_to_preserve.end()); in Optimize()
473 if (nodes_to_preserve.find(node->name()) != nodes_to_preserve.end()) { in Optimize()
498 if (nodes_to_delete.empty() && nodes_to_preserve.empty()) { in Optimize()
|
| D | loop_optimizer.cc | 457 const std::unordered_set<string>& nodes_to_preserve, int stack_node_idx) { in GetStackPushNodesToConvert() argument
493 if (nodes_to_preserve.find(stack_node->name()) == in GetStackPushNodesToConvert()
494 nodes_to_preserve.end()) { in GetStackPushNodesToConvert()
506 nodes_to_preserve.find(fanout_node.name()) != in GetStackPushNodesToConvert()
507 nodes_to_preserve.end())) { in GetStackPushNodesToConvert()
518 Status RemoveStackOps(const std::unordered_set<string>& nodes_to_preserve, in RemoveStackOps() argument
527 graph_view, nodes_to_preserve, node_idx)) { in RemoveStackOps()
762 const std::unordered_set<string>& nodes_to_preserve, in RemoveDeadBranches() argument
779 if (nodes_to_preserve.find(node.name()) != nodes_to_preserve.end()) { in RemoveDeadBranches()
811 if (nodes_to_preserve.find(dead.node->name()) != in RemoveDeadBranches()
[all …]
|
| D | loop_optimizer.h | 66 Status RemoveDeadBranches(const std::unordered_set<string>& nodes_to_preserve,
|
| D | layout_optimizer.cc | 407 const std::unordered_set<string>& nodes_to_preserve, in GraphProcessor() argument
411 nodes_to_preserve_(nodes_to_preserve), in GraphProcessor()
481 const std::unordered_set<string>& nodes_to_preserve, in OptimizeContext()
489 nodes_to_preserve(nodes_to_preserve), in OptimizeContext()
497 const std::unordered_set<string>& nodes_to_preserve; member
506 opt_cxt.nodes_to_preserve, opt_cxt.graph, in NodeProcessor()
2032 const std::unordered_set<string>& nodes_to_preserve, GraphDef* graph, in DataLayoutOptimizer() argument
2034 : GraphProcessor(graph_properties, virtual_placer, nodes_to_preserve, in DataLayoutOptimizer()
|
| D | arithmetic_optimizer.cc | 138 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfValuePreservingChain() argument
140 return nodes_to_preserve.find(node.name()) == nodes_to_preserve.end() && in GetTailOfValuePreservingChain()
149 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfIdempotentChain() argument
151 return nodes_to_preserve.find(node.name()) == nodes_to_preserve.end() && in GetTailOfIdempotentChain()
254 return ctx().nodes_to_preserve->find(node.name()) != in IsInPreserveSet()
255 ctx().nodes_to_preserve->end(); in IsInPreserveSet()
441 return ctx().nodes_to_preserve->find(node.name()) != in IsInPreserveSet()
442 ctx().nodes_to_preserve->end(); in IsInPreserveSet()
1127 *ctx().nodes_to_preserve); in TrySimplify()
1244 *ctx().nodes_to_preserve); in TrySimplify()
[all …]
|
| D | generic_layout_optimizer_transposer.cc | 238 const auto& nodes_to_preserve = item.NodesToPreserve(); in InitializeTransposeContext() local
239 context->nodes_to_preserve = absl::flat_hash_set<string>( in InitializeTransposeContext()
240 nodes_to_preserve.begin(), nodes_to_preserve.end()); in InitializeTransposeContext()
284 !context.nodes_to_preserve.contains(node_def->name()) && in ShouldProcess()
647 return !context.nodes_to_preserve.contains(node.GetName()) && in CanProcessNode()
|
| D | remapper.cc | 80 : nodes_to_preserve(item->NodesToPreserve()), in RemapperContext()
85 std::unordered_set<string> nodes_to_preserve; member
211 return ctx.nodes_to_preserve.count(node->name()) > 0; in IsInPreserveSet()
|
| D | generic_layout_optimizer_transposer.h | 69 absl::flat_hash_set<string> nodes_to_preserve; member
|
| D | auto_mixed_precision.cc | 929 const std::unordered_set<string>& nodes_to_preserve, in AutoMixedPrecisionImpl() argument
933 nodes_to_preserve_(nodes_to_preserve), in AutoMixedPrecisionImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.cc | 214 std::vector<string> nodes_to_preserve; in Optimize() local
228 nodes_to_preserve.push_back(s); in Optimize()
233 cp.output_names = &nodes_to_preserve; in Optimize()
|