Home
last modified time | relevance | path

Searched refs:ReshapeTransposeFusion (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dreshape_transpose_fusion.h31 class ReshapeTransposeFusion : public PatternProcessPass {
33 …explicit ReshapeTransposeFusion(bool multigraph = true) : PatternProcessPass("reshape_transpose_fu…
36 ~ReshapeTransposeFusion() override = default;
Dreshape_transpose_fusion.cc42 const BaseRef ReshapeTransposeFusion::DefinePattern() const { in DefinePattern()
49 const AnfNodePtr ReshapeTransposeFusion::Process(const FuncGraphPtr &func_graph, const AnfNodePtr &… in Process()
71 if (!LicManager::GetInstance().GetPassSwitch(OptPassEnum::ReshapeTransposeFusion)) { in Process()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dreshape_transpose_fusion_test.cc55 pm->AddPass(std::make_shared<opt::ReshapeTransposeFusion>()); in TEST_F()
78 pm->AddPass(std::make_shared<opt::ReshapeTransposeFusion>()); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dlic_manager.h28 ReshapeTransposeFusion, enumerator
Dlic_manager.cc37 {std::to_string(26), OptPassEnum::ReshapeTransposeFusion},
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_backend_optimization.cc187 ir_fusion_pm->AddPass(std::make_shared<ReshapeTransposeFusion>()); in AddAscendIRFusionPass()