aten.slice, aten.select, aten.expand
-
Function Schema:
torch.ops.aten.expand.default: ((torch.int64,), {})
torch.ops.aten.slice.Tensor: ((torch.float32,), {})
torch.ops.aten.slice.Tensor: ((torch.int32,), {})
torch.ops.aten.slice.Tensor: ((torch.float32,), {})
torch.ops.aten.slice.Tensor: ((torch.int32,), {})
torch.ops.aten.select.int: ((torch.float32,), {})
-
Original PyTorch API: torch.expand, torch.slice, torch.select
-
Relevant TensorRT Documentation: IGatherLayer, IShuffleLayer
Add support for slice, select, and expand as aten converters.