| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | RNN.cu | 33 const TensorArg& input_gates, const TensorArg& hidden_gates, in checkSizes() argument
37 checkSameSize(c, input_gates, hidden_gates); in checkSizes()
49 checkAllSameGPU(c, {input_gates, hidden_gates, input_bias, hidden_bias, prev_hidden}); in checkSizes()
370 void lstm_forward_impl(const Tensor& input_gates, const Tensor& hidden_gates, in lstm_forward_impl() argument
382 auto hidden_gatesI = getTensorInfo<scalar_t, index_type>(hidden_gates); in lstm_forward_impl()
392 if (allContiguous({input_gates, hidden_gates, input_bias, hidden_bias, cx, hy, cy, workspace})) { in lstm_forward_impl()
443 void gru_forward_impl(const Tensor& input_gates, const Tensor& hidden_gates, in gru_forward_impl() argument
455 auto hidden_gatesI = getTensorInfo<scalar_t, index_type>(hidden_gates); in gru_forward_impl()
464 if (allContiguous({input_gates, hidden_gates, input_bias, hidden_bias, hx, hy, workspace})) { in gru_forward_impl()
518 const Tensor& input_gates, const Tensor& hidden_gates, in _thnn_fused_lstm_cell_cuda() argument
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 1546 …const Tensor& hidden_gates, const std::optional<Tensor>& input_bias_opt, const std::optional<Tenso… in _thnn_differentiable_lstm_cell_backward() argument
1559 Tensor gates = input_gates + hidden_gates; in _thnn_differentiable_lstm_cell_backward()
1601 const Tensor& hidden_gates, in _thnn_differentiable_gru_cell_backward() argument
1609 Tensor h_g = hidden_gates; in _thnn_differentiable_gru_cell_backward()
|
| H A D | native_functions.yaml | 7747 - func: _thnn_fused_lstm_cell(Tensor input_gates, Tensor hidden_gates, Tensor cx, Tensor? input_bia…
7762 …backward(Tensor? grad_hy, Tensor? grad_cy, Tensor input_gates, Tensor hidden_gates, Tensor? input_…
7764 - func: _thnn_fused_gru_cell(Tensor input_gates, Tensor hidden_gates, Tensor hx, Tensor? input_bias…
7774 …entiable_gru_cell_backward(Tensor grad_hy, Tensor input_gates, Tensor hidden_gates, Tensor hx, Ten…
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | derivatives.yaml | 2861 - name: _thnn_fused_lstm_cell(Tensor input_gates, Tensor hidden_gates, Tensor cx, Tensor? input_bia…
2863 …hidden_gates, cx, input_bias, hidden_bias: "GradMode::is_enabled() ? _thnn_differentiable_lstm_cel…
2865 - name: _thnn_fused_gru_cell(Tensor input_gates, Tensor hidden_gates, Tensor hx, Tensor? input_bias…
2866 …hidden_gates, hx, input_bias, hidden_bias: "grad.defined() ? (GradMode::is_enabled() ? _thnn_diffe…
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 5864 hidden_gates, argument
5872 input_gates.shape == hidden_gates.shape,
5873 lambda: f"{input_gates.shape} != {hidden_gates.shape}",
5895 for x in [hidden_gates, input_bias, hidden_bias, prev_hidden]
5904 hidden_gates, argument
5909 rnn_cell_checkSizes(input_gates, hidden_gates, input_bias, hidden_bias, 4, cx)
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/generated/ |
| H A D | c_shim_cuda.h | 45 …__thnn_fused_lstm_cell(AtenTensorHandle input_gates, AtenTensorHandle hidden_gates, AtenTensorHand…
|