diff options
Diffstat (limited to 'reference_model/src/ops/data_layout.cc')
| -rw-r--r-- | reference_model/src/ops/data_layout.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc index 3e3770e..3b8d13a 100644 --- a/reference_model/src/ops/data_layout.cc +++ b/reference_model/src/ops/data_layout.cc @@ -54,6 +54,8 @@ int OpConcat<Rank, Dtype>::checkTensorAttributes() } int32_t num_inputs = inputs.size(); + LEVEL_CHECK(num_inputs <= tosa_level.MAX_TENSOR_LIST_SIZE, + "num_inputs should be smaller than or equal to MAX_TENSOR_LIST_SIZE"); // output and input must be the same types and rank for (int32_t i = 0; i < num_inputs; i++) |