def call(self, input_tensor, attention_mask, head_mask, output_attentions, training=False):
    self_outputs = self.self_attention(input_tensor, attention_mask, head_mask, output_attentions, training=training)
    attention_output = self.dense_output(self_outputs[0], input_tensor, training=training)
    outputs = ((attention_output,) + self_outputs[1:])
    return outputs
def call(self, input_tensor, attention_mask, head_mask, output_attentions, training=False):
    self_outputs = self.self_attention(input_tensor, attention_mask, head_mask, output_attentions, training=training)
    attention_output = self.dense_output(self_outputs[0], input_tensor, training=training)
    outputs = ((attention_output,) + self_outputs[1:])
    return outputs