Skip to content

↔ [Converter] Add support for amax in Torch-TensorRT #2095

Closed
@narendasan

Description

@narendasan

  • Function Schema:

torch.ops.aten.amax.default

  • Original PyTorch API:

https://pytorch.org/docs/stable/generated/torch.amax.html

  • Relevant TensorRT Documentation:

cc: @zewenli98

Alternatives

Additional context

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions