This repository has been archived by the owner on Jul 7, 2023. It is now read-only.
Enable depth_k != depth_v in local_attention_2d and masked_local_attention_2d#1899
Open
sgrigory wants to merge 1 commit intotensorflow:masterfrom sgrigory:attention-2d-depth-kv
+14-12
