Thanks to visit codestin.com
Credit goes to github.com

Skip to content

↔ [Converter] Add support for Tensor reshape and permute operators in the FX aten path #1724

@gs-olive

Description

@gs-olive

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.

Metadata

Metadata

Assignees

Type

No type
No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions