| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_lstm_op.py | 32 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
164 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
321 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
595 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
869 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 38 hidden_size, argument
102 …def __init__(self, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropo… argument
205 …def __init__(self, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropo… argument
293 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
|
| /third_party/mindspore/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
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
|
| D | test_rnn_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | lstm_grad.cc | 39 void LSTMGrad::set_hidden_size(const int64_t hidden_size) { in set_hidden_size()
79 void LSTMGrad::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, … in Init()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | lstm.cc | 56 int64_t hidden_size = GetValue<int64_t>(primitive->GetAttr(kHidden_size)); in LstmInfer() local
111 void LSTM::set_hidden_size(const int64_t hidden_size) { in set_hidden_size()
151 void LSTM::Init(const int64_t input_size, const int64_t hidden_size, const int64_t num_layers, cons… in Init()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | lstm_infer_test.cc | 32 int hidden_size = 2; in TEST_F() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | lstm.py | 134 hidden_size, argument
378 hidden_size, argument
|
| D | rnns.py | 202 def __init__(self, mode, input_size, hidden_size, num_layers=1, has_bias=True, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | lstm_infer.c | 44 int hidden_size = weight_i->shape_[1] / 4; in LstmInferShape() local
|
| D | gru_infer.c | 59 int hidden_size = w_gate_shape[1] / 3; in GruInferShape() local
|
| /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/mindspore/parallel/nn/ |
| D | transformer.py | 320 def __init__(self, hidden_size, argument
687 hidden_size, argument
1152 hidden_size, argument
1447 def __init__(self, hidden_size, argument
1872 hidden_size, argument
2063 hidden_size, argument
2279 hidden_size, argument
|
| D | moe.py | 89 def __init__(self, hidden_size, argument
|
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | anomaly_detection.py | 44 def __init__(self, encoder, decoder, hidden_size=400, latent_size=20): argument
|
| /third_party/mindspore/mindspore/nn/probability/dpn/vae/ |
| D | vae.py | 52 def __init__(self, encoder, decoder, hidden_size, latent_size): argument
|
| D | cvae.py | 57 def __init__(self, encoder, decoder, hidden_size, latent_size, num_classes): argument
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_lstm.py | 31 def InitialLstmWeight(input_size, hidden_size, num_layers, bidirectional, has_bias=False): argument
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 66 hidden_size=768, argument
605 hidden_size, argument
674 hidden_size=768, argument
743 hidden_size, argument
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_bidirection_gru_fusion.cc | 395 int *hidden_size) { in GetInputAndHiddenSize()
475 … const int input_size, const int hidden_size, float *gate_tensor_data, in ConvertWeightData()
530 const int hidden_size, float *tensor_data) { in ConvertBiasData()
624 int hidden_size = 0; in CreateBiDirectionGruNode() local
|
| D | tf_lstm_cell_fusion.cc | 186 … const float *const data_ptr, const int hidden_size) { in SetWeightAbstractAndDefault()
223 const ParameterPtr &weight_c, int hidden_size) { in SplitWeights()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | lstm_fp32.c | 120 float *state_buffer, int batch, int hidden_size, const float zoneout) { in UpdataState()
138 int batch, int hidden_size, const float zoneout) { in UpdataOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | lstm_fp16.c | 135 … const float16_t *cell_gate, float16_t *state_buffer, int batch, int hidden_size, in UpdataStateFp16()
154 float16_t *state_buffer, int batch, int hidden_size, float16_t zoneout) { in UpdataOutputFp16()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | lstm_adjust_pass.cc | 152 size_t hidden_size = static_cast<size_t>(primitive_c->get_hidden_size()); in ReplaceLstmNode() local
|
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | bert_attention_submodules.py | 368 hidden_size=768, argument
|