def METHOD_NAME(self, src_tokens, src_lengths):
    encoder_out: Dict[str, List[Tensor]] = self.conv_transformer_encoder(src_tokens, src_lengths.to(src_tokens.device))
    output = encoder_out["encoder_out"][0]
    encoder_padding_masks = encoder_out["encoder_padding_mask"]
    return {
        "encoder_out": [output],
        # This is because that in the original implementation
        # the output didn't consider the last segment as right context.
        "encoder_padding_mask": [encoder_padding_masks[0][:, : output.size(0)]] if len(encoder_padding_masks) > 0
        else [],
        "encoder_embedding": [],
        "encoder_states": [],
        "src_tokens": [],
        "src_lengths": [],
    }