FullyConnected.h 1.1 KB

123456789101112131415161718192021222324252627282930313233343536
  1. #include "Test.h"
  2. #include "Pattern.h"
  3. class FullyConnected:public Client::Suite
  4. {
  5. public:
  6. FullyConnected(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 "FullyConnected_decl.h"
  11. int32_t output_mult = 1073741824;
  12. int16_t output_shift = -1;
  13. int32_t filter_offset = 1;
  14. int32_t input_offset = 1;
  15. int32_t output_offset = -1;
  16. int32_t act_min =-128;
  17. int32_t act_max= 127;
  18. int32_t nb_batches=1;
  19. int32_t rowDim;
  20. int32_t colDim;
  21. Client::Pattern<q7_t> input;
  22. Client::Pattern<q31_t> bias;
  23. Client::Pattern<q7_t> weight;
  24. Client::LocalPattern<q7_t> output;
  25. Client::LocalPattern<q15_t> temp;
  26. // Reference patterns are not loaded when we are in dump mode
  27. Client::RefPattern<q7_t> ref;
  28. };