Searched defs:PackedLinearWeight (Results 1 – 2 of 2) sorted by relevance
14 struct TORCH_API PackedLinearWeight struct16 PackedLinearWeight(std::unique_ptr<fbgemm::BCSRMatrix<int8_t>> w, in PackedLinearWeight() argument33 std::unique_ptr<fbgemm::BCSRMatrix<int8_t>> w;34 std::optional<at::Tensor> bias_;35 std::vector<int32_t> col_offsets;36 std::vector<float> w_scale;37 std::vector<int32_t> w_zp;38 c10::QScheme q_scheme;49 at::Tensor apply_dynamic(const at::Tensor& input) override { in apply_dynamic()56 at::Tensor apply_dynamic_relu(const at::Tensor& input) override { in apply_dynamic_relu()[all …]
25 struct TORCH_API PackedLinearWeight : public LinearPackedParamsBase { struct26 PackedLinearWeight( in PackedLinearWeight() argument39 std::unique_ptr<fbgemm::PackBMatrix<int8_t>> w;40 std::optional<at::Tensor> bias_;41 std::vector<int32_t> col_offsets;42 std::vector<float> w_scale;43 std::vector<int32_t> w_zp;44 c10::QScheme q_scheme;86 std::optional<at::Tensor> bias() override { in bias()