Searched defs:QuantizedCellParamsDynamic (Results 1 – 1 of 1) sorted by relevance
387 struct QuantizedCellParamsDynamic : public CellParamsBase { struct388 QuantizedCellParamsDynamic( in QuantizedCellParamsDynamic() argument402 c10::intrusive_ptr<LinearPackedParamsBase> packed_w_ih;403 c10::intrusive_ptr<LinearPackedParamsBase> packed_w_hh;404 const Tensor b_ih_;405 const Tensor b_hh_;406 bool reduce_range_;408 Tensor matmul_ih(const Tensor& input) const override { in matmul_ih()411 Tensor matmul_hh(const Tensor& h) const override { in matmul_hh()415 Tensor linear_ih(const Tensor& input_ih) const override { in linear_ih()[all …]