We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6791412 commit 956808aCopy full SHA for 956808a
test/pytest/test_optimization/test_attributes.py
@@ -1,7 +1,7 @@
1
from tensorflow.keras.layers import Conv2D, Dense, Flatten, ReLU
2
from tensorflow.keras.models import Sequential
3
4
-from hls4ml.optimization import get_attributes_from_keras_model_and_hls4ml_config
+from hls4ml.optimization.dsp_aware_pruning import get_attributes_from_keras_model_and_hls4ml_config
5
from hls4ml.utils.config import config_from_keras_model
6
7
0 commit comments