2 ms·scaled_dot_product_attention isn’t CUDA specific, it even works on TPUs.by upbeat_general 2y agoscaled_dot_product_attention isn’t CUDA specific, it even works on TPUs.