Skip to content

Make MultiHeadAttention op return attention probabilities #1746

Make MultiHeadAttention op return attention probabilities

Make MultiHeadAttention op return attention probabilities #1746