Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc547 auto c2o_weight = utils::cast<AnfNodePtr>((*equiv)[vars[16]]); in CreateLSTMNode() local
548 MS_ASSERT(c2o_weight); in CreateLSTMNode()
575 std::vector<AnfNodePtr> c_weights{c2i_weight, c2o_weight, c2f_weight, c2c_weight}; in CreateLSTMNode()