diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-31 16:14:03 -0700 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2021-09-15 23:58:06 +0000 |
| commit | 93a1628bc3dd48d9ba099de503b586a561b4751f (patch) | |
| tree | adab77805c3d78cf3b30b00684e8a76316e11477 /reference_model | |
| parent | e3d6a8ffe0fffaf9d29167b03509b85a2f4d8308 (diff) | |
| download | reference_model-93a1628bc3dd48d9ba099de503b586a561b4751f.tar.gz | |
Rename attribute: Pool2d, Conv2d, TransposeConv2d -> Pool, Conv, TransposeConv
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I466dd1dcf5230e8e07df202ba88515e775e04a1e
Diffstat (limited to 'reference_model')
| -rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 10 | ||||
| -rw-r--r-- | reference_model/src/ops/tensor_ops.h | 10 |
2 files changed, 10 insertions, 10 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 045c0a5..a150656 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -78,7 +78,7 @@ OpAvgPool2d<Dtype>::OpAvgPool2d(SubgraphTraverser* sgt_, setRequiredOperands(1, 1); setRequiredRank(4); - INIT_ATTRIBUTE(Pool2d); + INIT_ATTRIBUTE(Pool); INIT_QINFO(Unary); } @@ -299,7 +299,7 @@ OpConv2d<InDtype, WeightDtype>::OpConv2d(SubgraphTraverser* sgt_, setRequiredOperands(3, 1); setRequiredRank(4); - INIT_ATTRIBUTE(Conv2d); + INIT_ATTRIBUTE(Conv); INIT_QINFO(Conv); } @@ -491,7 +491,7 @@ OpDepthwiseConv2d<InDtype, WeightDtype>::OpDepthwiseConv2d(SubgraphTraverser* sg setRequiredOperands(3, 1); setRequiredRank(4); - INIT_ATTRIBUTE(Conv2d); + INIT_ATTRIBUTE(Conv); INIT_QINFO(Conv); } @@ -892,7 +892,7 @@ OpMaxPool2d<Dtype>::OpMaxPool2d(SubgraphTraverser* sgt_, setRequiredOperands(1, 1); setRequiredRank(4); - INIT_ATTRIBUTE(Pool2d); + INIT_ATTRIBUTE(Pool); } template <DType Dtype> @@ -1034,7 +1034,7 @@ OpTransposeConv2d<InDtype, OutDtype>::OpTransposeConv2d(SubgraphTraverser* sgt_, setRequiredOperands(3, 1); setRequiredRank(4); - INIT_ATTRIBUTE(TransposeConv2d); + INIT_ATTRIBUTE(TransposeConv); INIT_QINFO(Conv); } 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; }; |