We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7a4d15b commit d69df60Copy full SHA for d69df60
1 file changed
examples/arm/aot_arm_compiler.py
@@ -198,13 +198,15 @@ def _load_serialized_model(
198
199
return model, example_inputs
200
201
+
202
def _apply_replace_quant_nodes(edge, args):
- """ Apply the replace_quant_nodes pass to the edge graph module. """
203
+ """Apply the replace_quant_nodes pass to the edge graph module."""
204
205
if args.target != "vgf" and not args.direct_drive:
206
edge = edge.transform([ReplaceQuantNodesPass()])
207
return edge
208
209
210
def get_model_and_inputs_from_name(
211
model_name: str, model_input: str | None
212
) -> Tuple[torch.nn.Module, Any]:
0 commit comments