Enable unrolling for the generated tanh kernel.

PiperOrigin-RevId: 317080498
Change-Id: Idaac4b2efee78cd2cd44c68ede4ac56f1a4cde40
This commit is contained in:
Stephan Herhut 2020-06-18 05:00:40 -07:00 committed by TensorFlower Gardener
parent de5620b74c
commit 16e6f9e792
1 changed files with 1 additions and 0 deletions

View File

@ -37,4 +37,5 @@ gen_kernel_library(
"f32", "f32",
"f64", "f64",
], ],
unroll_factors = "4",
) )