1 files changed, 139 insertions, 0 deletions
diff --git a/compiler/luci/service/src/ShapeDescription.cpp b/compiler/luci/service/src/ShapeDescription.cpp
new file mode 100644
index 000000000..cbc302f70
--- /dev/null
+++ b/compiler/luci/service/src/ShapeDescription.cpp
@@ -0,0 +1,139 @@
+/*
+ * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "luci/Service/ShapeDescription.h"
+
+#include <oops/InternalExn.h>
+
+#include <cassert>
+
+namespace luci
+{
+
+ShapeDescription to_shape_description(const loco::TensorShape &shape)
+{
+  ShapeDescription res;
+
+  res._rank_known = true;
+
+  res._dims.resize(shape.rank());
+  for (uint32_t axis = 0; axis < shape.rank(); ++axis)
+  {
+    // All the dimensions SHOULD be known
+    assert(shape.dim(axis).known());
+    res._dims.at(axis) = shape.dim(axis).value();
+  }
+
+  return res;
+}
+
+ShapeDescription to_shape_description(const loco::FeatureShape &shape)
+{
+  ShapeDescription res;
+
+  res._rank_known = true;
+
+  // T/F Lite encodes a feature map as a NHWC tensor
+  res._dims.resize(4);
+  res._dims.at(0) = shape.count().value();
+  res._dims.at(1) = shape.height().value();
+  res._dims.at(2) = shape.width().value();
+  res._dims.at(3) = shape.depth().value();
+
+  return res;
+}
+
+ShapeDescription to_shape_description(const loco::FilterShape &shape)
+{
+  ShapeDescription res;
+
+  res._rank_known = true;
+
+  // T/F Lite encodes a convolution filter as a NHWC tensor
+  res._dims.resize(4);
+  res._dims.at(0) = shape.count().value();
+  res._dims.at(1) = shape.height().value();
+  res._dims.at(2) = shape.width().value();
+  res._dims.at(3) = shape.depth().value();
+
+  return res;
+}
+
+ShapeDescription to_shape_description(const loco::DepthwiseFilterShape &shape)
+{
+  ShapeDescription res;
+
+  res._rank_known = true;
+
+  // T/F Lite encodes a depthwise convolution filter as a [1, H, W, C*M] tensor
+  res._dims.resize(4);
+  res._dims.at(0) = 1;
+  res._dims.at(1) = shape.height().value();
+  res._dims.at(2) = shape.width().value();
+  res._dims.at(3) = shape.depth().value() * shape.multiplier().value();
+
+  return res;
+}
+
+ShapeDescription to_shape_description(const loco::BiasShape &shape)
+{
+  ShapeDescription res;
+
+  res._rank_known = true;
+
+  res._dims.resize(1);
+  res._dims.at(0) = shape.length().value();
+
+  return res;
+}
+
+ShapeDescription to_shape_description(const loco::MatrixShape &shape)
+{
+  ShapeDescription res;
+
+  res._rank_known = true;
+
+  res._dims.resize(2);
+  res._dims.at(0) = shape.height().value();
+  res._dims.at(1) = shape.width().value();
+
+  return res;
+}
+
+ShapeDescription to_shape_description(const loco::NodeShape &shape)
+{
+  switch (shape.domain())
+  {
+    case loco::Domain::Tensor:
+      return to_shape_description(shape.as<loco::TensorShape>());
+    case loco::Domain::Feature:
+      return to_shape_description(shape.as<loco::FeatureShape>());
+    case loco::Domain::Filter:
+      return to_shape_description(shape.as<loco::FilterShape>());
+    case loco::Domain::DepthwiseFilter:
+      return to_shape_description(shape.as<loco::DepthwiseFilterShape>());
+    case loco::Domain::Bias:
+      return to_shape_description(shape.as<loco::BiasShape>());
+    case loco::Domain::Matrix:
+      return to_shape_description(shape.as<loco::MatrixShape>());
+    default:
+      break;
+  }
+
+  INTERNAL_EXN_V("Unsupported loco domain", oops::to_uint32(shape.domain()));
+}
+
+} // namespace luci