This is a 2D attention module, which only returns the attention softmax (no multiplication by value)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected