compiler/tflchef/tflite/src/FillerHelper.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 "FillerHelper.h"
  18
  19 #include "Convert.h"
  20
  21 namespace tflchef
  22 {
  23
  24 void fill_tensor_to_import(int32_t idx, TFliteImport *import)
  25 {
  26   const tflite::Tensor *tensor = import->tensors()->Get(idx);
  27   if (tensor != nullptr)
  28   {
  29     if (tensor->type() == tflite::TensorType::TensorType_INT32)
  30     {
  31       const tflite::Buffer *buffer = import->buffers()->Get(tensor->buffer());
  32       if (buffer && buffer->data())
  33       {
  34         auto vec = extract_buffer<int32_t>(buffer);
  35         import->set_tensor_filler(idx, vec);
  36       }
  37     }
  38     else if (tensor->type() == tflite::TensorType::TensorType_FLOAT32)
  39     {
  40       const tflite::Buffer *buffer = import->buffers()->Get(tensor->buffer());
  41       if (buffer && buffer->data())
  42       {
  43         auto vec = extract_buffer<float>(buffer);
  44         import->set_tensor_filler(idx, vec);
  45       }
  46     }
  47   }
  48 }
  49
  50 } // namespace tflchef