Update on "[quant] ReflectionPad2d"

Differential Revision: [D25000347](https://our.internmc.facebook.com/intern/diff/D25000347) [ghstack-poisoned]
pytorch · Nov 16, 2020 · 7cc6b2e · 7cc6b2e
1 parent e1af264
commit 7cc6b2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/quantization/test_quantized_op.py b/test/quantization/test_quantized_op.py
@@ -4117,7 +4117,7 @@ def test_reflection_pad1d(self, batch_size, channels, width, qtype):
            width=st.integers(16, 128),
            qtype=st.sampled_from(hu._ALL_QINT_TYPES))
     def test_reflection_pad2d(self, batch_size, channels, height, width, qtype):
-        padding = width // 4
+        padding = (width // 4, width // 4, height // 4, height // 4)
 
         x = torch.arange(batch_size * channels * height * width).to(torch.float)
         x = x.resize(batch_size, channels, height, width)