(qmin, qmax, inp, scale)
| 30 | |
| 31 | |
| 32 | def tqt_forward(qmin, qmax, inp, scale): |
| 33 | op = builtin.TQT(qmin=qmin, qmax=qmax) |
| 34 | (output,) = apply(op, inp, scale) |
| 35 | return output |
| 36 | |
| 37 | |
| 38 | def lsq_forward(qmin, qmax, inp, step_size, zero_point=None, scale_grad=None): |