Function at::quantized_gru_cell

• Defined in File Functions.h

Function Documentation

inline at::Tensor at::quantized_gru_cell(const at::Tensor &input, const at::Tensor &hx, const at::Tensor &w_ih, const at::Tensor &w_hh, const at::Tensor &b_ih, const at::Tensor &b_hh, const at::Tensor &packed_ih, const at::Tensor &packed_hh, const at::Tensor &col_offsets_ih, const at::Tensor &col_offsets_hh, const at::Scalar &scale_ih, const at::Scalar &scale_hh, const at::Scalar &zero_point_ih, const at::Scalar &zero_point_hh)