Home
last modified time | relevance | path

Searched refs:LSTM (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/
Dlstm.cc106 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 …]
Dlstm.h36 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/
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
41 res = LSTM(x, h, c, w)
46 res = LSTM(x, h, c, w)
/third_party/mindspore/tests/st/ops/ascend/
Dtest_lstm_op.py40 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/
Dlstm_gpu_kernel.cc21 MS_REG_GPU_KERNEL_ONE(LSTM,
33 MS_REG_GPU_KERNEL_ONE(LSTM,
/third_party/mindspore/mindspore/nn/layer/
Dlstm.py53 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/
Donnx_lstm_parser.cc25 auto prim = std::make_unique<ops::LSTM>(); in Parse()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Dlstm_cpu_kernel.h60 MS_REG_CPU_KERNEL(LSTM,
/third_party/mindspore/tests/st/ops/gpu/
Dtest_lstm_op.py39 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/
Dlstm_infer.c72 REG_INFER(LSTM, PrimType_LSTM, LstmInferShape)
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dschema.fbs234 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/
Dschema.fbs234 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).
Dschema_2.4.1.fbs237 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/
Dtest_lstm.py29 self.lstm = nn.LSTM(input_size=input_size,
/third_party/mindspore/tests/st/networks/
Dtest_gpu_lstm.py75 self.encoder = P.LSTM(input_size=embed_size, hidden_size=self.num_hiddens,
/third_party/mindspore/mindspore/lite/src/ops/
Dops_func_declare.h348 FUNC_MSOP2SCHEMAOP_DECLARE(LSTM)
Dops_def.cc106 OP_TYPE(LSTM)
654 OP_SCHEMA_DEF(LSTM)
664 OP_SCHEMA_DEF_END(LSTM)
/third_party/mindspore/mindspore/ops/operations/
D__init__.py68 from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSparseLazyAdam, AdamNo…
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs107 LSTM,
654 table LSTM {
Dops_generated.h272 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 …]
Dmodel_generated.h725 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/
Dlstm_adjust_pass.cc142 auto primitive_c = GetValueNode<std::shared_ptr<mindspore::ops::LSTM>>(lstm_cnode->input(0)); in ReplaceLstmNode()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc358 auto lstm_prim = std::make_shared<ops::LSTM>(); in CreateLSTMNode()
Dtflite_lstm_cell_fusion.cc523 auto lstm_prim = std::make_shared<ops::LSTM>(); in CreateLSTMNode()
/third_party/mindspore/
DRELEASE.md280 - [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

12