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
) => { ... };
($queries:expr, $keys:expr, $values:expr, $scale:expr, sinks = $sinks:expr) => { ... };
(
$queries:expr, $keys:expr, $values:expr, $scale:expr, sinks =
$sinks:expr, stream = $stream:expr
) => { ... };
(
$queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
$mask:expr, sinks = $sinks:expr
) => { ... };
(
$queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
$mask:expr, sinks = $sinks:expr, stream = $stream:expr
) => { ... };
(
$queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
$mask:expr, sinks = $sinks:expr
) => { ... };
(
$queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
$mask:expr, sinks = $sinks:expr, stream = $stream:expr
) => { ... };
}👎Deprecated since 0.26.0:
use with_stream or with_device around scaled_dot_product_attention
Expand description
Macro generated for the function crate::fast::scaled_dot_product_attention. See the function documentation for more details.