diff options
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]; |