onert-micro/luci-interpreter/src/kernels/LogicalOr.cpp

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   3  * Copyright 2019 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 #include "kernels/LogicalOr.h"
  19
  20 #include "kernels/Utils.h"
  21 #include "kernels/BinaryOpCommon.h"
  22
  23 namespace luci_interpreter
  24 {
  25 namespace kernels
  26 {
  27
  28 LogicalOr::LogicalOr(const Tensor *input1, const Tensor *input2, Tensor *output)
  29   : Kernel({input1, input2}, {output})
  30 {
  31 }
  32
  33 void LogicalOr::configure()
  34 {
  35   LUCI_INTERPRETER_CHECK(input1()->element_type() == input2()->element_type());
  36   LUCI_INTERPRETER_CHECK(input1()->element_type() == DataType::BOOL);
  37   // TODO: enable it only if kernel with dynamic shapes
  38   output()->resize(calculateShapeForBroadcast(input1()->shape(), input2()->shape()));
  39 }
  40
  41 void LogicalOr::execute() const
  42 {
  43   BinaryOpBroadcastSlow(getTensorShape(input1()), getTensorData<bool>(input1()),
  44                         getTensorShape(input2()), getTensorData<bool>(input2()),
  45                         getTensorShape(output()), getTensorData<bool>(output()),
  46                         [](bool x, bool y) { return x || y; });
  47 }
  48
  49 } // namespace kernels
  50 } // namespace luci_interpreter