Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dlstm_populate_v0.cc27 auto lstm_prim = primitive->value_as_Lstm(); in PopulateLstmParameter() local
28 if (lstm_prim == nullptr) { in PopulateLstmParameter()
40 lstm_param->bidirectional_ = lstm_prim->bidirection(); in PopulateLstmParameter()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc358 auto lstm_prim = std::make_shared<ops::LSTM>(); in CreateLSTMNode() local
359 MS_CHECK_TRUE_RET(lstm_prim != nullptr, nullptr); in CreateLSTMNode()
360 lstm_prim->set_bidirectional(false); in CreateLSTMNode()
361 lstm_prim->set_zoneout_cell(zoneout_cell); in CreateLSTMNode()
362 lstm_prim->set_zoneout_hidden(zoneout_hidden); in CreateLSTMNode()
363 auto value_node = NewValueNode(lstm_prim); in CreateLSTMNode()
Dtflite_lstm_cell_fusion.cc523 auto lstm_prim = std::make_shared<ops::LSTM>(); in CreateLSTMNode() local
524 MS_CHECK_TRUE_RET(lstm_prim != nullptr, nullptr); in CreateLSTMNode()
525 lstm_prim->set_bidirectional(false); in CreateLSTMNode()
526 lstm_prim->set_zoneout_cell(zoneout_cell); in CreateLSTMNode()
527 lstm_prim->set_zoneout_hidden(zoneout_hidden); in CreateLSTMNode()
528 auto value_node = NewValueNode(lstm_prim); in CreateLSTMNode()