Searched defs:input_bias_opt (Results 1 – 2 of 2) sorted by relevance
519 …const Tensor& cx, const std::optional<Tensor>& input_bias_opt, const std::optional<Tensor>& hidden… in _thnn_fused_lstm_cell_cuda()605 …const Tensor& hx, const std::optional<Tensor>& input_bias_opt, const std::optional<Tensor>& hidden… in _thnn_fused_gru_cell_cuda()
1546 …const Tensor& hidden_gates, const std::optional<Tensor>& input_bias_opt, const std::optional<Tenso… in _thnn_differentiable_lstm_cell_backward()1602 …const Tensor& hx, const std::optional<Tensor>& input_bias_opt, const std::optional<Tensor>& hidden… in _thnn_differentiable_gru_cell_backward()