diff options
Diffstat (limited to 'tests/validation/ReferenceCPP.h')
| -rw-r--r-- | tests/validation/ReferenceCPP.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/validation/ReferenceCPP.h b/tests/validation/ReferenceCPP.h index 10e5ab6bc6..d3c77a2243 100644 --- a/tests/validation/ReferenceCPP.h +++ b/tests/validation/ReferenceCPP.h @@ -28,6 +28,7 @@ #include "RawTensor.h" +#include <memory> #include <ostream> #include <vector> @@ -262,6 +263,12 @@ public: * @param[in] conv_info Pads and strides information for the convolution layer. */ static void convolution_layer(const RawTensor &src, const RawTensor &weights, const RawTensor &bias, RawTensor &dst, const PadStrideInfo &conv_info); + /** Depth concatenate layer from @p srcs to @p dst + * + * @param[in] srcs Input tensors. + * @param[out] dst Result tensor. + */ + static void depth_concatenate_layer(const std::vector<std::unique_ptr<RawTensor>> &srcs, RawTensor &dst); /** Fully connected layer function * * @param[in] src Input tensor |