FullyConnectedBench.h 951 B

123456789101112131415161718192021222324252627
  1. #include "Test.h"
  2. #include "Pattern.h"
  3. class FullyConnectedBench:public Client::Suite
  4. {
  5. public:
  6. FullyConnectedBench(Testing::testID_t id);
  7. virtual void setUp(Testing::testID_t,std::vector<Testing::param_t>& params,Client::PatternMgr *mgr);
  8. virtual void tearDown(Testing::testID_t,Client::PatternMgr *mgr);
  9. private:
  10. #include "FullyConnectedBench_decl.h"
  11. int repeatNb;
  12. Client::Pattern<q7_t> input;
  13. Client::Pattern<q31_t> bias;
  14. Client::Pattern<q7_t> weight;
  15. Client::LocalPattern<q7_t> output;
  16. Client::LocalPattern<q15_t> temp;
  17. // Reference patterns are not loaded when we are in dump mode
  18. Client::RefPattern<q7_t> ref;
  19. q7_t *inp;
  20. q31_t *biasp;
  21. q7_t *weightp;
  22. q7_t *outp;
  23. q7_t *refp;
  24. q15_t *tempp;
  25. };