| /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 | 211 const Tensor& b_hh() const override { in b_hh() function
286 const Tensor& b_hh() const override { in b_hh() function
312 b_ih = std::move(tensors[2]), b_hh = std::move(tensors[3]), in __setstate__() local
341 at::Tensor b_hh) { in make_quantized_cell_params()
425 const Tensor& b_hh() const override { in b_hh() function
508 const Tensor& b_hh() const override { in b_hh() function
566 const Tensor& b_hh() const { in b_hh() function
1532 const Tensor& b_hh = c10::value_or_else(b_hh_opt, [] {return Tensor();}); in lstm_cell() local
1646 const Tensor& b_hh = c10::value_or_else(b_hh_opt, [] {return Tensor();}); in gru_cell() local
1660 const Tensor& b_hh = c10::value_or_else(b_hh_opt, [] {return Tensor();}); in rnn_tanh_cell() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Gru.cpp | 90 const auto& b_hh = params_cpu[i * 4 + 3]; in gru_input() local
156 const auto& b_hh = params_cpu.at(i * 4 + 3); in pack_linear_op_contexts() local
|
| H A D | Lstm.cpp | 115 const auto& b_hh = params_cpu[l * 4 + 3]; in lstm_input() local
192 const auto& b_hh = params_cpu[l * 4 + 3]; 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 | 131 auto b_hh = 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
|