@dskhudia: I tried using the qfn to multiply 2 matrices and I get the below error.
For a of shape (1,10) and b of shape (10,20)  if i do
qfn.mul(a_quant, b_quant)

I get,
r = ops.quantized.mul(x, y, scale=self.scale, zero_point=self.zero_point)
RuntimeError: The size of tensor a (10) must match the size of tensor b (20) at non-singleton dimension 1

Is there any function for matrix multiplication of quantized tensors?