Searched refs:next_op_name (Results 1 – 6 of 6) sorted by relevance
160 auto next_op_name = AnfAlgo::GetCNodeName(next_cnode); in MergeCastToNextOp() local161 if (next_op_name == prim::kPrimSend->name() || next_op_name == kStackPushOpName) { in MergeCastToNextOp()
58 std::string next_op_name = ""; member
1149 op_exec_info->next_op_name = op_name; in DoAutoCast()1957 op_exec_info->next_op_name, in RunOpInMs()1966 op_exec_info->next_op_name, in RunOpInMs()
337 auto next_op_name = AnfAlgo::GetNodeAttr<std::string>(kernel_node, kAttrPynativeNextOpName); in SetCastAndWeightFormat() local338 auto iter = kNextOpFormatList.find(next_op_name); in SetCastAndWeightFormat()340 MS_LOG(INFO) << "The op name " << next_op_name << "has not been set in the next op map "; in SetCastAndWeightFormat()
66 std::string next_op_name = ""; member
2163 AnfAlgo::SetNodeAttr(kAttrPynativeNextOpName, MakeValue(op_run_info.next_op_name), cnode); in ConstructSingleOpGraph()