Lines Matching defs:QuantizedCellParamsFP16

481 struct QuantizedCellParamsFP16 : public CellParamsBase {  struct
482   QuantizedCellParamsFP16(  in QuantizedCellParamsFP16()  argument
487   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()
505   const Tensor& b_ih() const override {  in b_ih()
508   const Tensor& b_hh() const override {  in b_hh()
511   CellParamsSerializationType __getstate__() const override {  in __getstate__()
518   static c10::intrusive_ptr<CellParamsBase> __setstate__(  in __setstate__()