guides/understanding_masking_and_padding.py/TemporalSplit/compute_mask
class TemporalSplit:    def compute_mask(self, inputs, mask=None):
        # Also split the mask into 2 if it presents.
        if mask is None:
            return None
        return ops.split(mask, 2, axis=1)