Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc563 void AkgKernelJsonGenerator::GenStitchJson(const std::vector<AnfNodePtr> &anf_nodes, in GenStitchJson()
588 bool AkgKernelJsonGenerator::CollectFusedJson(const std::vector<AnfNodePtr> &anf_nodes, in CollectFusedJson()
655 bool AkgKernelJsonGenerator::GenSingleJsons(const std::vector<AnfNodePtr> &anf_nodes, in GenSingleJsons()
679 void AkgKernelJsonGenerator::UpdateTensorName(const std::vector<AnfNodePtr> &anf_nodes, in UpdateTensorName()
708 nlohmann::json AkgKernelJsonGenerator::CreateInputsJson(const std::vector<AnfNodePtr> &anf_nodes, in CreateInputsJson()
732 void AkgKernelJsonGenerator::GenParallelJson(const std::vector<AnfNodePtr> &anf_nodes, in GenParallelJson()
796 nlohmann::json AkgKernelJsonGenerator::CreateOutputsJson(const std::vector<AnfNodePtr> &anf_nodes, in CreateOutputsJson()
837 bool AkgKernelJsonGenerator::CollectFusedJson(const std::vector<AnfNodePtr> &anf_nodes, in CollectFusedJson()
Dakg_kernel_build.cc526 bool AkgKernelBuilder::AkgKernelParallelBuild(const std::vector<AnfNodePtr> &anf_nodes) { in AkgKernelParallelBuild()
/third_party/mindspore/tests/ut/cpp/stub/kernel/
Dkernel_fusion_stub.cc40 bool AscendKernelCompileManager::AscendSingleOpCompile(const std::vector<AnfNodePtr> &anf_nodes) { … in AscendSingleOpCompile()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dfusion_base_pass.h52 std::vector<AnfNodePtr> anf_nodes; member
Dub_pattern_fusion.cc64const std::vector<AnfNodePtr> &anf_nodes, session::KernelGraph *kernel_graph) { in CreateFusionOp()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_parallel_build.cc38 bool TbeOpParallelBuild(const std::vector<AnfNodePtr> &anf_nodes) { in TbeOpParallelBuild()
Dascend_kernel_compile.cc541 std::vector<AnfNodePtr> anf_nodes; in AscendPreBuild() local
586 bool AscendKernelCompileManager::AscendSingleOpCompile(const std::vector<AnfNodePtr> &anf_nodes) { in AscendSingleOpCompile()