Home
last modified time | relevance | path

Searched refs:IsGraphKernel (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Doptimize_assign.cc149 if (!AnfAlgo::IsGraphKernel(n)) continue; in RepalceOutputByParameter()
196 if (!AnfAlgo::IsGraphKernel(n)) continue; in RepalceAssignByInplaceAssign()
Dcast_matmul_fusion.cc101 if (!AnfAlgo::IsGraphKernel(node)) continue; in Run()
Daxis_normalizer.cc85 if (AnfAlgo::IsGraphKernel(node)) { in Run()
Dgraph_kernel_cluster.cc113 if (AnfAlgo::IsGraphKernel(node)) { in IsClusterableOp()
146 } else if (AnfAlgo::IsGraphKernel(node)) { in Cluster()
409 …if (AnfAlgo::IsGraphKernel(node) || IsPrimitiveCNode(node, prim::kPrimAssign) || !IsClusterableOp(… in Process()
Dshape_ops_splitter.cc120 if (AnfAlgo::IsGraphKernel(anf_node)) { in Run()
Draise_reduction_precision.cc115 if (AnfAlgo::IsGraphKernel(node)) { in Run()
Dadd_stitch_atomic_clean_gpu.cc109 if (!AnfAlgo::IsGraphKernel(anf_node)) return false; in IsStitchWithAtomic()
Dgraph_kernel_helper.cc268 if (!AnfAlgo::IsGraphKernel(n)) { in MixedNodesTransToGraph()
348 if (outputs.size() == 1 && AnfAlgo::IsGraphKernel(outputs[0])) { in SetNewKernelInfo()
479 } else if (AnfAlgo::IsGraphKernel(fuse_node)) { in FuseNodesToSubGraph()
501 bool has_graph_kernel = std::any_of(nodes.begin(), nodes.end(), AnfAlgo::IsGraphKernel); in AnfToJsonDesc()
531 if (nodes.size() == 1 && AnfAlgo::IsGraphKernel(nodes[0])) { in AnfToJsonDesc()
Dparallel_fusion.cc264 return session::AnfRuntimeAlgorithm::IsGraphKernel(node) || IsOneOf(node, whiteable_ops); in WhiteOpsFilter()
351 …buf << (AnfAlgo::IsGraphKernel(cnode) ? "[graph]" : "[primitive]") << cnode->fullname_with_scope()… in DumpNode()
642 if (!AnfAlgo::IsGraphKernel(fuse_nodes[i])) { in SetFusedParallelOpAttrToReturnNode()
Deliminate_redundant_output.cc94 return AnfAlgo::IsGraphKernel(node) && in FindGraphKernelsWithMultiOutput()
Ddecrease_compute_precision.cc253 if (AnfAlgo::IsGraphKernel(node)) { in Run()
Ddecrease_transfer_precision.cc66 bool is_gk = AnfAlgo::IsGraphKernel(node); in IsCandidateNode()
Dgraph_kernel_expander.cc225 …if (node == nullptr || AnfAlgo::IsGraphKernel(node) || IsKeepBasicNode(node) || !AnfAlgo::IsRealKe… in DoExpand()
Dreorder_ops.cc359 if (AnfAlgo::IsGraphKernel(node)) { in Run()
Dinsert_pad.cc282 if (!AnfAlgo::IsGraphKernel(node)) continue; in Run()
Dtransform_op_optimizer.cc431 if (!AnfAlgo::IsGraphKernel(node)) continue; in Run()
Dadd_atomic_clean.cc167 return (AnfAlgo::IsGraphKernel(node) && CanActivateAtomicAdd(node)); in Check()
Darithmetic_simplify.cc640 if (AnfAlgo::IsGraphKernel(node)) { in Run()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dnode_pass_extends.cc65 if (IsGraphKernel(new_node)) { in Run()
Dgllo_utils.h63 bool IsGraphKernel(const AnfNodePtr &node);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dnode_pass.cc102 if (AnfAlgo::IsGraphKernel(new_node)) { in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc499 if (AnfAlgo::IsGraphKernel(kernel_node)) { in SelectKernelInfo()
551 if (AnfAlgo::IsGraphKernel(kernel_node)) { in SetKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h240 static bool IsGraphKernel(const AnfNodePtr &node);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_build.cc535 if (AnfAlgo::IsGraphKernel(cnode)) { in AkgKernelParallelBuild()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc400 if (AnfAlgo::IsGraphKernel(kernel_node)) { in SetKernelInfo()

12