Struct Df64Expand 

pub struct Df64Expand {
    pub shape: Box<[usize]>,
}

Broadcast a scalar external value to a declared shape.

The total sum’s adjoint needs to place the output cotangent back into the input’s shape, and a preset broadcast is not available for a scalar tenferro does not declare, so the contribution owns this operation too.

Examples

use tenferro_df64_proof::extension::Df64Expand;
use tenferro_ad::extension::ExtensionOp;

let expand = Df64Expand::new(vec![2, 3]);
assert_eq!(<Df64Expand as ExtensionOp>::family_id(&expand), "tenferro-df64-proof.df64_ops.v1");
assert_eq!(&*expand.shape, &[2, 3]);

Fields

shape: Box<[usize]>

Shape the scalar is broadcast to.

Implementations

impl Df64Expand

pub fn new(shape: Vec<usize>) -> Self

Construct the operation for one output shape.

Examples

use tenferro_df64_proof::extension::Df64Expand;

let expand = Df64Expand::new(vec![2, 3]);
assert_eq!(&*expand.shape, &[2, 3]);

Trait Implementations

impl Clone for Df64Expand

fn clone(&self) -> Df64Expand

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Df64Expand

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

Formats the value using the given formatter.

impl ExtensionOp for Df64Expand

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.