Home
last modified time | relevance | path

Searched refs:MomentumLossscaleFusion (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dmomentum_lossscale_fusion.h23 class MomentumLossscaleFusion : public PatternProcessPass {
25 explicit MomentumLossscaleFusion(bool multigraph = true)
28 ~MomentumLossscaleFusion() override = default;
Dmomentum_lossscale_fusion.cc41 const BaseRef MomentumLossscaleFusion::DefinePattern() const { in DefinePattern()
52 const AnfNodePtr MomentumLossscaleFusion::Process(const FuncGraphPtr &func_graph, const AnfNodePtr … in Process()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dmomentum_lossscale_fusion_test.cc44 pm->AddPass(std::make_shared<opt::MomentumLossscaleFusion>()); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dlic_manager.h33 MomentumLossscaleFusion, enumerator
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_backend_optimization.cc196 ir_fusion_pm->AddPass(std::make_shared<MomentumLossscaleFusion>()); in AddAscendIRFusionPass()