Searched defs:b_ih_opt (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1528 …const Tensor& w_ih, const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional… in lstm_cell()
1642 …const Tensor& w_ih, const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional… in gru_cell()
1656 …const Tensor& w_ih, const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional… in rnn_tanh_cell()
1670 …const Tensor& w_ih, const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional… in rnn_relu_cell()
|