Searched defs:batch_first (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | lstm.py | 137 batch_first=False, argument
380 batch_first=False, argument
|
| D | rnns.py | 203 batch_first=False, dropout=0.0, bidirectional=False): argument
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_lstm.py | 27 def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional): argument
|
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_rnn_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
|
| D | test_gru_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
|
| D | test_lstm_op.py | 41 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 41 batch_first=False, argument
|