Searched defs:b_hh_opt (Results 1 – 1 of 1) sorted by relevance
1528 …const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional<Tensor>& b_hh_opt) { in lstm_cell()1642 …const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional<Tensor>& b_hh_opt) { in gru_cell()1656 …const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional<Tensor>& b_hh_opt) { in rnn_tanh_cell()1670 …const Tensor& w_hh, const std::optional<Tensor>& b_ih_opt, const std::optional<Tensor>& b_hh_opt) { in rnn_relu_cell()