TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

✨[Feature] Support `torch.ops.aten.repeat.default` converter

Open dgcnz opened this issue 5 months ago • 0 comments

Tasks

  • [ ] torch_tensorrt.dynamo.conversion.impl.slice.ops.repeat
    • Reference: https://github.com/pytorch/TensorRT/blob/fa02fd3e3a85a6042d11a00cd386f6b69c1d6c4b/py/torch_tensorrt/dynamo/conversion/impl/slice/ops.py#L219-L324
  • [ ] torch_tensorrt.dynamo.conversion.aten_ops_converters.aten_ops_repeat https://github.com/pytorch/TensorRT/blob/fa02fd3e3a85a6042d11a00cd386f6b69c1d6c4b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py#L1152-L1172
  • [ ] Unit test, example: expand

dgcnz avatar Sep 21 '24 10:09 dgcnz