Skip to main content

topo_sort_grad_dag

tidu::backward

Function topo_sort_grad_dag 

Source 
pub fn topo_sort_grad_dag<Op: GraphOp>(
    output_node: &Option<Arc<GradNode<Op>>>,
) -> Vec<Arc<GradNode<Op>>>
Expand description

Topologically sort the reachable grad DAG in dependency-first order.