Simplify table type names in softmax tests

fastmachinelearning · Nov 6, 2023 · 3db1e83 · 3db1e83
1 parent c7aa067
commit 3db1e83
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/test/pytest/test_softmax.py b/test/pytest/test_softmax.py
@@ -39,14 +39,12 @@ def test_softmax(backend, strategy, generate_data, input_bits, input_shape, tabl
     model.add(tf.keras.layers.Activation(input_shape=input_shape, activation='softmax', name='softmax'))
     model.compile()
 
-    f_type = (
-        f'ac_fixed<{table_bits},true,AC_RND,AC_SAT>' if backend == 'Quartus' else f'ap_fixed<{table_bits},AP_RND,AP_SAT>'
-    )
+    table_type = f'fixed<{table_bits}, RND, SAT>'
 
     cfg = hls4ml.utils.config_from_keras_model(model, granularity='name')
     cfg['LayerName']['softmax']['Strategy'] = strategy
-    cfg['LayerName']['softmax']['inv_table_t'] = f_type
-    cfg['LayerName']['softmax']['exp_table_t'] = f_type
+    cfg['LayerName']['softmax']['inv_table_t'] = table_type
+    cfg['LayerName']['softmax']['exp_table_t'] = table_type
     cfg['LayerName']['softmax_input']['Precision']['result'] = f'fixed<{input_bits}>'
 
     odir = str(