src/transformers/models/gptj/modeling_tf_gptj.py/TFGPTJAttention/get_masked_bias
class TFGPTJAttention:    def get_masked_bias(dtype: tf.DType) -> tf.Tensor:
        return tf.cast(tf.constant(-1e9), dtype)