torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

[WIP] fix various converters for dynamic shapes

Open jaybdub opened this issue 3 years ago • 0 comments

  • adds shape tracking with wrapper classes

Fixed:

  • [x] flatten
  • [x] squeeze
  • [x] unsqueeze
  • [x] view
  • [x] reshape
  • [x] interpolate
  • [x] getitem
  • [ ] chunk
  • [ ] split

Chunk / split can return variable number of outputs for dynamic shapes. They currently don't work with ONNX in this context. A good default behavior in this instance needs to be defined, but for now can probably be omitted as not supported by dynamic shapes.

jaybdub avatar Jul 29 '22 03:07 jaybdub