✨[Feature] Handling Scalar type outputs on the boundry between TensorRT and PyTorch subgraphs #2810
Closed
Description
Is your feature request related to a problem? Please describe.
See #2752 for an example. Some operators return a scalar, if these are on the graph boundary there may be issues in the runtime.
Describe the solution you'd like
Describe alternatives you've considered
Additional context