Code

support argmax converter#2291

Code

Merged

gs-olive merged 4 commits intomainpytorch/TensorRT:mainfrom argmax_converter_dynamopytorch/TensorRT:argmax_converter_dynamoCopy head branch name to clipboardOct 10, 2023

+137-1

Commits

Commits on Oct 10, 2023

• support argmax converter

  Show description for 6581906
  

  bowang007

  authored and

  gs-olive

  committed
  6581906

• handle edge cases

  

  bowang007

  authored and

  gs-olive

  committed
  6e06377

• fix: Formatting + TRTTensor casting

  gs-olive

  committed
  756a41a

• fix: Bugfixes and review comments

  Show description for 668f897
  gs-olive

  committed
  668f897