Struct Df64QrVjp 

pub struct Df64QrVjp {
    pub has_q: bool,
    pub has_r: bool,
}

Reverse-mode adjoint of the reduced QR factorization.

The adjoint is a numerical body of its own: it needs a triangular solve against the primal factor, so it is an operation rather than a graph of preset operations, which a scalar tenferro does not declare could not execute anyway.

Examples

use tenferro_df64_proof::extension::Df64QrVjp;
use tenferro_ad::extension::ExtensionOp;

// A loss that depends on the triangular factor alone supplies one cotangent.
let adjoint = Df64QrVjp::of(false, true);
assert_eq!(<Df64QrVjp as ExtensionOp>::input_count(&adjoint), 3);
assert_eq!(<Df64QrVjp as ExtensionOp>::output_count(&adjoint), 1);
assert_eq!(
    <Df64QrVjp as ExtensionOp>::input_count(&Df64QrVjp::of(true, true)),
    4
);

Fields

has_q: bool

Whether the caller supplied the factor’s cotangent.

has_r: bool

Whether the caller supplied the triangular factor’s cotangent.

Implementations

impl Df64QrVjp

pub const fn of(has_q: bool, has_r: bool) -> Self

Construct the adjoint for one cotangent availability.

Examples

use tenferro_df64_proof::extension::Df64QrVjp;

// Only the triangular factor carries a cotangent here.
let adjoint = Df64QrVjp::of(false, true);
assert!(!adjoint.has_q);
assert!(adjoint.has_r);

Trait Implementations

impl Clone for Df64QrVjp

fn clone(&self) -> Df64QrVjp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for Df64QrVjp

impl Debug for Df64QrVjp

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

Formats the value using the given formatter.

impl Default for Df64QrVjp

fn default() -> Df64QrVjp

Returns the “default value” for a type.

impl ExtensionOp for Df64QrVjp

fn family_id(&self) -> &'static str

Stable, process-independent family identifier.

fn payload_hash(&self, hasher: &mut dyn Hasher)

Hash the payload (everything except family_id).

fn payload_eq(&self, other: &dyn ExtensionOp) -> bool

Structural equality against another extension value.

fn clone_arc(&self) -> Arc<dyn ExtensionOp>

Deep-clone the payload behind an Arc.

fn as_any(&self) -> &dyn Any

Upcast this extension to &dyn Any for downcasting in payload_eq.

fn input_count(&self) -> usize

Number of primal inputs. MUST be constant for any given Arc<dyn ExtensionOp> value.

fn output_count(&self) -> usize

Number of outputs. MUST match the length of the vector returned by a successful Self::infer_output_meta call.

fn semantic_effects(&self) -> ExtensionEffectDeclaration<'_>

Declare observable semantic effects for this extension payload.

fn semantic_aliases(&self) -> ExtensionAliasDeclaration<'_>

Declare semantic output aliasing for this extension payload.

fn scalar_identity(&self) -> Option<&'static str>

Canonical identity of an externally defined scalar this operation carries.

fn infer_output_meta( &self, ctx: &mut ExtensionShapeContext<'_>, ) -> Result<Vec<(DType, Vec<SymDim>)>>

Infer output dtypes and shapes for each output slot.

fn lower_to_standard_ops( &self, _builder: &mut GraphBuilder<StdTensorOp>, _inputs: &[ValueRef<StdTensorOp>], _input_dtypes: &[DType], _input_shapes: &[&[SymDim]], ) -> Result<ExtensionStandardLowering, ExtensionLoweringError>

Try to expand this extension into standard tensor graph operations.

fn prune_outputs(&self, _live_outputs: &[bool]) -> Option<Arc<dyn ExtensionOp>>

Optionally return an equivalent op that produces only live outputs.