Home
last modified time | relevance | path

Searched refs:anf_nodes (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc64 … const std::vector<AnfNodePtr> &anf_nodes, session::KernelGraph *kernel_graph) { in CreateFusionOp() argument
68 for (auto &node : anf_nodes) { in CreateFusionOp()
87 for (auto &node : anf_nodes) { in CreateFusionOp()
103 buffer_fusion_kernel->set_scope((anf_nodes.back())->scope()); in CreateFusionOp()
220 (*buffer_fusion_infos)[fusion_id].anf_nodes.push_back(cnode); in GetFusionScopeComputeNodeList()
234 for (const auto &node : fusion_info.anf_nodes) { in GetFusionScopeInputNodeList()
239 … if (std::find(fusion_info.anf_nodes.begin(), fusion_info.anf_nodes.end(), real_input.first) == in GetFusionScopeInputNodeList()
240 fusion_info.anf_nodes.end()) { in GetFusionScopeInputNodeList()
296 for (const auto &node : fusion_info.anf_nodes) { in GetFusionScopeOutputNodeList()
307 … if (std::find(fusion_info.anf_nodes.begin(), fusion_info.anf_nodes.end(), use_node.first) == in GetFusionScopeOutputNodeList()
[all …]
Dfusion_base_pass.h52 std::vector<AnfNodePtr> anf_nodes; member
/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() argument
567 for (auto const &anf_node : anf_nodes) { in GenStitchJson()
588 bool AkgKernelJsonGenerator::CollectFusedJson(const std::vector<AnfNodePtr> &anf_nodes, in CollectFusedJson() argument
591 if (anf_nodes.empty()) { in CollectFusedJson()
592 …MS_LOG(ERROR) << "Invalid input size, anf_nodes [" << anf_nodes.size() << "], input_list [" << inp… in CollectFusedJson()
596 MS_LOG(DEBUG) << "Fusion nodes: [" << output_list.size() << "], input_list: [" << anf_nodes.size() in CollectFusedJson()
601 if (!GenSingleJsons(anf_nodes, &node_json_map)) return false; in CollectFusedJson()
603 UpdateTensorName(anf_nodes, &node_json_map); in CollectFusedJson()
606 std::transform(anf_nodes.begin(), anf_nodes.end(), std::back_inserter(node_json_desc), in CollectFusedJson()
610 auto inputs_json = CreateInputsJson(anf_nodes, input_list, node_json_map); in CollectFusedJson()
[all …]
Dakg_kernel_json_generator.h96 …bool CollectFusedJson(const std::vector<AnfNodePtr> &anf_nodes, const std::vector<AnfNodePtr> &inp…
99 …bool CollectFusedJson(const std::vector<AnfNodePtr> &anf_nodes, const std::vector<AnfNodePtr> &inp…
121 …void GenStitchJson(const std::vector<AnfNodePtr> &anf_nodes, std::map<AnfNodePtr, nlohmann::json> …
125 …bool GenSingleJsons(const std::vector<AnfNodePtr> &anf_nodes, std::map<AnfNodePtr, nlohmann::json>…
126 …void UpdateTensorName(const std::vector<AnfNodePtr> &anf_nodes, std::map<AnfNodePtr, nlohmann::jso…
127 …nlohmann::json CreateInputsJson(const std::vector<AnfNodePtr> &anf_nodes, const std::vector<AnfNod…
129 …nlohmann::json CreateOutputsJson(const std::vector<AnfNodePtr> &anf_nodes, const std::vector<AnfNo…
149 …void GenParallelJson(const std::vector<AnfNodePtr> &anf_nodes, const std::vector<AnfNodePtr> &inpu…
Dakg_kernel_build.h47 bool AkgKernelParallelBuild(const std::vector<AnfNodePtr> &anf_nodes);
Dakg_kernel_build.cc526 bool AkgKernelBuilder::AkgKernelParallelBuild(const std::vector<AnfNodePtr> &anf_nodes) { in AkgKernelParallelBuild() argument
528 for (const auto &anf_node : 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() argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dascend_kernel_compile.cc541 std::vector<AnfNodePtr> anf_nodes; in AscendPreBuild() local
542 GetAllAscendNodes(kernel_graph, &anf_nodes); in AscendPreBuild()
543 if (anf_nodes.empty()) { in AscendPreBuild()
548 for (const auto &node : anf_nodes) { in AscendPreBuild()
586 bool AscendKernelCompileManager::AscendSingleOpCompile(const std::vector<AnfNodePtr> &anf_nodes) { in AscendSingleOpCompile() argument
592 for (const auto &node : anf_nodes) { in AscendSingleOpCompile()
Dascend_kernel_compile.h60 bool AscendSingleOpCompile(const std::vector<AnfNodePtr> &anf_nodes);
Dtbe_kernel_parallel_build.h31 bool TbeOpParallelBuild(const std::vector<AnfNodePtr> &anf_nodes);
Dtbe_kernel_parallel_build.cc38 bool TbeOpParallelBuild(const std::vector<AnfNodePtr> &anf_nodes) { in TbeOpParallelBuild() argument
57 for (const auto &anf_node : anf_nodes) { in TbeOpParallelBuild()