Enable unrolling for the generated tanh kernel.
PiperOrigin-RevId: 317080498 Change-Id: Idaac4b2efee78cd2cd44c68ede4ac56f1a4cde40
This commit is contained in:
parent
de5620b74c
commit
16e6f9e792
|
@ -37,4 +37,5 @@ gen_kernel_library(
|
||||||
"f32",
|
"f32",
|
||||||
"f64",
|
"f64",
|
||||||
],
|
],
|
||||||
|
unroll_factors = "4",
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue