Searched refs:fused_cnode (Results 1 – 2 of 2) sorted by relevance
173 auto fused_cnode = func_graph->NewCNode(fused_node_inputs); in DoFusion() local174 MS_EXCEPTION_IF_NULL(fused_cnode); in DoFusion()175 AnfAlgo::SetNodeAttr(kAttrPsKey, MakeValue(weight_names), fused_cnode); in DoFusion()176 AnfAlgo::SetNodeAttr(kAttrIndex, MakeValue(indices), fused_cnode); in DoFusion()177 AnfAlgo::SetNodeAttr(kAttrPrimitiveTarget, MakeValue(kCPUDevice), fused_cnode); in DoFusion()181 fused_cnode->set_kernel_info(kernel_info); in DoFusion()183 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, fused_cnode.get()); in DoFusion()193 fused_cnode->set_abstract(abstract_tuple); in DoFusion()198 if (!manager->Replace(node, fused_cnode)) { in DoFusion()
386 auto fused_cnode = fused_node->cast<CNodePtr>(); in CreateFusedCommunicationOp() local387 fused_cnode->AddAttr("duplicated", MakeValue(true)); in CreateFusedCommunicationOp()388 auto fused_prim = GetCNodePrimitive(fused_cnode); in CreateFusedCommunicationOp()