Searched refs:num_doutputs (Results 1 – 1 of 1) sorted by relevance
668 def forward_backward(self, num_doutputs=None): argument670 if num_doutputs is None:671 num_doutputs = self._num_inference_outputs672 forward_backward = self._cached_function_pairs.get(num_doutputs)675 forward, backward = self._construct_forward_backward(num_doutputs)676 self._cached_function_pairs[num_doutputs] = (forward, backward)679 def _construct_forward_backward(self, num_doutputs): argument699 output for output in self._func_graph.outputs[:num_doutputs]