Yeah that don’t seems like a perfect solution, and with the in_chan, out_chan unknown as well.
However the in_chan, out_chan of grad_fn must can be read in backend level to enable performing forward.
Anyway thank you alban! Really nice of you to answer patiently!