Function at::lstm_mps_backward_outf¶

Function Documentation¶

inline void at::lstm_mps_backward_outf(const ::std::optional<at::Tensor> &grad_y, const ::std::optional<at::Tensor> &grad_hy, const ::std::optional<at::Tensor> &grad_cy, const at::Tensor &z_state, const at::Tensor &cell_state_fwd, const at::Tensor &input, const at::Tensor &layersOutputs, at::TensorList hx, at::TensorList params, bool has_biases, int64_t num_layers, double dropout, bool train, bool bidirectional, bool batch_first, at::Tensor &out0, at::TensorList out1, at::TensorList out2)¶