summaryrefslogtreecommitdiff
path: root/thirdparty/opus/mlp.h
diff options
context:
space:
mode:
Diffstat (limited to 'thirdparty/opus/mlp.h')
-rw-r--r--thirdparty/opus/mlp.h35
1 files changed, 9 insertions, 26 deletions
diff --git a/thirdparty/opus/mlp.h b/thirdparty/opus/mlp.h
index d7670550fd..618e246e2c 100644
--- a/thirdparty/opus/mlp.h
+++ b/thirdparty/opus/mlp.h
@@ -1,4 +1,5 @@
-/* Copyright (c) 2017 Jean-Marc Valin */
+/* Copyright (c) 2008-2011 Octasic Inc.
+ Written by Jean-Marc Valin */
/*
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
@@ -27,34 +28,16 @@
#ifndef _MLP_H_
#define _MLP_H_
-#include "opus_types.h"
-
-#define WEIGHTS_SCALE (1.f/128)
-
-#define MAX_NEURONS 32
+#include "arch.h"
typedef struct {
- const opus_int8 *bias;
- const opus_int8 *input_weights;
- int nb_inputs;
- int nb_neurons;
- int sigmoid;
-} DenseLayer;
-
-typedef struct {
- const opus_int8 *bias;
- const opus_int8 *input_weights;
- const opus_int8 *recurrent_weights;
- int nb_inputs;
- int nb_neurons;
-} GRULayer;
-
-extern const DenseLayer layer0;
-extern const GRULayer layer1;
-extern const DenseLayer layer2;
+ int layers;
+ const int *topo;
+ const float *weights;
+} MLP;
-void compute_dense(const DenseLayer *layer, float *output, const float *input);
+extern const MLP net;
-void compute_gru(const GRULayer *gru, float *state, const float *input);
+void mlp_process(const MLP *m, const float *in, float *out);
#endif /* _MLP_H_ */