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