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(getShape(_input), getBuffer<float>(_input), getShape(_output),
  48                                      getBuffer<float>(_output));
  49       break;
  50
  51     case OperandType::QUANT_UINT8_ASYMM:
  52     {
  53       nnfw::cker::L2NormParams params;
  54       assert(_input->data_zero_point() == 128);
  55       params.input_zero_point = _input->data_zero_point();
  56       nnfw::cker::L2NormalizeQuant8(params, getShape(_input), getBuffer<uint8_t>(_input),
  57                                     getShape(_output), getBuffer<uint8_t>(_output));
  58     }
  59     break;
  60
  61     default:
  62       throw std::runtime_error{"L2Norm: Unsupported data type"};
  63   }
  64 }
  65
  66 } // namespace ops
  67 } // namespace cpu
  68 } // namespace backend
  69 } // namespace onert