Searched defs:init_h (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_dynamic_gru_v2.py | 31 def construct(self, x, weight_i, weight_h, bias_i, bias_h, init_h): argument
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | lstm.py | 212 def _stacked_bi_dynamic_rnn(self, x, init_h, init_c, weight, bias): argument
240 def _stacked_dynamic_rnn(self, x, init_h, init_c, weight, bias): argument
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/tbe/ |
| D | tbe_json_creator_test.py | 62 def func_dynamic_rnn(x, w, b, seq_length, init_h, init_c): argument
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_nn_ops.py | 980 def bprop(x, w, b, seq_length, init_h, init_c, out, dout): argument
1001 def bprop(x, winput, whidden, binput, bhidden, seq, init_h, out, dout): argument
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 1002 def construct(self, x, w_i, w_h, b_i, b_h, init_h): argument
|