Searched defs:forward_wrapper (Results 1 – 2 of 2) sorted by relevance
15 def forward_wrapper(fetcher): function
992 self, backward_function, gradients_wrt_outputs, forward_wrapper): argument1067 def _shuffle_forward_outputs(self, forward_wrapper): argument