change signature of quantize_input

[chenbohua3]

As described here:

Note that this logic won’t traverse lists/dicts/any other data structures and will only consider Tensor s that are direct arguments to the call

So if we directly pass inputs (should be a tuple) to quant_grad, inputs will not appear in the training graph.