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