diff options
author | 2018-08-16 15:12:02 -0700 | |
---|---|---|
committer | 2018-08-16 15:18:10 -0700 | |
commit | fb10df08e8a53cfae51dd693d40504ff6179b633 (patch) | |
tree | c10c3380af84465fc1b12fe8d4ddc89f30bbb3bf | |
parent | f35866577cb8d27a51b31f49f472e926c4359e1f (diff) |
Add a ReduceWindow test with a large window.
PiperOrigin-RevId: 209056517
-rw-r--r-- | tensorflow/compiler/xla/tests/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/compiler/xla/tests/reduce_window_test.cc | 6 |
2 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/tests/BUILD b/tensorflow/compiler/xla/tests/BUILD index 76f2e519ae..49c7c680fc 100644 --- a/tensorflow/compiler/xla/tests/BUILD +++ b/tensorflow/compiler/xla/tests/BUILD @@ -1156,6 +1156,7 @@ xla_test( name = "reduce_window_test", timeout = "long", srcs = [], + shard_count = 20, tags = [ "enable_for_xla_interpreter", "optonly", diff --git a/tensorflow/compiler/xla/tests/reduce_window_test.cc b/tensorflow/compiler/xla/tests/reduce_window_test.cc index 92c93f08b2..cae029fd70 100644 --- a/tensorflow/compiler/xla/tests/reduce_window_test.cc +++ b/tensorflow/compiler/xla/tests/reduce_window_test.cc @@ -1261,6 +1261,12 @@ struct R1ReduceWindowTestData { /*pad_low=*/{5}, /*pad_high=*/{0}, /*reducer=*/Reducer::kAdd}, + + {/*base_bounds=*/{4096}, /*window_bounds=*/{4096}, + /*strides=*/{1}, + /*pad_low=*/{4095}, + /*pad_high=*/{0}, + /*reducer=*/Reducer::kMax}, }; string R1ReduceWindowTestDataToString( |