compute/cker/include/cker/operation/Fill.h

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   3  * Copyright 2018 The TensorFlow Authors. All Rights Reserved.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #ifndef __NNFW_CKER_FILL_H__
  19 #define __NNFW_CKER_FILL_H__
  20
  21 #include "cker/Shape.h"
  22
  23 namespace nnfw
  24 {
  25 namespace cker
  26 {
  27 template <typename T>
  28 inline void Fill(const Shape &input_shape, int *input_data, const T value_data,
  29                  const Shape &output_shape, T output_data)
  30 {
  31   int input_size = input_shape.FlatSize();
  32   int output_size = 1;
  33   for (int i = 0; i < input_size; i++)
  34   {
  35     output_size *= input_data[i];
  36   }
  37
  38   if (output_size == output_shape.FlatSize())
  39   {
  40     for (int i = 0; i < output_size; i++)
  41     {
  42       output_data[i] = *value_data;
  43     }
  44   }
  45   else
  46   {
  47     throw std::runtime_error("Cker Fill.h: output's size is not matched inferred size of output");
  48   }
  49 }
  50
  51 } // namespace cker
  52 } // namespace nnfw
  53
  54 #endif // __NNFW_CKER_FILL_H__