compiler/tflchef/tflite/src/Op/NonMaxSuppressionV5.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 "NonMaxSuppressionV5.h"
  18
  19 #include "Convert.h"
  20 #include "FillerHelper.h"
  21
  22 namespace tflchef
  23 {
  24
  25 void TFliteOpNonMaxSuppressionV5::filler(const tflite::Operator *op, TFliteImport *import,
  26                                          tflchef::ModelRecipe *model_recipe) const
  27 {
  28   const auto &inputs = *op->inputs();
  29
  30   const tflite::Tensor *max_output_size_tensor = import->tensors()->Get(inputs[2]);
  31   assert(max_output_size_tensor->type() == tflite::TensorType::TensorType_INT32);
  32
  33   const tflite::Tensor *iou_threshold_tensor = import->tensors()->Get(inputs[3]);
  34   assert(iou_threshold_tensor->type() == tflite::TensorType::TensorType_FLOAT32);
  35
  36   const tflite::Tensor *score_threshold_tensor = import->tensors()->Get(inputs[4]);
  37   assert(score_threshold_tensor->type() == tflite::TensorType::TensorType_FLOAT32);
  38
  39   const tflite::Tensor *soft_nms_sigma_tensor = import->tensors()->Get(inputs[5]);
  40   assert(soft_nms_sigma_tensor->type() == tflite::TensorType::TensorType_FLOAT32);
  41
  42   for (int32_t index = 2; index < 6; ++index)
  43   {
  44     fill_tensor_to_import(index, import);
  45   }
  46 }
  47
  48 tflchef::Operation *TFliteOpNonMaxSuppressionV5::build(const tflite::Operator *op,
  49                                                        TFliteImport *import,
  50                                                        tflchef::ModelRecipe *model_recipe) const
  51 {
  52   auto operation = model_recipe->add_operation();
  53
  54   operation->set_type("NonMaxSuppressionV5");
  55
  56   return operation;
  57 }
  58
  59 } // namespace tflchef