Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmemory_optimizer.cc481 std::unordered_set<string> cheap_to_recompute_ops = in RecomputationRewritingPass() local
485 [&cheap_to_recompute_ops, &feeds, &is_target](const NodeDef& node) { in RecomputationRewritingPass()
487 (cheap_to_recompute_ops.count(node.op()) > 0 || in RecomputationRewritingPass()