Skip to main content

saved_forward_values

Function saved_forward_values 

Source
pub fn saved_forward_values<Op: GraphOp>(
    op: &Op,
    input_aliases: &[GlobalValKey<Op>],
    inputs: &[EagerValue<Op>],
    outputs: &[Arc<Op::Operand>],
) -> HashMap<GlobalValKey<Op>, Arc<Op::Operand>>
Expand description

Build saved forward data for one eager op.

Inputs are saved under stable input aliases. Outputs are saved under the derived keys produced by replaying op with those aliases.

§Examples

let saved = tidu::saved_forward_values(&op, &input_aliases, &inputs, &outputs);