Rename attribute: Pool2d, Conv2d, TransposeConv2d -> Pool, Conv, TransposeConv
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I466dd1dcf5230e8e07df202ba88515e775e04a1e
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 @@
protected:
TosaReference::TensorTemplate<TIn>* in;
TosaReference::TensorTemplate<TOut>* out;
- tosa::TosaPool2dAttribute* attribute;
+ tosa::TosaPoolAttribute* attribute;
tosa::TosaUnaryQuantInfo* qinfo;
protected:
@@ -104,7 +104,7 @@
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 @@
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 @@
protected:
TosaReference::TensorTemplate<TIn>* in;
TosaReference::TensorTemplate<TOut>* out;
- tosa::TosaPool2dAttribute* attribute;
+ tosa::TosaPoolAttribute* attribute;
};
template <DType InDtype, DType WeightDtype>
@@ -250,7 +250,7 @@
TosaReference::TensorTemplate<TWeight>* weight;
TosaReference::TensorTemplate<TBias>* bias;
TosaReference::TensorTemplate<TAcc>* output;
- TosaTransposeConv2dAttribute* attribute;
+ TosaTransposeConvAttribute* attribute;
TosaConvQuantInfo* qinfo;
};