Remap the outputs of the subgraph in-place.
(self, new_output_indices)
| 301 | self._input_ts = [self._input_ts[i] for i in new_input_indices] |
| 302 | |
| 303 | def _remap_outputs(self, new_output_indices): |
| 304 | """Remap the outputs of the subgraph in-place.""" |
| 305 | new_output_indices = _finalize_indices(new_output_indices, self._output_ts) |
| 306 | _check_within_range( |
| 307 | new_output_indices, len(self._output_ts), repetition=True) |
| 308 | self._output_ts = [self._output_ts[i] for i in new_output_indices] |
| 309 | |
| 310 | def _remap_outputs_make_unique(self): |
| 311 | """Remap the outputs in place so that all the tensors appears only once.""" |
no test coverage detected