Searched defs:current_prim (Results 1 – 2 of 2) sorted by relevance
116 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in HasRecompute() local230 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in GetSeqLengthAndAttentionHeads() local293 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in GetPerBatch() local360 auto current_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in GetFPFromNodes() local
2439 auto current_prim = GetValueNode<PrimitivePtr>(pre_cnode->input(0)); in FindLossCNode() local