special handling for Pyg dim_names wasn't necessary

Author: abdelabd

hls4ml/converters/pyg_to_hls.py

@@ -55,7 +55,6 @@ def pyg_to_hls(config):
         'class_name': 'InputLayer',
         'input_shape': input_shapes['NodeAttr'],
         'inputs': 'input',
-        'dim_names': ['N_NODE', 'NODE_DIM'],
         'precision': fp_type
     }
     layer_list.append(NodeAttr_layer)
@@ -64,7 +63,6 @@ def pyg_to_hls(config):
         'class_name': 'InputLayer',
         'input_shape': input_shapes['EdgeAttr'],
         'inputs': 'input',
-        'dim_names': ['N_EDGE', 'EDGE_DIM'],
         'precision': fp_type
     }
     layer_list.append(EdgeAttr_layer)
@@ -73,7 +71,6 @@ def pyg_to_hls(config):
         'class_name': 'InputLayer',
         'input_shape': input_shapes['EdgeIndex'],
         'inputs': 'input',
-        'dim_names': ['N_EDGE', 'TWO'],
         'precision': int_type
     }
     layer_list.append(EdgeIndex_layer)

hls4ml/model/hls_layers.py

@@ -564,10 +564,7 @@ def initialize(self):
         shape = self.attributes['input_shape']
         if shape[0] is None:
             shape = shape[1:]
-        try:
-            dims = self.attributes['dim_names']
-        except KeyError:
-            dims = ['N_INPUT_{}_{}'.format(i, self.index) for i in range(1, len(shape) + 1)]
+        dims = ['N_INPUT_{}_{}'.format(i, self.index) for i in range(1, len(shape) + 1)]
         if self.index == 1:
             default_type_name = 'input_t'
         else:
@@ -2281,7 +2278,7 @@ def initialize(self):
 
         aggr_name = f"layer{self.index}_out"
         aggr_shape = [self.n_node, self.out_dim]
-        aggr_dims = ['N_NODE', f'LAYER{self.index}_OUT_DIM']
+        aggr_dims = ['N_NODE', f'LAYER{self.index}_OUT_DIM'] #todo: see if we can do without 'N_NODE'
         self.add_output_variable(shape=aggr_shape, dim_names=aggr_dims, out_name=aggr_name, var_name=aggr_name,
                                  precision=self.attributes.get('precision', None))