diff options
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; }; |