Home
last modified time | relevance | path

Searched defs:nodes_to_preserve (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc239 std::vector<string> nodes_to_preserve; in Optimize() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc458 const std::unordered_set<string>& nodes_to_preserve, int stack_node_idx) { in GetStackPushNodesToConvert()
519 Status RemoveStackOps(const std::unordered_set<string>& nodes_to_preserve, in RemoveStackOps()
737 const std::unordered_set<string>& nodes_to_preserve, in RemoveDeadBranches()
Dgraph_optimizer_stage.h61 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
Dlayout_optimizer.cc399 const std::unordered_set<string>& nodes_to_preserve, in GraphProcessor()
487 const std::unordered_set<string>& nodes_to_preserve; member
1965 const std::unordered_set<string>& nodes_to_preserve, GraphDef* graph, in DataLayoutOptimizer()
Dremapper.cc56 std::unordered_set<string> nodes_to_preserve; member
Darithmetic_optimizer.cc136 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfValuePreservingChain()
147 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfIdempotentChain()