| 13 | |
| 14 | namespace MNN { |
| 15 | CPUNormalize::CPUNormalize(Backend* b, const MNN::Op* op) : MNN::Execution(b) { |
| 16 | auto normalize = op->main_as_Normalize(); |
| 17 | mAcrossSpatial = normalize->acrossSpatial(); |
| 18 | mChannelShared = normalize->channelShared(); |
| 19 | |
| 20 | mEps = normalize->eps(); |
| 21 | mScale.reset(normalize->scale()->size()); |
| 22 | ::memcpy(mScale.get(), normalize->scale()->data(), normalize->scale()->size() * sizeof(float)); |
| 23 | } |
| 24 | |
| 25 | ErrorCode CPUNormalize::onResize(const std::vector<Tensor*>& inputs, const std::vector<Tensor*>& outputs) { |
| 26 | auto inputTensor = inputs[0]; |