diff options
-rw-r--r-- | reference_model/src/ops/data_layout.cc | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc index df7084d..49f53e8 100644 --- a/reference_model/src/ops/data_layout.cc +++ b/reference_model/src/ops/data_layout.cc @@ -497,11 +497,8 @@ int OpTileBase<Rank, Dtype>::checkTensorAttributes() for (int32_t d = 0; d < Rank; d++) { - if (in->getShape()[d] * attribute->multiples()[d] != out->getShape()[d]) - { - printNodeValidationError("unexpected output shape"); - return 1; - } + ERROR_IF(in->getShape()[d] * attribute->multiples()[d] != out->getShape()[d], + "Output shape not equal to input * multiples;") } return 0; |