Merge a54c06b into d7b7ac3

stanford-mast · Aug 24, 2020 · 03abd06 · 03abd06
2 parents d7b7ac3 + a54c06b
commit 03abd06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nn_dataflow/tools/nn_layer_stats.py b/nn_dataflow/tools/nn_layer_stats.py
@@ -68,8 +68,8 @@ def layer_stats(args):
         elif isinstance(layer, ConvLayer):
             convs += 1
 
-        ifmap_size = layer.total_ifmap_size(word_bytes) * batch / KILO
-        ofmap_size = layer.total_ofmap_size(word_bytes) * batch / KILO
+        ifmap_size = layer.total_ifmap_size(batch, word_bytes) / KILO
+        ofmap_size = layer.total_ofmap_size(batch, word_bytes) / KILO
         try:
             filter_size = layer.total_filter_size(word_bytes) / KILO
         except AttributeError: