Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dloop_optimizer.cc457 const std::unordered_set<string>& nodes_to_preserve, int stack_node_idx) { in GetStackPushNodesToConvert()
518 Status RemoveStackOps(const std::unordered_set<string>& nodes_to_preserve, in RemoveStackOps()
762 const std::unordered_set<string>& nodes_to_preserve, NodeMap& node_map, in RemoveDeadBranches()
H A Dgraph_optimizer_stage.h61 const std::unordered_set<string>* nodes_to_preserve; member
H A Dgeneric_layout_optimizer_transposer.h75 absl::flat_hash_set<string> nodes_to_preserve; member
H A Dmodel_pruner.cc423 const std::unordered_set<string> nodes_to_preserve = item.NodesToPreserve(); in Optimize() local
H A Darithmetic_optimizer.cc141 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfValuePreservingChain()
152 const std::unordered_set<string>& nodes_to_preserve) { in GetTailOfIdempotentChain()
H A Dauto_mixed_precision.cc1036 const std::unordered_set<string>& nodes_to_preserve, in AutoMixedPrecisionImpl()
H A Dgeneric_layout_optimizer_transposer.cc240 const auto& nodes_to_preserve = item.NodesToPreserve(); in InitializeTransposeContext() local
H A Dremapper.cc97 std::unordered_set<string> nodes_to_preserve; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dpattern_utils.cc170 const std::unordered_set<string>& nodes_to_preserve, in GetMatchedNodes()
H A Dpattern_utils.h206 const std::unordered_set<string>& nodes_to_preserve) { in IsSafeNodesToRemove()
H A Dpattern_utils_test.cc257 std::unordered_set<string> nodes_to_preserve = {"foo"}; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dshuffle_and_repeat_fusion.cc197 const auto nodes_to_preserve = item.NodesToPreserve(); in OptimizeAndCollectStats() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dtrt_optimization_pass.cc209 std::vector<string> nodes_to_preserve; in Optimize() local