* Add layers.MultiHeadAttention
* Update build/compute_output_signature argument checks
We do not definitively know which arguments are tensor arguments
at any given invocation (e.g. arguments with a None value may be tensor
arguments).
So rather than check that the build signature matches perfectly with
tensor call arguments, we will check the build signature arguments match
with some call argument.
cc053ac309