Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc57 return full_name_with_scope.find(kAttrRecompute) == 0; in WithRecomputedScope()
66 return cnode->GetAttr(kAttrRecompute); in GetRecomputeCNodeAttr()
304 auto prim_recompute_attr = prim->GetAttr(kAttrRecompute); in SetRecomputedAttr()
310 cnode->AddAttr(kAttrRecompute, MakeValue(true)); in SetRecomputedAttr()
321 output_cnode->AddAttr(kAttrRecompute, MakeValue(true)); in SetRecomputedAttr()
Dcse.cc42 return full_name_with_scope.find(kAttrRecompute) == 0; in WithRecomputedScope()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dadjust_depend_for_parallel_optimizer_recompute_all_gather.cc62 … AnfAlgo::HasNodeAttr(kAttrRecompute, cnode) && AnfAlgo::GetNodeAttr<bool>(cnode, kAttrRecompute); in Run()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Drecompute_prepare.h49 real_output->AddAttr(kAttrRecompute, MakeValue(false)); in operator()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h461 constexpr auto kAttrRecompute = "recompute"; variable