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); }