chore: Make the tf version loose

peri044 · peri044 · commit acb77f4b6f85 · 2022-02-09T22:40:16.000-08:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/tf2onnx/tf_loader.py b/tf2onnx/tf_loader.py
@@ -686,7 +686,7 @@ def tf_optimize_grappler(input_names, output_names, graph_def, fold_constant=Non
         # 'pruning', 'constfold', 'arithmetic', 'dependency', 'function',
         'function', 'dependency'
     ]
-    if is_tf2():
+    if LooseVersion(tf.__version__) > "2.4"::
         # This flag disables folding QDQ nodes around constants in the network (eg: around conv/FC weights)
         rewrite_options.experimental_disable_folding_quantization_emulation = True
 

Original file line number	Diff line number	Diff line change
`@@ -686,7 +686,7 @@ def tf_optimize_grappler(input_names, output_names, graph_def, fold_constant=Non`
`686`	`686`	`# 'pruning', 'constfold', 'arithmetic', 'dependency', 'function',`
`687`	`687`	`'function', 'dependency'`
`688`	`688`	`]`
`689`		`- if is_tf2():`
	`689`	`+ if LooseVersion(tf.__version__) > "2.4"::`
`690`	`690`	`# This flag disables folding QDQ nodes around constants in the network (eg: around conv/FC weights)`
`691`	`691`	`rewrite_options.experimental_disable_folding_quantization_emulation = True`
`692`	`692`