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

   1 /*
   2  * Copyright (c) 2020 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/Pow.h"
  19 #include "kernels/Utils.h"
  20
  21 #include <tensorflow/lite/kernels/internal/reference/reference_ops.h>
  22
  23 namespace luci_interpreter
  24 {
  25 namespace kernels
  26 {
  27
  28 Pow::Pow(const Tensor *input1, const Tensor *input2, Tensor *output)
  29   : Kernel({input1, input2}, {output})
  30 {
  31 }
  32
  33 void Pow::configure()
  34 {
  35   LUCI_INTERPRETER_CHECK(input1()->element_type() == input2()->element_type());
  36   LUCI_INTERPRETER_CHECK(input1()->element_type() == output()->element_type());
  37   // TODO: enable it only if kernel with dynamic shapes
  38   output()->resize(calculateShapeForBroadcast(input1()->shape(), input2()->shape()));
  39 }
  40
  41 void Pow::execute() const
  42 {
  43   switch (input1()->element_type())
  44   {
  45     case DataType::FLOAT32:
  46       eval<float>();
  47       break;
  48     case DataType::S32:
  49       eval<int32_t>();
  50       break;
  51     default:
  52       assert(false && "Unsupported type.");
  53   }
  54 }
  55
  56 template <typename T> void Pow::eval() const
  57 {
  58   tflite::ArithmeticParams params{};
  59
  60   const bool need_broadcast = tflite::reference_ops::ProcessBroadcastShapes(
  61     getTensorShape(input1()), getTensorShape(input2()), &params);
  62
  63   if (need_broadcast)
  64   {
  65     tflite::reference_ops::BroadcastPow4DSlow(getTensorShape(input1()), getTensorData<T>(input1()),
  66                                               getTensorShape(input2()), getTensorData<T>(input2()),
  67                                               getTensorShape(output()), getTensorData<T>(output()));
  68   }
  69   else
  70   {
  71     tflite::reference_ops::Pow(getTensorShape(input1()), getTensorData<T>(input1()),
  72                                getTensorShape(input2()), getTensorData<T>(input2()),
  73                                getTensorShape(output()), getTensorData<T>(output()));
  74   }
  75 }
  76
  77 } // namespace kernels
  78 } // namespace luci_interpreter