diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-07-05 17:02:25 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:15:39 +0100 |
commit | ac4e873dad6aa6291fc36aff62047a896db04f6a (patch) | |
tree | 2c5397c6c04b97cedb88ff809f5d40cbe7dc79c9 /tests/validation/Reference.h | |
parent | df24618b53cffed1c574e11e9fd4ba7740f8c009 (diff) | |
download | ComputeLibrary-ac4e873dad6aa6291fc36aff62047a896db04f6a.tar.gz |
COMPMID-417: Port DepthConcatenate to QS8/QS16 for NEON/CL.
Change-Id: I3dddae63043c7aa18d908a4fc8abacf3c64f98ca
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/80081
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Steven Niu <steven.niu@arm.com>
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 |