Trait PrimitiveBuilder 

pub trait PrimitiveBuilder<Op: GraphOperation> {
    // Required method
    fn add_primitive(
        &mut self,
        op: Op,
        inputs: Vec<PrimitiveValue<Op>>,
        role: OperationRole,
    ) -> Vec<LocalValueId> ;
}

Builder used by primitive JVP and transpose rules to append primitive applications.

Required Methods

fn add_primitive( &mut self, op: Op, inputs: Vec<PrimitiveValue<Op>>, role: OperationRole, ) -> Vec<LocalValueId>

Add one primitive application and return local ids for its outputs.

Implementors