Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc534 auto i2f_weight = utils::cast<AnfNodePtr>((*equiv)[vars[10]]); in CreateLSTMNode() local
535 MS_ASSERT(i2f_weight); in CreateLSTMNode()
566 std::vector<AnfNodePtr> i_weights{i2i_weight, i2o_weight, i2f_weight, i2c_weight}; in CreateLSTMNode()