diff options
| author |  Michael Kuperstein <mkuper@google.com> | 2018-09-17 12:57:26 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-17 13:02:15 -0700 |
| commit | de3fa499bb05c595f9e341c7d507b47b8d44ae90 (patch) | |
| tree | 8da722e0b8f6568f5daa18e5dc8705d2884ad20c | |
| parent | adae337d05251963ef0905e024dfdc07b6d0aae2 (diff) | |
[XLA] Add ReduceWindow test.
PiperOrigin-RevId: 213322116
| -rw-r--r-- | tensorflow/compiler/xla/tests/reduce_window_test.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/tests/reduce_window_test.cc b/tensorflow/compiler/xla/tests/reduce_window_test.cc index 63491a90bf..c25ccafaf8 100644 --- a/tensorflow/compiler/xla/tests/reduce_window_test.cc +++ b/tensorflow/compiler/xla/tests/reduce_window_test.cc @@ -1303,11 +1303,19 @@ struct R1ReduceWindowTestData { /*pad_high=*/{0}, /*reducer=*/Reducer::kAdd}, + // The pattern generated by inclusive scan (cumsum/cumprod). {/*base_bounds=*/{4096}, /*window_bounds=*/{4096}, /*strides=*/{1}, /*pad_low=*/{4095}, /*pad_high=*/{0}, /*reducer=*/Reducer::kMax}, + + // The pattern generated by exclusive scan (cumsum/cumprod). + {/*base_bounds=*/{4096}, /*window_bounds=*/{4096}, + /*strides=*/{1}, + /*pad_low=*/{4096}, + /*pad_high=*/{0}, + /*reducer=*/Reducer::kMax}, }; string R1ReduceWindowTestDataToString( |