diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-03-04 15:15:03 -0800 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-04-30 10:04:40 -0700 |
| commit | ad15dfab0430b72015d13d19b8a696bb9bacd0a6 (patch) | |
| tree | a78f6e90c54451ca9da812c03349feb9ef87e071 /reference_model/src/ops/data_layout.h | |
| parent | 573ecd4373f4e8f5d1a219147b5a259125059cf0 (diff) | |
| download | reference_model-ad15dfab0430b72015d13d19b8a696bb9bacd0a6.tar.gz | |
Concat takes variadic inputs
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: Ic8fe6e1fd899b41d444fd4f477d0f515ce0e9cc9
Diffstat (limited to 'reference_model/src/ops/data_layout.h')
| -rw-r--r-- | reference_model/src/ops/data_layout.h | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/reference_model/src/ops/data_layout.h b/reference_model/src/ops/data_layout.h index 100bd6b..b180b4f 100644 --- a/reference_model/src/ops/data_layout.h +++ b/reference_model/src/ops/data_layout.h @@ -40,8 +40,7 @@ public: protected: Eigen::array<int, Rank> reverser; - TosaReference::TensorTemplate<TIn>* lhs; - TosaReference::TensorTemplate<TIn>* rhs; + std::vector<TosaReference::TensorTemplate<TIn>*> ins; TosaAxisAttribute* attribute; TosaReference::TensorTemplate<TOut>* out; }; |