Struct AutoDiffAttrs

pub struct AutoDiffAttrs {
    pub mode: DiffMode,
    pub ret_activity: DiffActivity,
    pub input_activity: Vec<DiffActivity>,
}

Fields

mode: DiffMode

Conceptually either forward or reverse mode AD, as described in various autodiff papers and e.g. in the JAX Documentation.

ret_activity: DiffActivity

input_activity: Vec<DiffActivity>

Implementations

impl AutoDiffAttrs

pub fn has_ret_activity(&self) -> bool

pub fn has_active_only_ret(&self) -> bool

pub fn error() -> Self

pub fn source() -> Self

pub fn is_active(&self) -> bool

pub fn is_source(&self) -> bool

pub fn apply_autodiff(&self) -> bool

pub fn into_item( self, source: String, target: String, inputs: Vec<TypeTree>, output: TypeTree, ) -> AutoDiffItem

Trait Implementations

impl Clone for AutoDiffAttrs

fn clone(&self) -> AutoDiffAttrs

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for AutoDiffAttrs

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

Formats the value using the given formatter.

impl<__D: SpanDecoder> Decodable<__D> for AutoDiffAttrs

fn decode(__decoder: &mut __D) -> Self

impl<__E: SpanEncoder> Encodable<__E> for AutoDiffAttrs

fn encode(&self, __encoder: &mut __E)

impl<__CTX> HashStable<__CTX> for AutoDiffAttrs

where __CTX: HashStableContext,

fn hash_stable(&self, __hcx: &mut __CTX, __hasher: &mut StableHasher)

impl PartialEq for AutoDiffAttrs

fn eq(&self, other: &AutoDiffAttrs) -> 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 Eq for AutoDiffAttrs

impl StructuralPartialEq for AutoDiffAttrs

Layout

Note: Most layout information is completely unstable and may even differ between compilations. The only exception is types with certain repr(...) attributes. Please see the Rust Reference's “Type Layout” chapter for details on type layout guarantees.

Size: 32 bytes