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 fa5048b commit 101746eCopy full SHA for 101746e
examples/arm/example_modules/README.md
@@ -0,0 +1,7 @@
1
+# Example of an external model for the ARM AOT Compiler
2
+Example of an external Python file to be used as a module by the `run.sh` (and the `aot_arm_compiler.py`) scripts in `examples/arm` directory.
3
+Just pass the path of the `add.py` file as `--model_name`:
4
+
5
+`ModelUnderTest` should be a `torch.nn.module` instance.
6
7
+`ModelInputs` should be a tuple of inputs to the forward function.
examples/arm/example_modules/add.py
@@ -0,0 +1,13 @@
+import torch
+class myModelAdd(torch.nn.Module):
+ def __init__(self):
+ super().__init__()
8
+ def forward(self, x):
9
+ return x + x
10
11
12
+ModelUnderTest = myModelAdd()
13
+ModelInputs = (torch.ones(5),)
0 commit comments