| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/ |
| D | pytorch_lstm_parser.cc | 47 auto batch_first = PytorchNodeParser::GetValueFromConstNode<bool>(torch_node->input(NINTH_INPUT)); in Parse() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 41 …def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout, p… argument
58 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
|
| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_gru_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
|
| D | test_rnn_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
|
| D | test_lstm_op.py | 40 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_lstm_op.py | 40 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/ops/ |
| D | test_dyn_shape_primitive.py | 108 … def __init__(self, input_s, hidden_s, num_layers, has_bias, batch_first, bidirectional, dropout): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | transformer.py | 138 … add_zero_attn=False, kdim=None, vdim=None, batch_first=False, dtype=mstype.float32): argument
|
| D | rnns.py | 375 batch_first=False, dropout=0., bidirectional=False, dtype=mstype.float32): argument
|