Home
last modified time | relevance | path

Searched defs:LSTM (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dlstm.h37 LSTM() : BaseOperator(kNameLSTM) {} in LSTM() function
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dconvert_tuple_output_to_maketuple_test.py21 LSTM = P.LSTM(input_size=10, hidden_size=2, num_layers=1, has_bias=True, bidirectional=False, dropo… variable
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_lstm_op.py39 class LSTM(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_lstm_op.py39 class LSTM(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_lstm_op.py57 class LSTM(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/ut/python/ops/
Dtest_dyn_shape_primitive.py107 class LSTM(nn.Cell): class
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/
Drnns.py751 class LSTM(_RNNBase): class
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py3069 class LSTM(Primitive): class