Searched refs:LstmNet (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 101 class LstmNet(nn.Cell): class
103 super(LstmNet, self).__init__()
166 net = LstmNet(batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropout)
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_lstm_op.py | 31 class LstmNet(nn.Cell): class
33 super(LstmNet, self).__init__()
126 …net = LstmNet(seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, d…
|