aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/ReshapeLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/reference/ReshapeLayer.cpp')
-rw-r--r--tests/validation/reference/ReshapeLayer.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/validation/reference/ReshapeLayer.cpp b/tests/validation/reference/ReshapeLayer.cpp
index 42f06e4f5a..85bf3fcb76 100644
--- a/tests/validation/reference/ReshapeLayer.cpp
+++ b/tests/validation/reference/ReshapeLayer.cpp
@@ -33,6 +33,7 @@ namespace validation
{
namespace reference
{
+/** [ReshapeLayer] **/
template <typename T>
SimpleTensor<T> reshape_layer(const SimpleTensor<T> &src, const TensorShape &output_shape)
{
@@ -51,6 +52,7 @@ template SimpleTensor<uint32_t> reshape_layer(const SimpleTensor<uint32_t> &src,
template SimpleTensor<int32_t> reshape_layer(const SimpleTensor<int32_t> &src, const TensorShape &output_shape);
template SimpleTensor<half> reshape_layer(const SimpleTensor<half> &src, const TensorShape &output_shape);
template SimpleTensor<float> reshape_layer(const SimpleTensor<float> &src, const TensorShape &output_shape);
+/** [ReshapeLayer] **/
} // namespace reference
} // namespace validation
} // namespace test