Searched refs:lstm_cnode (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | lstm_adjust_pass.cc | 137 auto lstm_cnode = lstm_node->cast<CNodePtr>(); in ReplaceLstmNode() local
138 if (lstm_cnode == nullptr) { in ReplaceLstmNode()
141 MS_CHECK_TRUE_MSG(lstm_cnode->input(0) != nullptr, RET_ERROR, "lstm_cnode->input(0) is nullptr."); in ReplaceLstmNode()
142 auto primitive_c = GetValueNode<std::shared_ptr<mindspore::ops::LSTM>>(lstm_cnode->input(0)); in ReplaceLstmNode()
147 auto inputs = lstm_cnode->inputs(); in ReplaceLstmNode()
168 auto lstm_weight_node = GetRealLstmWeightNode(func_graph, lstm_cnode, kLSTMWeightIndex); in ReplaceLstmNode()
172 …auto ret = FetchDataFromParameterNode(lstm_cnode, kLSTMWeightIndex, converter::kFmkTypeMs, false, … in ReplaceLstmNode()
178 …auto ret = FetchDataFromValueNode(lstm_cnode, kLSTMWeightIndex, converter::kFmkTypeMs, false, &dat… in ReplaceLstmNode()
228 auto lstm_name = lstm_cnode->fullname_with_scope(); in ReplaceLstmNode()
235 new_lstm_cnode->set_abstract(lstm_cnode->abstract()->Clone()); in ReplaceLstmNode()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.h | 59 … const CNodePtr &lstm_cnode, const CNodePtr &output_get_item);
|
| D | tflite_lstm_cell_fusion.cc | 631 … const CNodePtr &lstm_cnode, const CNodePtr &output_get_item) { in AdjustOtherGetItems() argument
633 MS_ASSERT(lstm_cnode != nullptr && output_get_item != nullptr); in AdjustOtherGetItems()
655 new_inputs[1] = lstm_cnode; in AdjustOtherGetItems()
672 …get_item->set_fullname_with_scope(lstm_cnode->fullname_with_scope() + "_getitem_" + std::to_string… in AdjustOtherGetItems()
|