Searched defs:PackedLinearWeightFp16 (Results 1 – 1 of 1) sorted by relevance
112 struct TORCH_API PackedLinearWeightFp16 : public LinearPackedParamsBase { struct113 PackedLinearWeightFp16( in PackedLinearWeightFp16() function118 std::unique_ptr<fbgemm::PackedGemmMatrixFP16> w;119 std::optional<at::Tensor> bias_;121 at::Tensor apply( in apply()127 at::Tensor apply_relu( in apply_relu()150 std::optional<at::Tensor> bias() override { in bias()