Home
last modified time | relevance | path

Searched refs:GetValidKernelNodes (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.h96 void GetValidKernelNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *node_list);
97 void GetValidKernelNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *node_list,
Dcommon_utils.cc674 void GetValidKernelNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *node_list) { in GetValidKernelNodes() function
690 void GetValidKernelNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *node_list, in GetValidKernelNodes() function
696 GetValidKernelNodes(func_graph, node_list); in GetValidKernelNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_stitch_atomic_clean_gpu.cc114 kernel::GetValidKernelNodes(sub_graph, &kernel_nodes); in IsStitchWithAtomic()
Dgraph_kernel_expander.cc172 kernel::GetValidKernelNodes(new_func_graph, &kernel_nodes); in CreateExpandGraphKernel()
Dgraph_kernel_splitter.cc792 GetValidKernelNodes(); in Run()
817 virtual void GetValidKernelNodes() { in GetValidKernelNodes() function in mindspore::opt::__anond4835ba50211::CostModelSplitSchemer
Dgraph_kernel_helper.cc508 kernel::GetValidKernelNodes(fg, &op_nodes, &inputs, &outputs); in AnfToJsonDesc()
539 kernel::GetValidKernelNodes(fg, &op_nodes, &inputs, &outputs); in AnfToJsonDesc()
Dgraph_kernel_cluster.cc107 kernel::GetValidKernelNodes(AnfAlgo::GetCNodeFuncGraphPtr(node), &node_list); in CountGraphKernelInnerNodes()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc489 kernel::GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in SelectGraphKernelInfo()
515 kernel::GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in SelectGraphKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_build.cc544 GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in AkgKernelParallelBuild()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc290 kernel::GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in SetGraphKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc448 kernel::GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in CreateKernelInfoFromNewParameter()