Searched refs:recomputation_targets_name_scope (Results 1 – 2 of 2) sorted by relevance
38 const string& recomputation_targets_name_scope = "gradients/")40 recomputation_targets_name_scope_(recomputation_targets_name_scope) {} in optimization_level_()
424 const string& recomputation_targets_name_scope, in RecomputationRewritingPass() argument449 [&recomputation_targets_name_scope](const NodeDef& node) { in RecomputationRewritingPass()455 return node.name().find(recomputation_targets_name_scope) == 0 || in RecomputationRewritingPass()456 node.name().find("/" + recomputation_targets_name_scope) != -1; in RecomputationRewritingPass()