1 /**
2 * Copyright 2021 Huawei Technologies Co., Ltd
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 #ifndef MINDSPORE_NNACL_SQUARED_DIFFERENCE_H_
17 #define MINDSPORE_NNACL_SQUARED_DIFFERENCE_H_
18
19 #include "nnacl/fp32/squared_difference.h"
20 #include "nnacl/fp32/sub_fp32.h"
21 #include "nnacl/fp32/mul_fp32.h"
22
ElementSquaredDifference(const float * in0,const float * in1,float * out,int size)23 int ElementSquaredDifference(const float *in0, const float *in1, float *out, int size) {
24 ElementSub(in0, in1, out, size);
25 return ElementMul(out, out, out, size);
26 }
27
ElementOptSquaredDifference(const float * in0,const float * in1,float * out,int size,bool scale)28 int ElementOptSquaredDifference(const float *in0, const float *in1, float *out, int size, bool scale) {
29 ElementOptSub(in0, in1, out, size, scale);
30 return ElementMul(out, out, out, size);
31 }
32 #endif // MINDSPORE_NNACL_SQUARED_DIFFERENCE_H_
33