Closed
Description
- Function Schema:
torch.ops.aten.amax.default
- Original PyTorch API:
https://pytorch.org/docs/stable/generated/torch.amax.html
- Relevant TensorRT Documentation:
cc: @zewenli98
torch.ops.aten.amax.default
https://pytorch.org/docs/stable/generated/torch.amax.html
cc: @zewenli98