Searched defs:custom_cnode (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | convert_custom_for_ge.cc | 34 auto custom_cnode = func_graph->NewCNode(inputs); in CreateCustomOp() local
113 auto custom_cnode = CreateCustomOp(func_graph, cnode); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/ |
| D | custom_creator.cc | 115 auto custom_cnode = func_graph->NewCNode(prim, subgraph_inputs); in CreateCustomOp() local
299 … const api::CNodePtr &custom_cnode, const ModelCoreInfoPtr &om_model_info) { in SetCustomOutputs()
329 const api::CNodePtr &custom_cnode, in SetCustomSingleOutput()
365 const api::CNodePtr &custom_cnode, in SetCustomMultiOutput()
|
| D | dpico_pass.cc | 443 auto custom_cnode = custom_op_creator_->CreateCustomOp(func_graph, &subgraph, om_model_info); in ReplaceSubgraphWithCustom() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/ |
| D | kernel_builder.cc | 66 auto custom_cnode = builder->CreateCustomOp(func_graph, cnode); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/akg/ |
| D | ascend_kernel_builder.cc | 40 auto custom_cnode = func_graph->NewCNode(inputs); in CreateCustomOp() local
|
| D | gpu_kernel_builder.cc | 63 auto custom_cnode = func_graph->NewCNode(inputs); in CreateCustomOp() local
|
| D | cpu_kernel_builder.cc | 73 auto custom_cnode = func_graph->NewCNode(inputs); in CreateCustomOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/examples/converter_extend/pass/ |
| D | pass_registry_tutorial.cc | 64 auto custom_cnode = func_graph->NewCNode(primc, inputs); in CreateCustomOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/ |
| D | cluster_cce_lib_ops.cc | 104 auto custom_cnode = custom_node->cast<CNodePtr>(); in ReorderSubGraphInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/registry/ |
| D | pass_registry_test.cc | 170 auto custom_cnode = func_graph->NewCNode(primc_c, inputs); in CreateCustomOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/model/ |
| D | model_impl.cc | 98 CNodePtr custom_cnode = func_graph->NewCNode(custom_prim_c, {param}); in CreateFuncGraphFromDataFlow() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_ge/ |
| D | ge_graph_executor.cc | 172 auto custom_cnode = (*itr)->cast<CNodePtr>(); in GetDataFlowGraph() local
|