Searched defs:QRNNCellParamsWrapper (Results 1 – 1 of 1) sorted by relevance
544 struct QRNNCellParamsWrapper { struct545 QRNNCellParamsWrapper(c10::intrusive_ptr<CellParamsBase> param) in QRNNCellParamsWrapper() argument548 Tensor matmul_ih(const Tensor& input) const { in matmul_ih()551 Tensor matmul_hh(const Tensor& h) const { in matmul_hh()554 Tensor matmul_hr(const Tensor& h) const { in matmul_hr()557 Tensor linear_ih(const Tensor& input) const { in linear_ih()560 Tensor linear_hh(const Tensor& h) const { in linear_hh()563 const Tensor& b_ih() const { in b_ih()566 const Tensor& b_hh() const { in b_hh()570 c10::intrusive_ptr<CellParamsBase> param_;