compiler/tflchef/tflite/src/Op/NonMaxSuppressionV4.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 "NonMaxSuppressionV4.h"
  18
  19 #include "Convert.h"
  20 #include "FillerHelper.h"
  21
  22 namespace tflchef
  23 {
  24
  25 void TFliteOpNonMaxSuppressionV4::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   for (int32_t index = 2; index < 5; ++index)
  40   {
  41     fill_tensor_to_import(index, import);
  42   }
  43 }
  44
  45 tflchef::Operation *TFliteOpNonMaxSuppressionV4::build(const tflite::Operator *op,
  46                                                        TFliteImport *import,
  47                                                        tflchef::ModelRecipe *model_recipe) const
  48 {
  49   auto operation = model_recipe->add_operation();
  50
  51   operation->set_type("NonMaxSuppressionV4");
  52
  53   return operation;
  54 }
  55
  56 } // namespace tflchef