diff --git a/src/finn/core/execute_custom_node.py b/src/finn/core/execute_custom_node.py index 744ecf1b72abfd4bf2e60c5e8d258e16f72363fd..c941b5011bbae20cef7f751351dd581086d05f9d 100644 --- a/src/finn/core/execute_custom_node.py +++ b/src/finn/core/execute_custom_node.py @@ -1,6 +1,6 @@ # import onnx.helper as helper -import finn.core.MultiThreshold as multiThresh +import finn.core.multithreshold as multiThresh def execute_custom_node(node, context, graph): @@ -12,7 +12,7 @@ def execute_custom_node(node, context, graph): # save inputs v = context[node.input[0]] thresholds = context[node.input[1]] - + # calculate output output = multiThresh.execute(v, thresholds) diff --git a/src/finn/core/MultiThreshold.py b/src/finn/core/multithreshold.py similarity index 100% rename from src/finn/core/MultiThreshold.py rename to src/finn/core/multithreshold.py diff --git a/tests/test_multi_thresholding.py b/tests/test_multi_thresholding.py index b5b952d1b726376089353f4f5901630dd0181bff..49305e5572f35eb4a2e5f7678c73038777eb8b92 100644 --- a/tests/test_multi_thresholding.py +++ b/tests/test_multi_thresholding.py @@ -1,6 +1,6 @@ import numpy as np -import finn.core.MultiThreshold as multi_thresh +import finn.core.multithreshold as multi_thresh def test_execute_multi_thresholding():