diff options
Diffstat (limited to 'reference_model/src/ops/tensor_ops.h')
-rw-r--r-- | reference_model/src/ops/tensor_ops.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/reference_model/src/ops/tensor_ops.h b/reference_model/src/ops/tensor_ops.h index 6ffc27d..eea351d 100644 --- a/reference_model/src/ops/tensor_ops.h +++ b/reference_model/src/ops/tensor_ops.h @@ -68,7 +68,7 @@ public: protected: TosaReference::TensorTemplate<TIn>* in; TosaReference::TensorTemplate<TOut>* out; - tosa::TosaPool2dAttribute* attribute; + tosa::TosaPoolAttribute* attribute; tosa::TosaUnaryQuantInfo* qinfo; protected: @@ -104,7 +104,7 @@ protected: TosaReference::TensorTemplate<TWeight>* weight; TosaReference::TensorTemplate<TBias>* bias; TosaReference::TensorTemplate<TAcc>* output; - tosa::TosaConv2dAttribute* attribute; + tosa::TosaConvAttribute* attribute; tosa::TosaConvQuantInfo* qinfo; }; @@ -136,7 +136,7 @@ protected: TosaReference::TensorTemplate<TWeight>* weight; TosaReference::TensorTemplate<TBias>* bias; TosaReference::TensorTemplate<TAcc>* output; - tosa::TosaConv2dAttribute* attribute; + tosa::TosaConvAttribute* attribute; tosa::TosaConvQuantInfo* qinfo; }; @@ -219,7 +219,7 @@ public: protected: TosaReference::TensorTemplate<TIn>* in; TosaReference::TensorTemplate<TOut>* out; - tosa::TosaPool2dAttribute* attribute; + tosa::TosaPoolAttribute* attribute; }; template <DType InDtype, DType WeightDtype> @@ -250,7 +250,7 @@ protected: TosaReference::TensorTemplate<TWeight>* weight; TosaReference::TensorTemplate<TBias>* bias; TosaReference::TensorTemplate<TAcc>* output; - TosaTransposeConv2dAttribute* attribute; + TosaTransposeConvAttribute* attribute; TosaConvQuantInfo* qinfo; }; |