Home
last modified time | relevance | path

Searched refs:hidden_zoneout_new_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.h81 mutable VarPtr hidden_zoneout_new_ = nullptr; variable
Dtflite_lstm_cell_fusion.cc162 hidden_zoneout_new_ = std::make_shared<Var>(); in Init()
163 MS_CHECK_TRUE_RET(hidden_zoneout_new_ != nullptr, false); in Init()
259 VectorRef zoneout_hidden_new = VectorRef({is_var_mul5, hidden_zoneout_new_, output}); in GetBodyGraphPattern()
386 auto hidden_zoneout_new_node = utils::cast<AnfNodePtr>((*equiv)[hidden_zoneout_new_]); in CheckBodyGraph()
Dtf_lstm_cell_fusion.cc158 VectorRef new_to_hidden = VectorRef({is_var_mul3, hidden_zoneout_new_, new_hidden}); in GetBodyGraphPattern()