(x, scale)
| 16 | |
| 17 | |
| 18 | def fake_quant(x, scale): |
| 19 | x = x / scale |
| 20 | x = F.round(x) |
| 21 | x = F.clip(x, -128, 127) |
| 22 | x = x * scale |
| 23 | return x |
| 24 | |
| 25 | |
| 26 | @pytest.mark.parametrize("kind", ["abs", "sin", "sub", "mul", "fuse_add_tanh"]) |
no test coverage detected