aboutsummaryrefslogtreecommitdiffstats
path: root/libavfilter
diff options
context:
space:
mode:
authorTing Fu <ting.fu@intel.com>2020-06-29 22:54:04 +0800
committerGuo, Yejun <yejun.guo@intel.com>2020-07-06 12:45:14 +0800
commitea71e731f413bc581c20c0f963b13e8410dd3b6f (patch)
treec0d25dce331e1a15a2a9f990d1f64085a16f950e /libavfilter
parentc5de77e33c9b573d27f5ba70a13caca12c32f4a8 (diff)
downloadffmpeg-ea71e731f413bc581c20c0f963b13e8410dd3b6f.tar.gz
dnn_backend_native_layer_mathunary: add tanh support
Signed-off-by: Ting Fu <ting.fu@intel.com>
Diffstat (limited to 'libavfilter')
-rw-r--r--libavfilter/dnn/dnn_backend_native_layer_mathunary.c4
-rw-r--r--libavfilter/dnn/dnn_backend_native_layer_mathunary.h1
2 files changed, 5 insertions, 0 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layer_mathunary.c b/libavfilter/dnn/dnn_backend_native_layer_mathunary.c
index ddb70996e7..ccdbcc21e0 100644
--- a/libavfilter/dnn/dnn_backend_native_layer_mathunary.c
+++ b/libavfilter/dnn/dnn_backend_native_layer_mathunary.c
@@ -112,6 +112,10 @@ int dnn_execute_layer_math_unary(DnnOperand *operands, const int32_t *input_oper
for (int i = 0; i < dims_count; ++i)
dst[i] = cosh(src[i]);
return 0;
+ case DMUO_TANH:
+ for (int i = 0; i < dims_count; ++i)
+ dst[i] = tanh(src[i]);
+ return 0;
default:
return -1;
}
diff --git a/libavfilter/dnn/dnn_backend_native_layer_mathunary.h b/libavfilter/dnn/dnn_backend_native_layer_mathunary.h
index 5a486b4f5f..ae0c1e1cdd 100644
--- a/libavfilter/dnn/dnn_backend_native_layer_mathunary.h
+++ b/libavfilter/dnn/dnn_backend_native_layer_mathunary.h
@@ -39,6 +39,7 @@ typedef enum {
DMUO_ATAN = 6,
DMUO_SINH = 7,
DMUO_COSH = 8,
+ DMUO_TANH = 9,
DMUO_COUNT
} DNNMathUnaryOperation;