From 4ed6bca4ae578e050d60aa96dbbdefd60f6d67c4 Mon Sep 17 00:00:00 2001 From: Mingyu Yin Date: Mon, 10 Aug 2020 19:05:42 +0800 Subject: dnn_backend_native_layer_mathunary: add round support Signed-off-by: Mingyu Yin Reviewed-by: Guo, Yejun --- tests/dnn/dnn-layer-mathunary-test.c | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tests') diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c index 0bfcdcbd85..e9235120f3 100644 --- a/tests/dnn/dnn-layer-mathunary-test.c +++ b/tests/dnn/dnn-layer-mathunary-test.c @@ -60,6 +60,8 @@ static float get_expected(float f, DNNMathUnaryOperation op) return ceil(f); case DMUO_FLOOR: return floor(f); + case DMUO_ROUND: + return round(f); default: av_assert0(!"not supported yet"); return 0.f; @@ -136,5 +138,7 @@ int main(int agrc, char **argv) return 1; if (test(DMUO_FLOOR)) return 1; + if (test(DMUO_ROUND)) + return 1; return 0; } -- cgit v1.2.3