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

   1 /*
   2  * Copyright (c) 2020 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 #include "kernels/L2Normalize.h"
  18 #include "kernels/Utils.h"
  19
  20 #include "PALL2Normalize.h"
  21
  22 namespace luci_interpreter
  23 {
  24
  25 namespace kernels
  26 {
  27
  28 L2Normalize::L2Normalize(const Tensor *input, Tensor *output, const L2NormParams &params)
  29   : KernelWithParams<L2NormParams>({input}, {output}, params)
  30 {
  31 }
  32
  33 void L2Normalize::configure()
  34 {
  35   LUCI_INTERPRETER_CHECK(input()->shape().num_dims() <= 4);
  36   LUCI_INTERPRETER_CHECK(output()->element_type() == DataType::FLOAT32 ||
  37                          output()->element_type() == DataType::U8);
  38   LUCI_INTERPRETER_CHECK(input()->element_type() == output()->element_type());
  39   if (output()->element_type() == DataType::U8)
  40   {
  41     LUCI_INTERPRETER_CHECK(output()->scale() == (1. / 128.));
  42     LUCI_INTERPRETER_CHECK(output()->zero_point() == 128);
  43   }
  44   LUCI_INTERPRETER_CHECK(params().activation == Activation::NONE);
  45   // TODO: enable it only if kernel with dynamic shapes
  46   output()->resize(input()->shape());
  47 }
  48
  49 void L2Normalize::execute() const
  50 {
  51   switch (output()->element_type())
  52   {
  53     case DataType::FLOAT32:
  54       eval<float>(0);
  55       break;
  56     case DataType::U8:
  57       eval<uint8_t>(input()->zero_point());
  58       break;
  59     default:
  60       assert(false && "Unsupported type.");
  61   }
  62 }
  63
  64 template <typename T> void L2Normalize::eval(int32_t zero_point) const
  65 {
  66   tflite::L2NormalizationParams op_params{};
  67   op_params.input_zero_point = zero_point;
  68   luci_interpreter_pal::L2Normalization(op_params, getTensorShape(input()),
  69                                         getTensorData<T>(input()), getTensorShape(output()),
  70                                         getTensorData<T>(output()));
  71 }
  72
  73 } // namespace kernels
  74 } // namespace luci_interpreter