diff options
| author | Suraj Sudhir <suraj.sudhir@arm.com> | 2024-04-16 16:14:36 -0700 |
|---|---|---|
| committer | Suraj Sudhir <suraj.sudhir@arm.com> | 2024-04-18 21:46:54 -0700 |
| commit | b5fcfc0cfeffeb3fe9f05c32dc678ab09b2cee31 (patch) | |
| tree | a4cf45d48fcc17db3d41d8663ffe8c05ef41624c /reference_model/src/ops/tensor_ops.cc | |
| parent | 7296e16351554a2c80daab18b2d69b44c67b807a (diff) | |
| download | reference_model-b5fcfc0cfeffeb3fe9f05c32dc678ab09b2cee31.tar.gz | |
[reference_model] Remove output_shape from transpose_conv2d
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
Change-Id: Ib2b95e73b226d64c4db5ad1ed22c123e04d7e6f9
Diffstat (limited to 'reference_model/src/ops/tensor_ops.cc')
| -rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 15 |
1 files changed, 0 insertions, 15 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index edc1793..f38f486 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -1958,12 +1958,6 @@ int OpTransposeConv2d<InDtype, WeightDtype, AccDtype, OutDtype>::checkTensorAttr return 1; } - if (attribute->output_shape().size() != 4) - { - printNodeValidationError("OpTransposeConv2d: illegal size for attribute output_shape"); - return 1; - } - for (int32_t i : attribute->stride()) { if (i < 1) @@ -1973,15 +1967,6 @@ int OpTransposeConv2d<InDtype, WeightDtype, AccDtype, OutDtype>::checkTensorAttr } } - for (int d = 0; d < 4; d++) - { - if (attribute->output_shape()[d] != this->output->getShape()[d]) - { - printNodeValidationError("OpTransposeConv2d: illegal size for attribute output_shape"); - return 1; - } - } - int32_t IH = input->getShape()[1]; int32_t IW = input->getShape()[2]; int32_t OH = output->getShape()[1]; |