1 //
2 //  CPUInstanceNorm.hpp
3 //  MNN
4 //
5 //  Created by MNN on 2019/02/28.
6 //  Copyright © 2018, Alibaba Group Holding Limited
7 //
8 
9 #ifndef CPUInstanceNorm_hpp
10 #define CPUInstanceNorm_hpp
11 
12 #include "core/AutoStorage.h"
13 #include "core/Execution.hpp"
14 
15 namespace MNN {
16 class CPUInstanceNorm : public Execution {
17 public:
18     CPUInstanceNorm(Backend *backend, const MNN::Op *op);
19     virtual ~CPUInstanceNorm() = default;
20     virtual ErrorCode onExecute(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) override;
21 
22 private:
23     AutoStorage<float> mScale;
24     AutoStorage<float> mBias;
25     float mEpsilon;
26 };
27 } // namespace MNN
28 
29 #endif /* CPUInstanceNorm_hpp */
30