onert-micro/luci-interpreter/src/kernels/Square.cpp

   1 /*
   2  * Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved
   3  * Copyright 2019 The TensorFlow Authors. All Rights Reserved.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *    http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #include "kernels/Square.h"
  19 #include "kernels/Utils.h"
  20
  21 #include <cmath>
  22
  23 namespace luci_interpreter
  24 {
  25
  26 namespace kernels
  27 {
  28
  29 Square::Square(const Tensor *input, Tensor *output) : Kernel({input}, {output}) {}
  30
  31 void Square::configure()
  32 {
  33   if (input()->element_type() != output()->element_type())
  34   {
  35     assert(false && "Input/output tensor data type mismatch.");
  36   }
  37   // TODO: enable it only if kernel with dynamic shapes
  38   output()->resize(input()->shape());
  39 }
  40
  41 void Square::execute() const
  42 {
  43   switch (input()->element_type())
  44   {
  45     case DataType::FLOAT32:
  46       evalFloat();
  47       break;
  48
  49     default:
  50       assert(false && "Unsupported type.");
  51   }
  52 }
  53
  54 void Square::evalFloat() const
  55 {
  56   auto in = getTensorData<float>(input());
  57   auto out = getTensorData<float>(output());
  58   auto size = getTensorShape(input()).FlatSize();
  59   for (auto i = in; i != in + size; ++i)
  60   {
  61     *out = (*i) * (*i);
  62     ++out;
  63   }
  64 }
  65
  66 } // namespace kernels
  67 } // namespace luci_interpreter