src/transformers/models/deberta/modeling_tf_deberta.py/TFDebertaXSoftmax/call
class TFDebertaXSoftmax:    def call(self, inputs: tf.Tensor, mask: tf.Tensor):
        rmask = tf.logical_not(tf.cast(mask, tf.bool))
        output = tf.where(rmask, float("-inf"), inputs)
        output = stable_softmax(output, self.axis)
        output = tf.where(rmask, 0.0, output)
        return output