diff options
Diffstat (limited to 'reference_model/src/ops/data_layout.h')
| -rw-r--r-- | reference_model/src/ops/data_layout.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/reference_model/src/ops/data_layout.h b/reference_model/src/ops/data_layout.h index 9f44fc7..bad88e4 100644 --- a/reference_model/src/ops/data_layout.h +++ b/reference_model/src/ops/data_layout.h @@ -63,8 +63,8 @@ protected: Eigen::array<std::pair<ptrdiff_t, ptrdiff_t>, Rank> paddings_array; TosaReference::TensorTemplate<TIn>* in; TosaReference::TensorTemplate<TOut>* out; - TosaReference::TensorTemplate<Eigen::Tensor<int32_t, 2>>* paddings; TosaPadQuantInfo* qinfo; + TosaPadAttribute* attribute; }; template <int InRank, int OutRank, DType Dtype> @@ -207,8 +207,8 @@ public: protected: Eigen::array<int, Rank> perm_array; + TosaTransposeAttribute* attribute; TosaReference::TensorTemplate<TIn>* in; - TosaReference::TensorTemplate<ETensor1<int32_t>>* perm_tensor; TosaReference::TensorTemplate<TOut>* out; }; }; // namespace TosaReference |