Skip to content

↔ [Converter] Add support for Tensor slicing and selection operators in the FX aten path #1714

Closed
@gs-olive

Description

@gs-olive

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.

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions