Skip to content

Refactor forwardDims / forwardDtype

We currently have forwardDtype and forwardDims that both propagate information toward the graph.

The idea of this issue is to propose to refactor these functions in one function forwardInfo. Forward info would do the same loop logic as the other two functions and would call a lambda passed by argument on each nodes.

This function would take as an input a list of lambdas so that we can propagate in one go every informations.

Of course, this function should be called by forwardDType and forwardDims to maintain compatibility.