|
|
@ -51,14 +51,14 @@ class L2NormCPUKernel : public LiteKernel {
|
|
|
|
int Run() override;
|
|
|
|
int Run() override;
|
|
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
protected:
|
|
|
|
L2NormParameter *l2_norm_param_;
|
|
|
|
L2NormParameter *l2_norm_param_ = nullptr;
|
|
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
private:
|
|
|
|
int MallocTmpBuffer();
|
|
|
|
int MallocTmpBuffer();
|
|
|
|
void FreeTmpBuffer();
|
|
|
|
void FreeTmpBuffer();
|
|
|
|
float sqrt_sum_;
|
|
|
|
float sqrt_sum_ = 0;
|
|
|
|
float *input_ptr_;
|
|
|
|
float *input_ptr_ = nullptr;
|
|
|
|
float *output_ptr_;
|
|
|
|
float *output_ptr_ = nullptr;
|
|
|
|
float *tmp_sum_ = nullptr;
|
|
|
|
float *tmp_sum_ = nullptr;
|
|
|
|
};
|
|
|
|
};
|
|
|
|
} // namespace mindspore::kernel
|
|
|
|
} // namespace mindspore::kernel
|
|
|
|