Searched refs:current_prim (Results 1 – 1 of 1) sorted by relevance
561 PrimitivePtr current_prim = current_value->value()->cast<PrimitivePtr>(); in StepRedistribution() local562 MS_EXCEPTION_IF_NULL(current_prim); in StepRedistribution()563 insert_node_new = ((current_prim->name() == prim::kTupleGetItem) ? node : insert_node); in StepRedistribution()2274 auto current_prim = GetValueNode<PrimitivePtr>(pre_cnode->input(0)); in FindLossCNode() local2276 if (INVALID_LOSS_OPS.find(current_prim->name()) != INVALID_LOSS_OPS.end()) { in FindLossCNode()2277 MS_LOG(INFO) << "The loss is: " << current_prim->name(); in FindLossCNode()2288 if (current_prim->name() == prim::kTupleGetItem) { in FindLossCNode()2301 if (current_prim->name() == MAKE_TUPLE) { in FindLossCNode()2308 MS_LOG(DEBUG) << "The loss name is " << current_prim->name(); in FindLossCNode()