dot_product_attention

Auto-generated from flax_nnx_code_defs

Keras

API: keras.ops.nn.dot_product_attention
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.nn.attention.dot_product_attention
Strategy: Direct Mapping
Official Docs ↗