diff options
author | TatWai Chong <tatwai.chong@arm.com> | 2024-02-05 11:56:46 -0800 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-02-23 22:41:41 +0000 |
commit | 97f1c0e602049ee43537ad6c8a08f476eb5c722b (patch) | |
tree | 48fe2ca626e4346bf5b328ceab7e2f1f8700d13e /reference_model/src/ops/data_layout.h | |
parent | 20ab3df3d3100af68c47825846eee31925ff592d (diff) | |
download | reference_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.h | 6 |
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; }; |