Closed
Description
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.