diff options
| -rw-r--r-- | reference_model/src/ops/data_layout.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc index f3e80f3..674227b 100644 --- a/reference_model/src/ops/data_layout.cc +++ b/reference_model/src/ops/data_layout.cc @@ -89,7 +89,7 @@ int OpConcat<Rank, Dtype>::checkTensorAttributes() } } - ERROR_IF(output_dim_on_axis == outputs[0]->getShape()[attribute->axis()], + ERROR_IF(output_dim_on_axis != outputs[0]->getShape()[attribute->axis()], "OpConcat: sum of input dimension on axis not equal to output dimension on axis"); out = dynamic_cast<TosaReference::TensorTemplate<TOut>*>(outputs[0]); |