Imported Upstream version 1.7.0

[platform/core/ml/nnfw.git] / compiler / loco / include / loco / IR / TensorShape.h

diff --git a/compiler/loco/include/loco/IR/TensorShape.h b/compiler/loco/include/loco/IR/TensorShape.h
index af1066d..44eca67 100644 (file)
--- a/compiler/loco/include/loco/IR/TensorShape.h
+++ b/compiler/loco/include/loco/IR/TensorShape.h
@@ -57,6 +57,11 @@ private:
  */
 uint32_t element_count(const loco::TensorShape *tensor_shape);
 
+/**
+ * @brief '==' operator for TensorShape
+ */
+bool operator==(const TensorShape &lhs, const TensorShape &rhs);
+
 } // namespace loco
 
 #endif // __LOCO_IR_TENSOR_SHAPE_H__