/third_party/mindspore/mindspore/core/ops/ |
D | lstm.cc | 106 void LSTM::set_input_size(const int64_t input_size) { in set_input_size() 110 int64_t LSTM::get_input_size() const { return GetValue<int64_t>(GetAttr(kInput_size)); } in get_input_size() 111 void LSTM::set_hidden_size(const int64_t hidden_size) { in set_hidden_size() 115 int64_t LSTM::get_hidden_size() const { return GetValue<int64_t>(GetAttr(kHidden_size)); } in get_hidden_size() 116 void LSTM::set_num_layers(const int64_t num_layers) { in set_num_layers() 120 int64_t LSTM::get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() 121 void LSTM::set_has_bias(const bool has_bias) { (void)AddAttr(kHasBias, MakeValue(has_bias)); } in set_has_bias() 122 bool LSTM::get_has_bias() const { in get_has_bias() 126 void LSTM::set_dropout(const float dropout) { in set_dropout() 130 float LSTM::get_dropout() const { in get_dropout() [all …]
|
D | lstm.h | 36 class MS_CORE_API LSTM : public PrimitiveC { 39 LSTM() : PrimitiveC(kNameLSTM) {} in LSTM() function 41 ~LSTM() = default; 42 MS_DECLARE_PARENT(LSTM, PrimitiveC); 108 using PrimLstmPtr = std::shared_ptr<LSTM>;
|
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
D | convert_tuple_output_to_maketuple_test.py | 21 LSTM = P.LSTM(input_size=10, hidden_size=2, num_layers=1, has_bias=True, bidirectional=False, dropo… variable 41 res = LSTM(x, h, c, w) 46 res = LSTM(x, h, c, w)
|
/third_party/mindspore/tests/st/ops/ascend/ |
D | test_lstm_op.py | 40 class LSTM(nn.Cell): class 43 …self.lstm = nn.LSTM(input_size=input_s, hidden_size=hidden_s, num_layers=num_layers, has_bias=has_… 111 …net = LSTM(input_s=input_s, hidden_s=16, num_layers=num_layers, has_bias=has_bias, batch_first=Fal… 119 …net_pynative = LSTM(input_s=input_s, hidden_s=16, num_layers=num_layers, has_bias=has_bias, batch_… 151 …net = LSTM(input_s=input_s, hidden_s=16, num_layers=num_layers, has_bias=has_bias, batch_first=Fal… 165 …net_pynative = LSTM(input_s=input_s, hidden_s=16, num_layers=num_layers, has_bias=has_bias, batch_…
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
D | lstm_gpu_kernel.cc | 21 MS_REG_GPU_KERNEL_ONE(LSTM, 33 MS_REG_GPU_KERNEL_ONE(LSTM,
|
/third_party/mindspore/mindspore/nn/layer/ |
D | lstm.py | 53 class LSTM(Cell): class 141 super(LSTM, self).__init__() 152 self.lstm = P.LSTM(input_size=input_size, 387 self.lstm = P.LSTM(input_size=input_size,
|
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
D | onnx_lstm_parser.cc | 25 auto prim = std::make_unique<ops::LSTM>(); in Parse()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
D | lstm_cpu_kernel.h | 60 MS_REG_CPU_KERNEL(LSTM,
|
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_lstm_op.py | 39 self.lstm = P.LSTM(input_size, hidden_size, num_layers, has_bias, bidirectional, dropout) 171 self.lstm = P.LSTM(input_size, hidden_size, num_layers, has_bias, bidirectional, dropout) 328 self.lstm = P.LSTM(input_size, hidden_size, num_layers, has_bias, bidirectional, dropout) 602 self.lstm = P.LSTM(input_size, hidden_size, num_layers, has_bias, bidirectional, dropout) 876 self.lstm = P.LSTM(input_size, hidden_size, num_layers, has_bias, bidirectional, dropout)
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
D | lstm_infer.c | 72 REG_INFER(LSTM, PrimType_LSTM, LstmInferShape)
|
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
D | schema.fbs | 234 LSTM = 16, 603 // Full LSTM kernel which supports peephole and projection. 605 // Basic LSTM kernels. Equivalent to TensorFlow BasicLSTMCell. 611 // Parameters for LSTM version 1 or above. 616 // Parameters for LSTM version 2 or above. 620 // Parameters for LSTM version 4 or above. 633 // Parameter for Unidirectional Sequence LSTM version 4. 1010 // this operator.(e.g. used by RNN and LSTM).
|
/third_party/mindspore/third_party/proto/tensorflow/lite/ |
D | schema.fbs | 234 LSTM = 16, 603 // Full LSTM kernel which supports peephole and projection. 605 // Basic LSTM kernels. Equivalent to TensorFlow BasicLSTMCell. 611 // Parameters for LSTM version 1 or above. 616 // Parameters for LSTM version 2 or above. 620 // Parameters for LSTM version 4 or above. 633 // Parameter for Unidirectional Sequence LSTM version 4. 1010 // this operator.(e.g. used by RNN and LSTM).
|
D | schema_2.4.1.fbs | 237 LSTM = 16, 610 // Full LSTM kernel which supports peephole and projection. 612 // Basic LSTM kernels. Equivalent to TensorFlow BasicLSTMCell. 618 // Parameters for LSTM version 1 or above. 623 // Parameters for LSTM version 2 or above. 627 // Parameters for LSTM version 4 or above. 640 // Parameter for Unidirectional Sequence LSTM version 4. 1033 // this operator.(e.g. used by RNN and LSTM).
|
/third_party/mindspore/tests/ut/python/nn/ |
D | test_lstm.py | 29 self.lstm = nn.LSTM(input_size=input_size,
|
/third_party/mindspore/tests/st/networks/ |
D | test_gpu_lstm.py | 75 self.encoder = P.LSTM(input_size=embed_size, hidden_size=self.num_hiddens,
|
/third_party/mindspore/mindspore/lite/src/ops/ |
D | ops_func_declare.h | 348 FUNC_MSOP2SCHEMAOP_DECLARE(LSTM)
|
D | ops_def.cc | 106 OP_TYPE(LSTM) 654 OP_SCHEMA_DEF(LSTM) 664 OP_SCHEMA_DEF_END(LSTM)
|
/third_party/mindspore/mindspore/ops/operations/ |
D | __init__.py | 68 from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSparseLazyAdam, AdamNo…
|
/third_party/mindspore/mindspore/lite/schema/ |
D | ops.fbs | 107 LSTM, 654 table LSTM {
|
D | ops_generated.h | 272 struct LSTM; 1589 template<> struct PrimitiveTypeTraits<mindspore::schema::LSTM> { 6750 struct LSTM FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { 6806 typedef LSTM Table; 6810 fbb_.AddElement<uint8_t>(LSTM::VT_BIDIRECTIONAL, static_cast<uint8_t>(bidirectional), 0); 6813 fbb_.AddElement<uint8_t>(LSTM::VT_HAS_BIAS, static_cast<uint8_t>(has_bias), 0); 6816 fbb_.AddElement<int64_t>(LSTM::VT_INPUT_SIZE, input_size, 0); 6819 fbb_.AddElement<int64_t>(LSTM::VT_HIDDEN_SIZE, hidden_size, 0); 6822 fbb_.AddElement<int64_t>(LSTM::VT_NUM_LAYERS, num_layers, 0); 6825 fbb_.AddElement<int64_t>(LSTM::VT_NUM_DIRECTIONS, num_directions, 0); [all …]
|
D | model_generated.h | 725 const mindspore::schema::LSTM *value_as_LSTM() const { in value_as_LSTM() 726 …dspore::schema::PrimitiveType_LSTM ? static_cast<const mindspore::schema::LSTM *>(value()) : nullp… in value_as_LSTM() 1420 template<> inline const mindspore::schema::LSTM *Primitive::value_as<mindspore::schema::LSTM>() con…
|
/third_party/mindspore/mindspore/lite/tools/converter/parser/ |
D | lstm_adjust_pass.cc | 142 auto primitive_c = GetValueNode<std::shared_ptr<mindspore::ops::LSTM>>(lstm_cnode->input(0)); in ReplaceLstmNode()
|
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
D | tf_lstm_cell_fusion.cc | 358 auto lstm_prim = std::make_shared<ops::LSTM>(); in CreateLSTMNode()
|
D | tflite_lstm_cell_fusion.cc | 523 auto lstm_prim = std::make_shared<ops::LSTM>(); in CreateLSTMNode()
|
/third_party/mindspore/ |
D | RELEASE.md | 280 - [STABLE] Add NLP models on Ascend: DGU, TextCNN, SentimentNet(LSTM). 693 - [STABLE] Add NLP models on Ascend: NAML, Fasttext, GRU, LSTM 722 - [STABLE] Optimize LSTM inference memory consumption in Graph mode with CPU. 1376 - [STABLE] GRU: a recurrent neural network architecture like the LSTM(Long-Short Term Memory) on Mu… 1378 - [STABLE] LSTM: a recurrent neural network architecture used to learn word vectors for sentiment a… 2379 4. Support BatchMatMul fusion and LSTM fusion in MindSpore Lite Converter. 2525 - LSTM API optimization([!6374](https://gitee.com/mindspore/mindspore/pulls/6374)) 2854 - New model supported: LSTM. 3119 - Supported models: AlexNet, LeNet, and LSTM
|