pytorch

1
import torch
2

3

4
def register_quantized_ops():
5
    from . import lowering
6

7
    quantized = torch.ops.quantized
8

9
    lowering.add_needs_realized_inputs(
10
        [
11
            quantized.max_pool2d,
12
        ]
13
    )
14

15
    lowering.make_fallback(quantized.max_pool2d)
16