conv_layers_with_time_padding: Fix Tensor comparison (similar to previous cl).
conv_layers_with_time_padding: Fix Tensor comparison (similar to previous cl).

PiperOrigin-RevId: 324754536

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo