import torch
def register_quantized_ops():
from . import lowering
quantized = torch.ops.quantized
lowering.add_needs_realized_inputs(
[
quantized.max_pool2d,
]
)
lowering.make_fallback(quantized.max_pool2d)
Back to Directory