1 /**
2 * Copyright 2020-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
17 #include "nnacl/base/arithmetic_base.h"
18 #include "nnacl/kernel/arithmetic.h"
19
CalcMultiplesAndStrides(ArithmeticParameter * param)20 void CalcMultiplesAndStrides(ArithmeticParameter *param) {
21 for (size_t i = 0; i < param->ndim_; i++) {
22 if (param->in_shape0_[i] != 0) {
23 param->multiples0_[i] = param->out_shape_[i] / param->in_shape0_[i];
24 }
25 if (param->in_shape1_[i] != 0) {
26 param->multiples1_[i] = param->out_shape_[i] / param->in_shape1_[i];
27 }
28 }
29 // cal strides
30 ComputeStrides(param->in_shape0_, param->in_strides0_, param->ndim_);
31 ComputeStrides(param->in_shape1_, param->in_strides1_, param->ndim_);
32 ComputeStrides(param->out_shape_, param->out_strides_, param->ndim_);
33 }
34
CalcStructMultiplesAndStrides(ArithmeticStruct * arithmetic)35 void CalcStructMultiplesAndStrides(ArithmeticStruct *arithmetic) {
36 for (size_t i = 0; i < arithmetic->ndim_; i++) {
37 if (arithmetic->in_shape0_[i] != 0) {
38 arithmetic->multiples0_[i] = arithmetic->out_shape_[i] / arithmetic->in_shape0_[i];
39 }
40 if (arithmetic->in_shape1_[i] != 0) {
41 arithmetic->multiples1_[i] = arithmetic->out_shape_[i] / arithmetic->in_shape1_[i];
42 }
43 }
44 // cal strides
45 ComputeStrides(arithmetic->in_shape0_, arithmetic->in_strides0_, arithmetic->ndim_);
46 ComputeStrides(arithmetic->in_shape1_, arithmetic->in_strides1_, arithmetic->ndim_);
47 ComputeStrides(arithmetic->out_shape_, arithmetic->out_strides_, arithmetic->ndim_);
48 }
49