Trait auto_diff::op::OpCall

source · [−]

pub trait OpCall {
    fn call(&mut self, inputs: &[&Var]) -> Result<Vec<Var>, AutoDiffError>;
}

Expand description

Ops that first created, then called needs to follow this behavior.

Required methods

fn call(&mut self, inputs: &[&Var ]) -> Result<Vec<Var>, AutoDiffError>

Implementors

impl OpCall for ArgSort

impl OpCall for Conv2d

impl OpCall for Cat

impl OpCall for Chunk

impl OpCall for ConditionalSelect

impl OpCall for Gather

impl OpCall for IndexExclude

impl OpCall for IndexSelect

impl OpCall for Permute

impl OpCall for Repeat

impl OpCall for Reshape

impl OpCall for Split

impl OpCall for Squeeze

impl OpCall for Stack

impl OpCall for Take

impl OpCall for Unsqueeze

impl OpCall for BiLinear

impl OpCall for Linear

impl OpCall for Argmax

impl OpCall for Argmin

impl OpCall for Logsumexp

impl OpCall for Max

impl OpCall for Mean

impl OpCall for Min

impl OpCall for Prod

impl OpCall for Std

impl OpCall for Sum

impl OpCall for Variance

impl OpCall for View

impl OpCall for GetPatch