fix: fix the error that collection input segmented into trt subgraph#1225

narendasan merged 1 commit intosquashed_collectionspytorch/TensorRT:squashed_collectionsfrom

fix_collection_partitioningpytorch/TensorRT:fix_collection_partitioningCopy head branch name to clipboard