cleaning and formatting

Author: mlomb

evaluation/measure.py

@@ -120,9 +120,9 @@ def postprocess(times_in_ns: list[int]) -> list[float]:
 
     return results | (
         {
-            #"tensorflow": postprocess(
-            #    measure_tf(tf_model, inputs, total, tqdm_leave=tqdm_leave)
-            #),
+            "tensorflow": postprocess(
+                measure_tf(tf_model, inputs, total, tqdm_leave=tqdm_leave)
+            ),
             "onnxruntime": postprocess(
                 measure_onnxruntime(model_proto, inputs, total, tqdm_leave=tqdm_leave)
             ),

evaluation/measure_models.py

@@ -1,9 +1,7 @@
 import setup  # noqa # isort:skip
 
-import keras
 import numpy as np
 import pandas as pd
-from keras import layers
 import onnx
 from measure import measure_all
 from tqdm import tqdm
@@ -23,7 +21,7 @@
     "../data/vision/classification/squeezenet/model/squeezenet1.1-7.onnx",
     "../data/vision/body_analysis/emotion_ferplus/model/emotion-ferplus-8.onnx",
     "../data/vision/classification/inception_and_googlenet/inception_v2/model/inception-v2-7.onnx",
-    "../data/vision/classification/resnet/model/resnet50-caffe2-v1-9.onnx"
+    "../data/vision/classification/resnet/model/resnet50-caffe2-v1-9.onnx",
 ]
 
 for model in tqdm(models):

onnx2code/ops/gemm_tiling/GEMM.py

@@ -40,7 +40,7 @@ def set_tiling_params(params: LoopTilingParams) -> None:
 
 
 def call_GEMM(M: int, K: int, N: int, params: str) -> str:
-    nc = min(2**math.ceil(math.log2(N)), tiling_params.nc)
+    nc = min(2 ** math.ceil(math.log2(N)), tiling_params.nc)
     kc = tiling_params.kc
     mc = tiling_params.mc
     mr = tiling_params.mr