Searched refs:new_output_indices (Results 1 – 2 of 2) sorted by relevance
303 def _remap_outputs(self, new_output_indices): argument305 new_output_indices = _finalize_indices(new_output_indices, self._output_ts)307 new_output_indices, len(self._output_ts), repetition=True)308 self._output_ts = [self._output_ts[i] for i in new_output_indices]386 def remap_outputs(self, new_output_indices): argument406 res._remap_outputs(new_output_indices) # pylint: disable=protected-access409 def remap(self, new_input_indices=None, new_output_indices=None): argument433 if new_output_indices is not None:434 res._remap_outputs(new_output_indices) # pylint: disable=protected-access
72 sgv = sgv.remap(new_input_indices=[], new_output_indices=[])84 sgv = sgv.remap_outputs(new_output_indices=[1]).remove_unused_ops()