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/src/ops/tensor_ops.cc | |
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/src/ops/tensor_ops.cc')
-rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 10 |
1 files changed, 5 insertions, 5 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); } |