TensorRT
TensorRT copied to clipboard
refactor: remove the code that covers the no output segmented block
Signed-off-by: Bo Wang [email protected]
Description
Before this feature https://github.com/pytorch/TensorRT/issues/1031 is supported, there might be some cases that TensorRT segmented block has NonTensor input, so we go through the nodes' output one by one to determine the output for this block if there is no output for this block. However, this this PR merged https://github.com/pytorch/TensorRT/pull/1140, we can ensure that every TensorRT segmented block has Tensor input, which means there will never be no output for segmented block.
Therefore, we can erase these lines.
support #1121
Checklist:
- [x] My code follows the style guidelines of this project (You can use the linters)
- [x] I have performed a self-review of my own code
- [x] I have commented my code, particularly in hard-to-understand areas and hacks
- [ ] I have made corresponding changes to the documentation
- [ ] I have added tests to verify my fix or my feature
- [ ] New and existing unit tests pass locally with my changes
- [ ] I have added the relevant labels to my PR in so that relevant reviewers are notified