black

nod-ai · Dec 8, 2023 · f548a5c · f548a5c
1 parent cd64f3a
commit f548a5c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/resnet-18/resnet-18.py b/examples/resnet-18/resnet-18.py
@@ -45,11 +45,13 @@ def forward(self, x=AbstractTensor(None, 3, 224, 224, dtype=torch.float32)):
         const = [x.dynamic_dim(0) < 16]
         return jittable(forward)(x, constraints=const)
 
+
 # build an mlir module to compile with 1-shot exporter
 exported = export(RN18)
 
 compiled_binary = exported.compile(save_to=None)
 
+
 # return type is rt.array_interop.DeviceArray
 # np.array of outputs can be accessed via to_host() method
 def shark_infer(x):