Use py_utils.ApplyPadding

PiperOrigin-RevId: 563449723
tensorflow · Sep 7, 2023 · 7f7e0ac · 7f7e0ac
1 parent c00a74b
commit 7f7e0ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lingvo/core/conv_layers_with_time_padding.py b/lingvo/core/conv_layers_with_time_padding.py
@@ -960,7 +960,7 @@ class PaddingLayer(base_layer.BaseLayer):
 
   def FProp(self, theta, inputs, paddings):
     paddings_expanded = tf.expand_dims(tf.expand_dims(paddings, -1), -1)
-    return inputs * (1.0 - paddings_expanded), paddings
+    return py_utils.ApplyPadding(paddings_expanded, inputs), paddings
 
 
 class GlobalPoolingLayer(base_layer.BaseLayer):