File tree Expand file tree Collapse file tree 2 files changed +0
-387
lines changed Expand file tree Collapse file tree 2 files changed +0
-387
lines changed Original file line number Diff line number Diff line change 34
34
warnings .warn ("WARNING: ONNX converter is not enabled!" , stacklevel = 1 )
35
35
__onnx_enabled__ = False
36
36
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
-
45
37
# ----------Layer handling register----------#
46
38
model_types = ['keras' , 'pytorch' , 'onnx' ]
47
39
@@ -139,11 +131,6 @@ def convert_from_config(config):
139
131
model = pytorch_to_hls (yamlConfig )
140
132
else :
141
133
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." )
147
134
else :
148
135
model = keras_to_hls (yamlConfig )
149
136
You can’t perform that action at this time.
0 commit comments