Searched defs:custom_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/om/ |
| D | om_model_parser.cc | 105 auto custom_node = func_graph->NewCNode(prim_c, graph_input); in CreateGraphOutputs() local
149 bool OMModelParser::SetCustomOutputs(const CNodePtr &custom_node) { in SetCustomOutputs()
175 …elParser::CreateMakeTupleGraphOutput(const FuncGraphPtr &func_graph, const CNodePtr &custom_node) { in CreateMakeTupleGraphOutput()
212 bool OMModelParser::SetMultiOutputs(const CNodePtr &custom_node) { in SetMultiOutputs()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/ |
| D | custom_ascend_utils.cc | 109 … CustomAscendUtils::SetCustomOutputs(const FuncGraphPtr &func_graph, const CNodePtr &custom_node) { in SetCustomOutputs()
203 CNodePtr custom_node = func_graph->NewCNode(prim_c, graph_input); in CreateCustomNode() local
239 …endUtils::CreateMakeTupleGraphOutput(const FuncGraphPtr &func_graph, const CNodePtr &custom_node) { in CreateMakeTupleGraphOutput()
270 …AscendUtils::ModifyGraphByCustomNode(const FuncGraphPtr &func_graph, const CNodePtr &custom_node) { in ModifyGraphByCustomNode()
370 CNodePtr custom_node = nullptr; in GetCustomNode() local
410 auto custom_node = GetCustomNode(func_graph); in ParseCustomFuncGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/akg/ |
| D | akg_kernel_builder.h | 34 virtual bool GenerateAkgKernelNodes(const FuncGraphPtr &func_graph, const AnfNodePtr &custom_node, in GenerateAkgKernelNodes()
|
| D | cpu_kernel_builder.cc | 113 …ernelBuilder::GenerateAkgKernelNodes(const FuncGraphPtr &func_graph, const AnfNodePtr &custom_node, in GenerateAkgKernelNodes()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/ |
| D | cluster_cce_lib_ops.cc | 60 std::vector<size_t> GetPerm(const CNodePtr &custom_node, const CNodePtr &old_node) { in GetPerm()
98 void ClusterCceLibOps::ReorderSubGraphInputs(const AnfNodePtr &custom_node, const AnfNodePtrList &o… in ReorderSubGraphInputs()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/third_party/ |
| D | third_party_model_parser.cc | 96 CNodePtr custom_node = nullptr; in CreateFuncGraph() local
|