added statistical model to dump_nfec_model

This commit is contained in:
Jan Buethe 2022-10-19 17:18:25 +00:00
parent 50966eecc5
commit ea4d8f54c3
2 changed files with 53 additions and 3 deletions

View file

@ -157,4 +157,4 @@ def dump_conv1d_layer(self, f, hf):
hf.write('#define {}_STATE_SIZE ({}*{})\n'.format(name.upper(), weights[0].shape[1], (weights[0].shape[0]-1)))
hf.write('#define {}_DELAY {}\n'.format(name.upper(), (weights[0].shape[0]-1)//2))
hf.write('extern const Conv1DLayer {};\n\n'.format(name));
return max_conv_inputs
return max_conv_inputs