We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1b7b10e commit 2f0518dCopy full SHA for 2f0518d
examples/xnnpack/aot_compiler.py
@@ -92,6 +92,7 @@
92
logging.info("Quantizing Model...")
93
# TODO(T165162973): This pass shall eventually be folded into quantizer
94
model = quantize(model, example_inputs)
95
+ ep = torch.export.export_for_training(model, example_inputs)
96
97
edge = to_edge_transform_and_lower(
98
ep,
0 commit comments