1 files changed, 3 insertions, 1 deletions
diff --git a/compute_kernel_writer/include/ckw/TileOperand.h b/compute_kernel_writer/include/ckw/TileOperand.h
index fe44b73e82..873a9825f3 100644
--- a/compute_kernel_writer/include/ckw/TileOperand.h
+++ b/compute_kernel_writer/include/ckw/TileOperand.h
@@ -29,6 +29,7 @@ namespace ckw
 {
 
 class KernelWriter;
+class TensorOperand;
 class ITile;
 
 /** A tile operand refers to a tile object that can be used for kernel writing. */
@@ -36,8 +37,9 @@ class TileOperand
 {
 public:
     // The constructor and _tile field is completely hidden from the public API to avoid any misuse.
-    // Only kernel writer class interacts with tile operand hence we allow it to access this field.
+    // Only kernel writer and tensor operand classes create and interact with tile operand hence we allow them to access this field.
     friend class KernelWriter;
+    friend class TensorOperand;
 
 private:
     // These are hidden from the public API to avoid any misuse.