aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/ops/data_layout.h
diff options
context:
space:
mode:
authorTatWai Chong <tatwai.chong@arm.com>2024-02-05 11:56:46 -0800
committerEric Kunze <eric.kunze@arm.com>2024-02-23 22:41:41 +0000
commit97f1c0e602049ee43537ad6c8a08f476eb5c722b (patch)
tree48fe2ca626e4346bf5b328ceab7e2f1f8700d13e /reference_model/src/ops/data_layout.h
parent20ab3df3d3100af68c47825846eee31925ff592d (diff)
downloadreference_model-97f1c0e602049ee43537ad6c8a08f476eb5c722b.tar.gz
Fix comparing unsigned int and int in slice evaluation
also check if tensors of start and size are allocated. Change-Id: I6a72d11ebcb8d0725fe267058dfd792102459427 Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Diffstat (limited to 'reference_model/src/ops/data_layout.h')
-rw-r--r--reference_model/src/ops/data_layout.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/reference_model/src/ops/data_layout.h b/reference_model/src/ops/data_layout.h
index 6ab5ebd..dee2ae0 100644
--- a/reference_model/src/ops/data_layout.h
+++ b/reference_model/src/ops/data_layout.h
@@ -149,14 +149,14 @@ public:
using InEigenShapeType = typename GetEigenType<TOSA_REF_TYPE_SHAPE>::type;
using OutEigenType = typename GetEigenType<Dtype>::type;
using TIn = Eigen::Tensor<InEigenType, Rank>;
- using TInShape = Eigen::Tensor<InEigenShapeType, 1>;
+ using TSlicing = Eigen::Tensor<InEigenShapeType, 1>;
using TOut = Eigen::Tensor<OutEigenType, Rank>;
protected:
Eigen::array<Eigen::Index, Rank> begin_array;
Eigen::array<Eigen::Index, Rank> size_array;
- TosaReference::TensorTemplate<TInShape>* start;
- TosaReference::TensorTemplate<TInShape>* size;
+ TosaReference::TensorTemplate<TSlicing>* start;
+ TosaReference::TensorTemplate<TSlicing>* size;
TosaReference::TensorTemplate<TIn>* in;
TosaReference::TensorTemplate<TOut>* out;
};