Fix tanh for large inputs

PiperOrigin-RevId: 671189741
Intel-tensorflow · Sep 5, 2024 · ff48a2f · ff48a2f
1 parent 5b1c4dd
commit ff48a2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/lite/delegates/gpu/common/tasks/elementwise.cc b/tensorflow/lite/delegates/gpu/common/tasks/elementwise.cc
@@ -144,8 +144,8 @@ std::string GetOneInputCode(const GpuInfo& gpu_info,
             "FLT4 exp_val = convert_half4(native_exp(2.0f * "
             "convert_float4($1)));\n";
         result +=
-            "$0 = ((exp_val - INIT_FLT4(1.0f)) / (exp_val + "
-            "INIT_FLT4(1.0f)));";
+            "$0 = isinf(exp_val) ? sign($1) : ((exp_val - INIT_FLT4(1.0f)) / "
+            "(exp_val + INIT_FLT4(1.0f)));\n";
       } else {
         result = "$0 = tanh($1);";
       }