experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[MLIR][KernelGen] Remove unroll factor when 1

Browse Source 
Skip loop unrolling logic entirely.

PiperOrigin-RevId: 346968086
Change-Id: I141697c1a43a4f64bd84d61994f2a29a1fd6a15f
This commit is contained in:
A. Unique TensorFlower 2020-12-11 03:33:55 -08:00 committed by TensorFlower Gardener
parent 32d5b268e2
commit 4e082d53bf
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/mlir_generated/BUILD
View File

@ -315,7 +315,6 @@ gen_kernel_library(
"f32",
"f64",
],
unroll_factors = "1",
)
gen_kernel_library(
@ -349,7 +348,6 @@ gen_kernel_library(
generate_unranked = True,
tile_size = "256",
types = ["i1"],
unroll_factors = "1",
)
gen_kernel_library(
@ -360,7 +358,6 @@ gen_kernel_library(
"f32",
"f64",
],
unroll_factors = "1",
)
gen_kernel_library(
@ -572,7 +569,6 @@ gen_kernel_library(
"f32",
"f64",
],
unroll_factors = "1",
)
for name in [
"cos",