Browse Source

unaryop tanh vulkan

tags/20191113
nihuini 6 years ago
parent
commit
80f898b079
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      src/layer/vulkan/shader/unaryop.comp
  2. +1
    -0
      src/layer/vulkan/shader/unaryop_pack4.comp

+ 1
- 0
src/layer/vulkan/shader/unaryop.comp View File

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

+ 1
- 0
src/layer/vulkan/shader/unaryop_pack4.comp View File

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

Loading…
Cancel
Save