Home
last modified time | relevance | path

Searched defs:current_prim (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dstage_compute.cc116 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in HasRecompute() local
230 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in GetSeqLengthAndAttentionHeads() local
293 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in GetPerBatch() local
360 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in GetFPFromNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2439 auto current_prim = GetValueNode<PrimitivePtr>(pre_cnode->input(0)); in FindLossCNode() local