Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc431 auto input_tensor_node = tensor_list_cnode->input(1); in CreateLSTMNode() local
433 …std::vector<AnfNodePtr> new_node_inputs = {value_node, input_tensor_node, i_weight, c_weight, bias… in CreateLSTMNode()
Dtflite_lstm_cell_fusion.cc598 auto input_tensor_node = tensor_list_cnode->input(1); in CreateLSTMNode() local
600 …std::vector<AnfNodePtr> new_node_inputs = {value_node, input_tensor_node, i_weight, c_weight, bias… in CreateLSTMNode()