Searched refs:lstm_node (Results 1 – 1 of 1) sorted by relevance
972 Node* lstm_node = graph->NewNode(); in ParseBasic() local973 lstm_node->operation.type = ToString(OperationType::LSTM); in ParseBasic()976 lstm_node->operation.attributes = lstm_attr; in ParseBasic()994 RETURN_IF_ERROR(graph->AddConsumer(lstm_node->id, activ_temp->id)); in ParseBasic()995 RETURN_IF_ERROR(reader->AddInput(lstm_node, 4)); // prev_state in ParseBasic()996 RETURN_IF_ERROR(reader->AddOutput(lstm_node, 1)); // new_state in ParseBasic()997 RETURN_IF_ERROR(reader->AddOutput(lstm_node, 0)); // activation in ParseBasic()