Searched refs:GetValidKernelNodes (Results 1 – 11 of 11) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
D | common_utils.h | 96 void GetValidKernelNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *node_list); 97 void GetValidKernelNodes(const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *node_list,
|
D | common_utils.cc | 674 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/ |
D | add_stitch_atomic_clean_gpu.cc | 114 kernel::GetValidKernelNodes(sub_graph, &kernel_nodes); in IsStitchWithAtomic()
|
D | graph_kernel_expander.cc | 172 kernel::GetValidKernelNodes(new_func_graph, &kernel_nodes); in CreateExpandGraphKernel()
|
D | graph_kernel_splitter.cc | 792 GetValidKernelNodes(); in Run() 817 virtual void GetValidKernelNodes() { in GetValidKernelNodes() function in mindspore::opt::__anond4835ba50211::CostModelSplitSchemer
|
D | graph_kernel_helper.cc | 508 kernel::GetValidKernelNodes(fg, &op_nodes, &inputs, &outputs); in AnfToJsonDesc() 539 kernel::GetValidKernelNodes(fg, &op_nodes, &inputs, &outputs); in AnfToJsonDesc()
|
D | graph_kernel_cluster.cc | 107 kernel::GetValidKernelNodes(AnfAlgo::GetCNodeFuncGraphPtr(node), &node_list); in CountGraphKernelInnerNodes()
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
D | kernel_select_graph_kernel.cc | 489 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/ |
D | akg_kernel_build.cc | 544 GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in AkgKernelParallelBuild()
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
D | kernel_info_setter.cc | 290 kernel::GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in SetGraphKernelInfo()
|
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
D | kernel_graph.cc | 448 kernel::GetValidKernelNodes(func_graph, &node_list, &input_list, &output_list); in CreateKernelInfoFromNewParameter()
|