macro_rules! scaled_dot_product_attention {
($queries:expr, $keys:expr, $values:expr, $scale:expr) => { ... };
($queries:expr, $keys:expr, $values:expr, $scale:expr, stream = $stream:expr) => { ... };
($queries:expr, $keys:expr, $values:expr, $scale:expr, mask = $mask:expr) => { ... };
(
$queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
$mask:expr, stream = $stream:expr
) => { ... };
}
Expand description
Macro generated for the function crate::fast::scaled_dot_product_attention
. See the function documentation for more details.