diff options
author | 2018-07-23 17:44:16 -0700 | |
---|---|---|
committer | 2018-07-23 17:49:35 -0700 | |
commit | b553a232a7537ca23efb36d19b4d6f5198ff46d1 (patch) | |
tree | 9a35aa160d26f8251e1ce94c222c53a76292d05f | |
parent | 86f63c717a354f342d1b714420a04c85434bb282 (diff) |
Add larger inputs to conv2d benchmark for better coverage
PiperOrigin-RevId: 205754951
-rw-r--r-- | tensorflow/python/ops/conv2d_benchmark.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/ops/conv2d_benchmark.py b/tensorflow/python/ops/conv2d_benchmark.py index aacdaa7ad0..28111c2730 100644 --- a/tensorflow/python/ops/conv2d_benchmark.py +++ b/tensorflow/python/ops/conv2d_benchmark.py @@ -175,7 +175,8 @@ class Conv2DBenchmark(test.Benchmark): data_types = [dtypes.float32, dtypes.float16] data_formats = ["NHWC", "NCHW"] - in_channels = list(range(3, 16)) + in_channels = list(range(1, 10)) + list(range(10, 20, 2)) + list( + range(20, 33, 4)) out_channels = [4, 16, 32] hw_strides = [[2, 2]] paddings = ["VALID", "SAME"] |