Plumbing for the frame rate network

This commit is contained in:
Jean-Marc Valin 2018-11-25 17:20:24 -05:00
parent 70fdf47471
commit 7119eaf33b
3 changed files with 58 additions and 2 deletions

View file

@ -187,10 +187,10 @@ hf.write('#define MAX_CONV_INPUTS {}\n\n'.format(max_conv_inputs))
hf.write('#define MAX_MDENSE_TMP {}\n\n'.format(max_mdense_tmp))
hf.write('struct RNNState {\n')
hf.write('typedef struct {\n')
for i, name in enumerate(layer_list):
hf.write(' float {}_state[{}_STATE_SIZE];\n'.format(name, name.upper()))
hf.write('};\n')
hf.write('} LPCNetState;\n')
hf.write('\n\n#endif\n')