diff options
Diffstat (limited to 'tests/validation_new/Helpers.h')
| -rw-r--r-- | tests/validation_new/Helpers.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/validation_new/Helpers.h b/tests/validation_new/Helpers.h index 3058b8eaee..30959161bb 100644 --- a/tests/validation_new/Helpers.h +++ b/tests/validation_new/Helpers.h @@ -128,6 +128,14 @@ std::pair<T, T> get_activation_layer_test_bounds(ActivationLayerInfo::Activation return bounds; } + +/** Calculate output tensor shape give a vector of input tensor to concatenate + * + * @param[in] input_shapes Shapes of the tensors to concatenate across depth. + * + * @return The shape of output concatenated tensor. + */ +TensorShape calculate_depth_concatenate_shape(const std::vector<TensorShape> &input_shapes); } // namespace validation } // namespace test } // namespace arm_compute |