cc053ac309
* 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. |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| additive_attention_test.py | ||
| additive_attention.py | ||
| attention_test.py | ||
| attention.py | ||
| multi_head_attention_test.py | ||
| multi_head_attention.py | ||