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

Skip to content

↔ [Converter] Add support for aten.matmul in the FX aten path #1709

@gs-olive

Description

@gs-olive

aten.matmul

  • Function Schema: torch.ops.aten.matmul: ((torch.float32, torch.float32), {})

  • Original PyTorch API: torch.matmul

  • Relevant TensorRT Documentation: IMatrixMultiplyLayer

Add support for matmul as an aten converter.

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