mirror of
https://github.com/xiph/opus.git
synced 2025-05-17 08:58:30 +00:00
Starting to actually test this -- fix a few OOB reads
This commit is contained in:
parent
8d62ba067e
commit
538f25565a
3 changed files with 22 additions and 3 deletions
|
@ -116,7 +116,7 @@ def dump_mdense_layer(self, f, hf):
|
|||
activation = self.activation.__name__.upper()
|
||||
max_mdense_tmp = max(max_mdense_tmp, weights[0].shape[0]*weights[0].shape[2])
|
||||
f.write('const MDenseLayer {} = {{\n {}_bias,\n {}_weights,\n {}_factor,\n {}, {}, {}, ACTIVATION_{}\n}};\n\n'
|
||||
.format(name, name, name, name, weights[0].shape[0], weights[0].shape[1], weights[0].shape[2], activation))
|
||||
.format(name, name, name, name, weights[0].shape[1], weights[0].shape[0], weights[0].shape[2], activation))
|
||||
hf.write('#define {}_OUT_SIZE {}\n'.format(name.upper(), weights[0].shape[0]))
|
||||
hf.write('extern const MDenseLayer {};\n\n'.format(name));
|
||||
return False
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue