Searched defs:forward_wrapper (Results 1 – 1 of 1) sorted by relevance
1106 self, backward_function, gradients_wrt_outputs, forward_wrapper): argument1181 def _shuffle_forward_outputs(self, forward_wrapper): argument