runtime/onert/backend/cpu/ops/L2NormLayer.cc

   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 "L2NormLayer.h"
  18
  19 #include "OperationUtils.h"
  20
  21 #include <cker/operation/L2Normalize.h>
  22 #include <cker/Types.h>
  23
  24 namespace onert
  25 {
  26 namespace backend
  27 {
  28 namespace cpu
  29 {
  30 namespace ops
  31 {
  32
  33 void L2NormLayer::configure(const IPortableTensor *input, IPortableTensor *output)
  34 {
  35   assert(input != nullptr);
  36   assert(output != nullptr);
  37
  38   _input = input;
  39   _output = output;
  40 }
  41
  42 void L2NormLayer::run()
  43 {
  44   switch (_input->data_type())
  45   {
  46     case OperandType::FLOAT32:
  47       nnfw::cker::L2NormalizeFloat32(
  48           getTensorShape(_input), reinterpret_cast<const float *>(_input->buffer()),
  49           getTensorShape(_output), reinterpret_cast<float *>(_output->buffer()));
  50       break;
  51
  52     case OperandType::QUANT_UINT8_ASYMM:
  53     {
  54       nnfw::cker::L2NormParams params;
  55       assert(_input->data_offset() == 128);
  56       params.input_zero_point = _input->data_offset();
  57       nnfw::cker::L2NormalizeQuant8(
  58           params, getTensorShape(_input), reinterpret_cast<const uint8_t *>(_input->buffer()),
  59           getTensorShape(_output), reinterpret_cast<uint8_t *>(_output->buffer()));
  60     }
  61     break;
  62
  63     default:
  64       throw std::runtime_error{"L2Norm: Unsupported data type"};
  65   }
  66 }
  67
  68 } // namespace ops
  69 } // namespace cpu
  70 } // namespace backend
  71 } // namespace onert