def forward(self, x):
        """
        Call the module

        Args:
            x (`torch.tensor`): The input tensor to apply dropout
        """
        if self.training and self.drop_prob > 0:
            return XDropout.apply(x, self.get_context())
        return x

        Call the module

        Args:
            x (`torch.tensor`): The input tensor to apply dropout
        
dpkd/transformers/src/transformers/models/deberta/modeling_deberta.py