From 80f898b079a09552efdea993ebcd94fef22b9524 Mon Sep 17 00:00:00 2001 From: nihuini Date: Thu, 12 Sep 2019 17:45:49 +0800 Subject: [PATCH] unaryop tanh vulkan --- src/layer/vulkan/shader/unaryop.comp | 1 + src/layer/vulkan/shader/unaryop_pack4.comp | 1 + 2 files changed, 2 insertions(+) diff --git a/src/layer/vulkan/shader/unaryop.comp b/src/layer/vulkan/shader/unaryop.comp index 16ed63004..2d94f0515 100644 --- a/src/layer/vulkan/shader/unaryop.comp +++ b/src/layer/vulkan/shader/unaryop.comp @@ -69,6 +69,7 @@ void main() if (op_type == 13) res = acos(v); if (op_type == 14) res = atan(v); if (op_type == 15) res = afp(1.f) / v; + if (op_type == 16) res = tanh(v); bottom_top_blob_data[gi] = afp2sfp(res); } diff --git a/src/layer/vulkan/shader/unaryop_pack4.comp b/src/layer/vulkan/shader/unaryop_pack4.comp index ddfc1e0c4..1f14e1fe7 100644 --- a/src/layer/vulkan/shader/unaryop_pack4.comp +++ b/src/layer/vulkan/shader/unaryop_pack4.comp @@ -69,6 +69,7 @@ void main() if (op_type == 13) res = acos(v); if (op_type == 14) res = atan(v); if (op_type == 15) res = afp(1.f) / v; + if (op_type == 16) res = tanh(v); bottom_top_blob_data[gi] = afp2sfpvec4(res); }