compiler/loco/src/IR/TensorShape.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 "loco/IR/TensorShape.h"
  18
  19 #include <cassert>
  20
  21 namespace loco
  22 {
  23
  24 uint32_t element_count(const loco::TensorShape *tensor_shape)
  25 {
  26   uint32_t res = 1;
  27
  28   for (uint32_t axis = 0; axis < tensor_shape->rank(); ++axis)
  29   {
  30     // Let's use "assert" here as "caller" is responsible for this check.
  31     // Please refer to the header for details.
  32     assert(tensor_shape->dim(axis).known());
  33     res *= tensor_shape->dim(axis).value();
  34   }
  35
  36   return res;
  37 }
  38
  39 } // namespace loco
  40
  41 namespace loco
  42 {
  43
  44 bool operator==(const TensorShape &lhs, const TensorShape &rhs)
  45 {
  46   if (lhs.rank() != rhs.rank())
  47     return false;
  48   for (uint32_t axis = 0; axis < lhs.rank(); ++axis)
  49   {
  50     if (!(lhs.dim(axis) == rhs.dim(axis)))
  51       return false;
  52   }
  53   return true;
  54 }
  55
  56 } // namespace loco