Home
last modified time | relevance | path

Searched defs:CreateCustomOp (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/akg/
Dascend_kernel_builder.cc34 AnfNodePtr AscendKernelBuilder::CreateCustomOp(const FuncGraphPtr &func_graph, const CNodePtr &cnod… in CreateCustomOp() function in mindspore::graphkernel::AscendKernelBuilder
Dgpu_kernel_builder.cc57 AnfNodePtr GpuKernelBuilder::CreateCustomOp(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in CreateCustomOp() function in mindspore::graphkernel::GpuKernelBuilder
Dcpu_kernel_builder.cc31 AnfNodePtr CpuKernelBuilder::CreateCustomOp(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in CreateCustomOp() function in mindspore::graphkernel::CpuKernelBuilder
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/converter_extend/pass/
Dpass_registry_tutorial.cc45 api::AnfNodePtr PassTutorial::CreateCustomOp(const api::FuncGraphPtr func_graph, const api::CNodePt… in CreateCustomOp() function in mindspore::opt::PassTutorial
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dconvert_custom_for_ge.cc26 AnfNodePtr ConvertCustomForGE::CreateCustomOp(const FuncGraphPtr &func_graph, const CNodePtr &cnode… in CreateCustomOp() function in mindspore::graphkernel::ConvertCustomForGE
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/registry/
Dpass_registry_test.cc150 AnfNodePtr CreateCustomOp(const FuncGraphPtr func_graph, const CNodePtr &cnode) { in CreateCustomOp() function in mindspore::opt::Test2Fusion
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dcustom_creator.cc96 api::CNodePtr CustomOpCreator::CreateCustomOp(const api::FuncGraphPtr &func_graph, Subgraph *subgra… in CreateCustomOp() function in mindspore::dpico::CustomOpCreator