aten.unsqueeze, aten.reshape, aten.permute, aten.transpose
-
Function Schema:
torch.ops.aten.unsqueeze.default: ((torch.int32,), {})
torch.ops.aten.reshape: ((torch.float32,), {})
torch.ops.aten.permute.default: ((torch.float32,), {})
torch.ops.aten.transpose.int: ((torch.float32,), {})
-
Original PyTorch API: torch.unsqueeze, torch.reshape, torch.permute, torch.transpose
-
Relevant TensorRT Documentation: IGatherLayer, IShuffleLayer
Add support for unsqueeze, reshape, permute, and transpose as aten converters.