Softmax.h 746 B

12345678910111213141516171819202122232425
  1. #include "Test.h"
  2. #include "Pattern.h"
  3. class Softmax:public Client::Suite
  4. {
  5. public:
  6. Softmax(Testing::testID_t id);
  7. virtual void setUp(Testing::testID_t,std::vector<Testing::param_t>& paramsArgs,Client::PatternMgr *mgr);
  8. virtual void tearDown(Testing::testID_t,Client::PatternMgr *mgr);
  9. private:
  10. #include "Softmax_decl.h"
  11. Client::Pattern<int16_t> dims;
  12. Client::Pattern<q7_t> input;
  13. Client::RefPattern<int16_t> ref;
  14. Client::RefPattern<q7_t> samples;
  15. Client::LocalPattern<int16_t> output;
  16. Client::LocalPattern<q7_t> temp;
  17. int nbSamples;
  18. int vecDim;
  19. };