| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | cells.py | 118 def gru_cell(input, hidden, w_ih, w_hh, b_ih, b_hh): argument
132 def rnn_relu_cell(input, hidden, w_ih, w_hh, b_ih, b_hh): argument
138 def rnn_tanh_cell(input, hidden, w_ih, w_hh, b_ih, b_hh): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 208 const Tensor& b_ih() const override { in b_ih() function
283 const Tensor& b_ih() const override { in b_ih() function
312 b_ih = std::move(tensors[2]), b_hh = std::move(tensors[3]), in __setstate__() local
340 at::Tensor b_ih, in make_quantized_cell_params()
422 const Tensor& b_ih() const override { in b_ih() function
505 const Tensor& b_ih() const override { in b_ih() function
563 const Tensor& b_ih() const { in b_ih() function
1531 const Tensor& b_ih = *b_ih_maybe_owned; in lstm_cell() local
1645 const Tensor& b_ih = *b_ih_maybe_owned; in gru_cell() local
1659 const Tensor& b_ih = *b_ih_maybe_owned; in rnn_tanh_cell() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Gru.cpp | 89 const auto& b_ih = params_cpu[i * 4 + 2]; in gru_input() local
155 const auto& b_ih = params_cpu.at(i * 4 + 2); in pack_linear_op_contexts() local
|
| H A D | Lstm.cpp | 114 const auto& b_ih = params_cpu[l * 4 + 2]; in lstm_input() local
191 const auto& b_ih = params_cpu[l * 4 + 2]; in pack_lstm_linear_op_contexts() local
|
| /aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
| H A D | rnn_eltwise.py | 34 def forward(self, input, hx, cx, b_ih, b_hh): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | rnn.cpp | 128 auto b_ih = torch::empty({gate_size}); in reset() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 207 def LSTMCell(input, hidden, w_ih, w_hh, b_ih=None, b_hh=None): argument
227 def LSTMCellS(x, hx, cx, w_ih, w_hh, b_ih, b_hh): argument
2843 def lstm(x, hx, cx, w_ih, w_hh, b_ih, b_hh): argument
11145 def lstm(x, hx, cx, w_ih, w_hh, b_ih, b_hh): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 4564 def lstm_cell(g: jit_utils.GraphContext, self, hidden, w_ih, w_hh, b_ih, b_hh): argument
|