Bug fix in conv2d due to merging

Niki Parmar · Copybara-Service · commit 4b4a180c3c98 · 2018-10-23T14:01:29.000-07:00
PiperOrigin-RevId: 218402537
diff --git a/mesh_tensorflow/ops.py b/mesh_tensorflow/ops.py
@@ -4090,4 +4090,4 @@ def conv2d_with_blocks(
       else:
         conv_input = pad(
             conv_input, [halo_size, halo_size], block_size_dim.name)
-    return conv2d(conv_input, conv_filter, strides, "VALID", name)
+  return conv2d(conv_input, conv_filter, strides, "VALID", name)
