Home
last modified time | relevance | path

Searched refs:nodes_to_preserve (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc458 const std::unordered_set<string>& nodes_to_preserve, int stack_node_idx) { in GetStackPushNodesToConvert() argument
494 if (nodes_to_preserve.find(stack_node->name()) == in GetStackPushNodesToConvert()
495 nodes_to_preserve.end()) { in GetStackPushNodesToConvert()
507 nodes_to_preserve.find(fanout_node.name()) != in GetStackPushNodesToConvert()
508 nodes_to_preserve.end())) { in GetStackPushNodesToConvert()
519 Status RemoveStackOps(const std::unordered_set<string>& nodes_to_preserve, in RemoveStackOps() argument
528 graph_view, nodes_to_preserve, node_idx)) { in RemoveStackOps()
737 const std::unordered_set<string>& nodes_to_preserve, in RemoveDeadBranches() argument
751 if (nodes_to_preserve.find(node.name()) != nodes_to_preserve.end()) { in RemoveDeadBranches()
783 if (nodes_to_preserve.find(dead.node->name()) != in RemoveDeadBranches()
[all …]
Dgraph_optimizer_stage.h49 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
Dmodel_pruner.cc425 const std::unordered_set<string> nodes_to_preserve = item.NodesToPreserve(); in Optimize() local
431 if (!nodes_to_preserve.empty()) { in Optimize()
432 std::vector<string> terminal_nodes(nodes_to_preserve.begin(), in Optimize()
433 nodes_to_preserve.end()); in Optimize()
465 if (nodes_to_preserve.find(node.name()) != nodes_to_preserve.end()) { in Optimize()
Dloop_optimizer.h62 Status RemoveDeadBranches(const std::unordered_set<string>& nodes_to_preserve,
Dlayout_optimizer.cc399 const std::unordered_set<string>& nodes_to_preserve, in GraphProcessor() argument
403 nodes_to_preserve_(nodes_to_preserve), in GraphProcessor()
473 const std::unordered_set<string>& nodes_to_preserve, in OptimizeContext()
480 nodes_to_preserve(nodes_to_preserve), in OptimizeContext()
487 const std::unordered_set<string>& nodes_to_preserve; member
495 opt_cxt.nodes_to_preserve, opt_cxt.graph, in NodeProcessor()
1965 const std::unordered_set<string>& nodes_to_preserve, GraphDef* graph, in DataLayoutOptimizer() argument
1967 : GraphProcessor(graph_properties, virtual_placer, nodes_to_preserve, in DataLayoutOptimizer()
Darithmetic_optimizer.cc136 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfValuePreservingChain() argument
138 return nodes_to_preserve.find(node.name()) == nodes_to_preserve.end() && in GetTailOfValuePreservingChain()
147 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfIdempotentChain() argument
149 return nodes_to_preserve.find(node.name()) == nodes_to_preserve.end() && in GetTailOfIdempotentChain()
237 return ctx().nodes_to_preserve->find(node.name()) != in IsInPreserveSet()
238 ctx().nodes_to_preserve->end(); in IsInPreserveSet()
416 return ctx().nodes_to_preserve->find(node.name()) != in IsInPreserveSet()
417 ctx().nodes_to_preserve->end(); in IsInPreserveSet()
1099 *ctx().nodes_to_preserve); in TrySimplify()
1204 *ctx().nodes_to_preserve); in TrySimplify()
[all …]
Dremapper.cc51 : nodes_to_preserve(item.NodesToPreserve()), in RemapperContext()
56 std::unordered_set<string> nodes_to_preserve; member
135 return ctx.nodes_to_preserve.count(node->name()) > 0; in IsInPreserveSet()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc239 std::vector<string> nodes_to_preserve; in Optimize() local
253 nodes_to_preserve.push_back(s); in Optimize()
256 cp.output_names = &nodes_to_preserve; in Optimize()