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