pub fn backward_dag<Op: PrimitiveOp>(
sorted_nodes: &[Arc<GradNode<Op>>],
output_key: &GlobalValKey<Op>,
seed: Arc<Op::Operand>,
callbacks: &mut impl BackwardCallbacks<Op>,
ctx: &mut Op::ADContext,
) -> HashMap<GlobalValKey<Op>, Arc<Op::Operand>>where
Op::InputKey: ADKey,Expand description
Execute reverse-mode AD over a grad DAG.