diff options
Diffstat (limited to 'tests/validation/reference')
| -rw-r--r-- | tests/validation/reference/ReshapeLayer.cpp | 2 | ||||
| -rw-r--r-- | tests/validation/reference/ReshapeLayer.h | 3 |
2 files changed, 5 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 diff --git a/tests/validation/reference/ReshapeLayer.h b/tests/validation/reference/ReshapeLayer.h index fc6c716d7a..9e42f80642 100644 --- a/tests/validation/reference/ReshapeLayer.h +++ b/tests/validation/reference/ReshapeLayer.h @@ -34,8 +34,11 @@ namespace validation { namespace reference { +/** [ReshapeLayer] **/ template <typename T> SimpleTensor<T> reshape_layer(const SimpleTensor<T> &src, const TensorShape &output_shape); +/** [ReshapeLayer] **/ + } // namespace reference } // namespace validation } // namespace test |