From 16e6f9e792741b98d27b4f1463057313d04acdd8 Mon Sep 17 00:00:00 2001 From: Stephan Herhut Date: Thu, 18 Jun 2020 05:00:40 -0700 Subject: [PATCH] Enable unrolling for the generated tanh kernel. PiperOrigin-RevId: 317080498 Change-Id: Idaac4b2efee78cd2cd44c68ede4ac56f1a4cde40 --- tensorflow/core/kernels/cubin_headers/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/core/kernels/cubin_headers/BUILD b/tensorflow/core/kernels/cubin_headers/BUILD index b8ba164fbc3..ec8b44050db 100644 --- 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", )