Enum LinalgAdRoute 

pub enum LinalgAdRoute {
    Unsupported,
    Linearize,
    LinearizeThenTranspose,
    LinearizeThenCustomLinearTranspose,
    CustomVjp,
    CustomPreferredWithLinearizeFallback,
}

Implementation route used for a user-visible AD mode.

Examples

use tenferro_linalg::{linalg_ad_support, LinalgAdOpKind, LinalgAdRoute};

let svd = linalg_ad_support(LinalgAdOpKind::Svd);
assert_eq!(svd.vjp.route, LinalgAdRoute::LinearizeThenTranspose);

Variants

Unsupported

No supported route exists.

Linearize

The mode is emitted directly by the operation’s linearize rule.

LinearizeThenTranspose

Reverse mode is supported by linearizing the primal graph and then transposing that linear graph.

LinearizeThenCustomLinearTranspose

Reverse mode is supported by linearizing the primal graph and using a custom transposed-linear rule for the emitted linear operation.

CustomVjp

Reverse mode is emitted by a direct operation-specific primal VJP rule.

CustomPreferredWithLinearizeFallback

A custom VJP is preferred, with the canonical linearize-then-transpose route retained as an intentional not-applicable fallback.

Trait Implementations

impl Clone for LinalgAdRoute

fn clone(&self) -> LinalgAdRoute

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for LinalgAdRoute

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for LinalgAdRoute

fn eq(&self, other: &LinalgAdRoute) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for LinalgAdRoute

impl Eq for LinalgAdRoute

impl StructuralPartialEq for LinalgAdRoute