diff options
Diffstat (limited to 'lib/rbcodec/codecs/libopus/mlp.h')
-rw-r--r-- | lib/rbcodec/codecs/libopus/mlp.h | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/lib/rbcodec/codecs/libopus/mlp.h b/lib/rbcodec/codecs/libopus/mlp.h new file mode 100644 index 0000000000..d7670550fd --- /dev/null +++ b/lib/rbcodec/codecs/libopus/mlp.h | |||
@@ -0,0 +1,60 @@ | |||
1 | /* Copyright (c) 2017 Jean-Marc Valin */ | ||
2 | /* | ||
3 | Redistribution and use in source and binary forms, with or without | ||
4 | modification, are permitted provided that the following conditions | ||
5 | are met: | ||
6 | |||
7 | - Redistributions of source code must retain the above copyright | ||
8 | notice, this list of conditions and the following disclaimer. | ||
9 | |||
10 | - Redistributions in binary form must reproduce the above copyright | ||
11 | notice, this list of conditions and the following disclaimer in the | ||
12 | documentation and/or other materials provided with the distribution. | ||
13 | |||
14 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | ||
15 | ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | ||
16 | LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | ||
17 | A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR | ||
18 | CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | ||
19 | EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | ||
20 | PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR | ||
21 | PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF | ||
22 | LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING | ||
23 | NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS | ||
24 | SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||
25 | */ | ||
26 | |||
27 | #ifndef _MLP_H_ | ||
28 | #define _MLP_H_ | ||
29 | |||
30 | #include "opus_types.h" | ||
31 | |||
32 | #define WEIGHTS_SCALE (1.f/128) | ||
33 | |||
34 | #define MAX_NEURONS 32 | ||
35 | |||
36 | typedef struct { | ||
37 | const opus_int8 *bias; | ||
38 | const opus_int8 *input_weights; | ||
39 | int nb_inputs; | ||
40 | int nb_neurons; | ||
41 | int sigmoid; | ||
42 | } DenseLayer; | ||
43 | |||
44 | typedef struct { | ||
45 | const opus_int8 *bias; | ||
46 | const opus_int8 *input_weights; | ||
47 | const opus_int8 *recurrent_weights; | ||
48 | int nb_inputs; | ||
49 | int nb_neurons; | ||
50 | } GRULayer; | ||
51 | |||
52 | extern const DenseLayer layer0; | ||
53 | extern const GRULayer layer1; | ||
54 | extern const DenseLayer layer2; | ||
55 | |||
56 | void compute_dense(const DenseLayer *layer, float *output, const float *input); | ||
57 | |||
58 | void compute_gru(const GRULayer *gru, float *state, const float *input); | ||
59 | |||
60 | #endif /* _MLP_H_ */ | ||