Enable unrolling for the generated tanh kernel.

PiperOrigin-RevId: 317080498 Change-Id: Idaac4b2efee78cd2cd44c68ede4ac56f1a4cde40
2020-06-18 05:00:40 -07:00 · 2020-06-18 05:00:40 -07:00 · 16e6f9e792
parent de5620b74c
commit 16e6f9e792
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/core/kernels/cubin_headers/BUILD
+++ b/tensorflow/core/kernels/cubin_headers/BUILD
@ -37,4 +37,5 @@ gen_kernel_library(
        "f32",
        "f64",
    ],
+    unroll_factors = "4",
 )