def call(self, hidden_states, attention_mask, head_mask, output_attentions, training=False):
    attention_outputs = self.attention(hidden_states, attention_mask, head_mask, output_attentions, training=training)
    attention_output = attention_outputs[0]
    intermediate_output = self.intermediate(attention_output)
    layer_output = self.bert_output(intermediate_output, attention_output, training=training)
    outputs = ((layer_output,) + attention_outputs[1:])
    return outputs
def call(self, hidden_states, attention_mask, head_mask, output_attentions, training=False):
    attention_outputs = self.attention(hidden_states, attention_mask, head_mask, output_attentions, training=training)
    attention_output = attention_outputs[0]
    intermediate_output = self.intermediate(attention_output)
    layer_output = self.bert_output(intermediate_output, attention_output, training=training)
    outputs = ((layer_output,) + attention_outputs[1:])
    return outputs