Searched defs:num_directions (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | lstm.cc | 55 int64_t num_directions = GetValue<int64_t>(primitive->GetAttr(kNumDirections)); in LstmInfer() local
139 void LSTM::set_num_directions(const int64_t num_directions) { in set_num_directions()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | lstm_adjust_pass.cc | 70 TypeId data_type, bool is_bias = false, size_t num_directions = 1) { in InitLstmWeight()
153 size_t num_directions = primitive_c->get_bidirectional() ? kBirectionalNums : 1; in ReplaceLstmNode() local
|
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_rnn_op.py | 50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
|
| D | test_gru_op.py | 50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
|
| D | test_lstm_op.py | 51 def __init__(self, num_layers, has_bias, input_s, num_directions, hidden_s, bidirectional): argument
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | lstm_grad.cc | 67 void LSTMGrad::set_num_directions(const int64_t num_directions) { in set_num_directions()
|