Add declare_constant_tile API function in CKW

Resolves: COMPMID-6535

Change-Id: I07d8aca96a0fcbd624f828b24513ee0500a14a74
Signed-off-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10200
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>

1 files changed, 19 insertions, 3 deletions

diff --git a/compute_kernel_writer/include/ckw/KernelWriter.h b/compute_kernel_writer/include/ckw/KernelWriter.h
index d59867fa6f..022ae83999 100644
--- a/compute_kernel_writer/include/ckw/KernelWriter.h
+++ b/compute_kernel_writer/include/ckw/KernelWriter.h
@@ -27,6 +27,7 @@
 
 #include "ckw/TensorOperand.h"
 #include "ckw/TileOperand.h"
+#include "ckw/types/ConstantData.h"
 #include "ckw/types/ConvertPolicy.h"
 #include "ckw/types/Operators.h"
 
@@ -37,13 +38,13 @@
 namespace ckw
 {
 
-class Kernel;
-
 /** Forward Declerations */
+class Kernel;
 class TensorInfo;
 class TensorSampler;
 class TileInfo;
 
+enum class DataType;
 enum class TargetArchitecture;
 enum class TargetLanguage;
 
@@ -175,10 +176,19 @@ public:
      * @param[in] name Name of the tile
      * @param[in] tile_info Shape and data type of the tile
      *
-     * @returns The created tile operand
+     * @return The created tile operand
      */
     virtual TileOperand declare_tile(const std::string &name, const TileInfo &tile_info) = 0;
 
+    /** Declare a constant tile given a @ref:ConstantData object
+     *
+     * @param[in] data a @ref ckw::ConstantData object that has the values and the
+     *                 underlying data type of the constant tile
+     *
+     * @return The created constant tile operand
+     */
+    virtual TileOperand declare_constant_tile(const ConstantData &data) = 0;
+
     /** Load the data from the tensor memory to the tile using the sampling information.
      *
      * @param[in] tile_op   The tile to be loaded.
@@ -240,6 +250,12 @@ protected:
     /** Get the reference to tensor object from the tensor operand. */
     static ITensor &get_tensor(const TensorOperand &operand);
 
+    /** Get the values of a constant data object. */
+    static const std::vector<std::vector<std::string>> &get_values(const ConstantData &data);
+
+    /** Get the data type of a constant data object. */
+    static DataType get_data_type(const ConstantData &data);
+
 private:
     int32_t _id_space{ 0 };
 };