Searched defs:QuantizedCellParamsFP16 (Results 1 – 1 of 1) sorted by relevance
481 struct QuantizedCellParamsFP16 : public CellParamsBase { struct482 QuantizedCellParamsFP16( in QuantizedCellParamsFP16() function487 c10::intrusive_ptr<LinearPackedParamsBase> packed_ih;488 c10::intrusive_ptr<LinearPackedParamsBase> packed_hh;489 const Tensor b_ih_;490 const Tensor b_hh_;492 Tensor matmul_ih(const Tensor& /* unused */) const override { in matmul_ih()495 Tensor matmul_hh(const Tensor& /* unused */) const override { in matmul_hh()498 Tensor linear_ih(const Tensor& input) const override { in linear_ih()501 Tensor linear_hh(const Tensor& h) const override { in linear_hh()[all …]