onert-micro/luci-interpreter/pal/common/PALNeg.h

   1 /*
   2  * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
   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 #ifndef LUCI_INTERPRETER_PAL_NEG_H
  18 #define LUCI_INTERPRETER_PAL_NEG_H
  19
  20 namespace luci_interpreter_pal
  21 {
  22 template <typename T>
  23 inline void Negate(const luci_interpreter::RuntimeShape &input_shape, const T *input_data,
  24                    const luci_interpreter::RuntimeShape &output_shape, T *output_data)
  25
  26 {
  27   // check that input and output dimensions are equal
  28   int N = input_shape.dimensionsCount();
  29   assert(N == output_shape.dimensionsCount());
  30
  31   // check that sizes of all dimensions are equal
  32   for (int i = 0; i < N; ++i)
  33   {
  34     assert(input_shape.dims(i) == output_shape.dims(i));
  35   }
  36
  37   const int flat_size = input_shape.flatSize();
  38
  39   for (int i = 0; i < flat_size; ++i)
  40   {
  41     output_data[i] = -input_data[i];
  42   }
  43 }
  44 } // namespace luci_interpreter_pal
  45
  46 #endif // LUCI_INTERPRETER_PAL_NEG_H