diff options
Diffstat (limited to 'tests/validation/Reference.h')
| -rw-r--r-- | tests/validation/Reference.h | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/tests/validation/Reference.h b/tests/validation/Reference.h index cbdeb012ce..37a072b60a 100644 --- a/tests/validation/Reference.h +++ b/tests/validation/Reference.h @@ -26,7 +26,6 @@ #include "RawTensor.h" #include "Types.h" -#include <vector> #include <vector> @@ -285,7 +284,7 @@ public: * @return Computed raw tensor. */ static RawTensor compute_reference_batch_normalization_layer(const TensorShape &shape0, const TensorShape &shape1, DataType dt, float epsilon, int fixed_point_position = 0); - /** Compute reference pixel-wise multiplication + /** Compute reference convolution layer * * @param[in] input_shape Shape for the input tensor * @param[in] weights_shape Shape for the weights tensor @@ -299,6 +298,15 @@ public: */ static RawTensor compute_reference_convolution_layer(const TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, DataType dt, const PadStrideInfo &conv_info, int fixed_point_position); + /** Compute reference depth concatenation layer + * + * @param[in] shapes Input tensor shapes (All dimensions should match apart from DimZ) + * @param[in] dt Data type to use + * @param[in] fixed_point_position (Optional) Number of bits for the fractional part of the fixed point numbers + * + * @return Computed raw tensor. + */ + static RawTensor compute_reference_depth_concatenate_layer(const std::vector<TensorShape> &shapes, DataType dt, int fixed_point_position = 0); /** Compute reference for fully connected layer function * * @param[in] input_shape Shape for the input tensor |