From c5c2a7e4be64fef198c150063db9f241f21299d2 Mon Sep 17 00:00:00 2001
From: Tai Ly <tai.ly@arm.com>
Date: Thu, 22 Feb 2024 23:26:28 +0000
Subject: [ref_model] Change resize attrs to inputs

This patch implements changes needed for resize op's
scale/offset/border changing from attributes to inputs

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I20db0687fad40711f3ded644af51096292dd05b3
---
 reference_model/src/ops/image.h | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

(limited to 'reference_model/src/ops/image.h')

diff --git a/reference_model/src/ops/image.h b/reference_model/src/ops/image.h
index 6d5a418..9c08def 100644
--- a/reference_model/src/ops/image.h
+++ b/reference_model/src/ops/image.h
@@ -32,18 +32,20 @@ public:
     virtual int checkTensorAttributes() final;
     virtual int eval();
 
-    using InEigenType  = typename GetEigenType<InDtype>::type;
-    using OutEigenType = typename GetEigenType<OutDtype>::type;
-    using TIn          = Eigen::Tensor<InEigenType, 4>;
-    using TOut         = Eigen::Tensor<OutEigenType, 4>;
+    using InEigenType      = typename GetEigenType<InDtype>::type;
+    using InEigenShapeType = typename GetEigenType<TOSA_REF_TYPE_SHAPE>::type;
+    using OutEigenType     = typename GetEigenType<OutDtype>::type;
+    using TIn              = Eigen::Tensor<InEigenType, 4>;
+    using TInShape         = Eigen::Tensor<InEigenShapeType, 1>;
+    using TOut             = Eigen::Tensor<OutEigenType, 4>;
 
 protected:
     TosaResizeAttribute* attribute;
-    std::vector<int16_t> scale;
-    std::vector<int16_t> offset;
-    std::vector<int16_t> border;
     ResizeMode mode;
     TosaReference::TensorTemplate<TIn>* in;
+    TosaReference::TensorTemplate<TInShape>* scale;
+    TosaReference::TensorTemplate<TInShape>* offset;
+    TosaReference::TensorTemplate<TInShape>* border;
     TosaReference::TensorTemplate<TOut>* out;
 };
 
-- 
cgit v1.2.1