Add Tensor3dMapper to CKW

In CKW, tensors are assumed to be 3d, except the batch dimension. Tensor3dMapper class defines how an Nd tensor is mapped to 3d. It also provides utility functions to access certain dimensions and the associated strides.

Partially Resolves: COMPMID-5791

Signed-off-by: Gunes Bayir <gunes.bayir@arm.com>
Change-Id: I17c176220201ff92954ab5808fa1c1f29966d4e9
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9993
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

1 files changed, 140 insertions, 0 deletions

diff --git a/compute_kernel_writer/src/Tensor3dMapper.cpp b/compute_kernel_writer/src/Tensor3dMapper.cpp
new file mode 100644
index 0000000000..60f4f57466
--- /dev/null
+++ b/compute_kernel_writer/src/Tensor3dMapper.cpp
@@ -0,0 +1,140 @@
+/*
+ * Copyright (c) 2023 Arm Limited.
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to
+ * deal in the Software without restriction, including without limitation the
+ * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
+ * sell copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#include "Tensor3dMapper.h"
+
+#include "ckw/Error.h"
+#include "ckw/types/TensorSamplerTypes.h"
+#include "src/ITensor.h"
+
+
+namespace ckw
+{
+Tensor3dMapper::Tensor3dMapper(ITensor *tensor, TensorSampler sampler)
+        : _tensor(tensor), _sampler(sampler)
+{
+}
+
+std::string Tensor3dMapper::tensor_component_x() const
+{
+    const TensorSamplerFormat format = _sampler.format();
+    switch(format)
+    {
+        case TensorSamplerFormat::Dim0_Dim1xDim2_1:
+        case TensorSamplerFormat::Dim0_Dim1_Dim2:
+            return _tensor->component(TensorComponentType::Dim0).scalar(0,0).str;
+        default:
+            CKW_THROW_MSG("Unsupported tensor format");
+            return "";
+    }
+}
+
+std::string Tensor3dMapper::tensor_component_y() const
+{
+    const TensorSamplerFormat format = _sampler.format();
+    switch(format)
+    {
+        case TensorSamplerFormat::Dim0_Dim1xDim2_1:
+            return _tensor->component(TensorComponentType::Dim1xDim2).scalar(0,0).str;
+        case TensorSamplerFormat::Dim0_Dim1_Dim2:
+            return _tensor->component(TensorComponentType::Dim1).scalar(0,0).str;
+        default:
+            CKW_THROW_MSG("Unsupported tensor format");
+            return "";
+    }
+}
+
+std::string Tensor3dMapper::tensor_component_z() const
+{
+    const TensorSamplerFormat format = _sampler.format();
+    switch(format)
+    {
+        case TensorSamplerFormat::Dim0_Dim1xDim2_1:
+            return "1";
+        case TensorSamplerFormat::Dim0_Dim1_Dim2:
+            return _tensor->component(TensorComponentType::Dim2).scalar(0,0).str;
+        default:
+            CKW_THROW_MSG("Unsupported tensor format");
+            return "";
+    }
+}
+
+std::string Tensor3dMapper::tensor_component_stride_x() const
+{
+    const TensorSamplerFormat format = _sampler.format();
+    switch(format)
+    {
+        case TensorSamplerFormat::Dim0_Dim1xDim2_1:
+        case TensorSamplerFormat::Dim0_Dim1_Dim2:
+            return _tensor->component(TensorComponentType::Stride0).scalar(0,0).str;
+        default:
+            CKW_THROW_MSG("Unsupported tensor format");
+            return "";
+    }
+}
+
+std::string Tensor3dMapper::tensor_component_stride_y() const
+{
+    const TensorSamplerFormat format = _sampler.format();
+    switch(format)
+    {
+        case TensorSamplerFormat::Dim0_Dim1xDim2_1:
+        case TensorSamplerFormat::Dim0_Dim1_Dim2:
+            return _tensor->component(TensorComponentType::Stride1).scalar(0,0).str;
+        default:
+            CKW_THROW_MSG("Unsupported tensor format");
+            return "";
+    }
+}
+
+std::string Tensor3dMapper::tensor_component_stride_z() const
+{
+    const TensorSamplerFormat format = _sampler.format();
+    switch(format)
+    {
+        case TensorSamplerFormat::Dim0_Dim1xDim2_1:
+            return "0";
+        case TensorSamplerFormat::Dim0_Dim1_Dim2:
+            return _tensor->component(TensorComponentType::Stride2).scalar(0,0).str;
+        default:
+            CKW_THROW_MSG("Unsupported tensor format");
+            return "";
+    }
+}
+
+std::string Tensor3dMapper::tensor_component_stride_batch() const
+{
+    return _tensor->component(TensorComponentType::Stride3).scalar(0,0).str;
+}
+
+TensorSampler Tensor3dMapper::sampler() const
+{
+    return _sampler;
+}
+
+ITensor *Tensor3dMapper::tensor() const
+{
+    return _tensor;
+}
+} // namespace ckw
\ No newline at end of file