diff options
Diffstat (limited to 'tests/validation/reference')
| -rw-r--r-- | tests/validation/reference/UpsampleLayer.cpp | 12 | ||||
| -rw-r--r-- | tests/validation/reference/UpsampleLayer.h | 2 |
2 files changed, 8 insertions, 6 deletions
diff --git a/tests/validation/reference/UpsampleLayer.cpp b/tests/validation/reference/UpsampleLayer.cpp index 3a340d0905..876f6d794a 100644 --- a/tests/validation/reference/UpsampleLayer.cpp +++ b/tests/validation/reference/UpsampleLayer.cpp @@ -35,10 +35,10 @@ namespace reference { template <typename T> SimpleTensor<T> upsample_layer(const SimpleTensor<T> &src, - const Size2D &info, const InterpolationPolicy upsampling_policy) + const Size2D &info, const InterpolationPolicy policy) { - ARM_COMPUTE_ERROR_ON(upsampling_policy != InterpolationPolicy::NEAREST_NEIGHBOR); - ARM_COMPUTE_UNUSED(upsampling_policy); + ARM_COMPUTE_ERROR_ON(policy != InterpolationPolicy::NEAREST_NEIGHBOR); + ARM_COMPUTE_UNUSED(policy); TensorShape output_shape = src.shape(); output_shape.set(0, src.shape().x() * info.x()); @@ -77,9 +77,11 @@ SimpleTensor<T> upsample_layer(const SimpleTensor<T> &src, } template SimpleTensor<float> upsample_layer(const SimpleTensor<float> &src, - const Size2D &info, const InterpolationPolicy upsampling_policy); + const Size2D &info, const InterpolationPolicy policy); template SimpleTensor<half> upsample_layer(const SimpleTensor<half> &src, - const Size2D &info, const InterpolationPolicy upsampling_policy); + const Size2D &info, const InterpolationPolicy policy); +template SimpleTensor<uint8_t> upsample_layer(const SimpleTensor<uint8_t> &src, + const Size2D &info, const InterpolationPolicy policy); } // namespace reference } // namespace validation } // namespace test diff --git a/tests/validation/reference/UpsampleLayer.h b/tests/validation/reference/UpsampleLayer.h index fc1da39186..ecb458a0c6 100644 --- a/tests/validation/reference/UpsampleLayer.h +++ b/tests/validation/reference/UpsampleLayer.h @@ -37,7 +37,7 @@ namespace reference { template <typename T> SimpleTensor<T> upsample_layer(const SimpleTensor<T> &src, - const Size2D &info, const InterpolationPolicy upsampling_policy); + const Size2D &info, const InterpolationPolicy policy); } // namespace reference } // namespace validation } // namespace test |