[mlir/quant] fix a small typo in the quant utility

This is an edge case where the input type is a primitive type.

Differential Revision: https://reviews.llvm.org/D76442