| 32 | #define BatchNormName "BatchNormalization" |
| 33 | |
| 34 | class BatchNorm : public OperatorWithParam<BatchNorm, BatchNormParam> |
| 35 | { |
| 36 | public: |
| 37 | BatchNorm() |
| 38 | { |
| 39 | name_ = BatchNormName; |
| 40 | } |
| 41 | BatchNorm(const BatchNorm& src) = default; |
| 42 | virtual ~BatchNorm(){}; |
| 43 | |
| 44 | void SetSchema(void) override; |
| 45 | |
| 46 | float GetFops(const std::vector<TShape>& inputs, const std::vector<TShape>& outputs) override; |
| 47 | }; |
| 48 | |
| 49 | } // namespace TEngine |
| 50 |
nothing calls this directly
no outgoing calls
no test coverage detected