Skip to content

Commit eca1ea3

Browse files
authored
Merge pull request #795 from vloncar/remove_tf2hls
Remove tf_to_hls
2 parents 1108c65 + 66b6ebc commit eca1ea3

File tree

2 files changed

+0
-387
lines changed

2 files changed

+0
-387
lines changed

hls4ml/converters/__init__.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -34,14 +34,6 @@
3434
warnings.warn("WARNING: ONNX converter is not enabled!", stacklevel=1)
3535
__onnx_enabled__ = False
3636

37-
try:
38-
from hls4ml.converters.tf_to_hls import tf_to_hls
39-
40-
__tensorflow_enabled__ = True
41-
except ImportError:
42-
warnings.warn("WARNING: Tensorflow converter is not enabled!", stacklevel=1)
43-
__tensorflow_enabled__ = False
44-
4537
# ----------Layer handling register----------#
4638
model_types = ['keras', 'pytorch', 'onnx']
4739

@@ -139,11 +131,6 @@ def convert_from_config(config):
139131
model = pytorch_to_hls(yamlConfig)
140132
else:
141133
raise Exception("PyTorch not found. Please install PyTorch.")
142-
elif 'TensorFlowModel' in yamlConfig:
143-
if __tensorflow_enabled__:
144-
model = tf_to_hls(yamlConfig)
145-
else:
146-
raise Exception("TensorFlow not found. Please install TensorFlow.")
147134
else:
148135
model = keras_to_hls(yamlConfig)
149136

0 commit comments

Comments
 (0)