diff options
author | Ting Fu <ting.fu@intel.com> | 2020-06-29 22:54:04 +0800 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2020-07-06 12:45:14 +0800 |
commit | ea71e731f413bc581c20c0f963b13e8410dd3b6f (patch) | |
tree | c0d25dce331e1a15a2a9f990d1f64085a16f950e /libavfilter | |
parent | c5de77e33c9b573d27f5ba70a13caca12c32f4a8 (diff) | |
download | ffmpeg-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.c | 4 | ||||
-rw-r--r-- | libavfilter/dnn/dnn_backend_native_layer_mathunary.h | 1 |
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; |