1 #include "Test.h"
2 #include "Pattern.h"
3 
4 #include "dsp/bayes_functions.h"
5 
6 class BayesF32:public Client::Suite
7     {
8         public:
9             BayesF32(Testing::testID_t id);
10             virtual void setUp(Testing::testID_t,std::vector<Testing::param_t>& paramsArgs,Client::PatternMgr *mgr);
11             virtual void tearDown(Testing::testID_t,Client::PatternMgr *mgr);
12         private:
13             #include "BayesF32_decl.h"
14 
15             Client::Pattern<float32_t> input;
16             Client::Pattern<float32_t> params;
17             Client::Pattern<int16_t> dims;
18 
19             Client::LocalPattern<float32_t> outputProbas;
20             Client::LocalPattern<float32_t> temp;
21             Client::LocalPattern<int16_t> outputPredicts;
22 
23             // Reference patterns are not loaded when we are in dump mode
24             Client::RefPattern<int16_t> predicts;
25 
26             int classNb,vecDim;
27             int nb=0;
28             const float32_t *theta;
29             const float32_t *sigma;
30             const float32_t *classPrior;
31             float32_t epsilon;
32 
33             arm_gaussian_naive_bayes_instance_f32 bayes;
34 
35             const float32_t *inp;
36             float32_t *bufp,*tempp;
37 
38     };
39