Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix: use bool instead of uint8/byte in Deberta/DebertaV2/SEW-D to make it compatible with TensorRT #23683

Merged
merged 2 commits into from
May 24, 2023
Merged

fix: use bool instead of uint8/byte in Deberta/DebertaV2/SEW-D to make it compatible with TensorRT #23683

merged 2 commits into from
May 24, 2023

Commits on May 23, 2023

  1. Use bool instead of uint8/byte in DebertaV2 to make it compatible wit… 

    …h TensorRT

TensorRT cannot accept onnx graph with uint8/byte intermediate tensors. This PR uses bool tensors instead of unit8/byte tensors to make the exported onnx file can work with TensorRT.
    @uchuhimo
    uchuhimo authored May 23, 2023
    Configuration menu
    Copy the full SHA
    5cbd9e7 View commit details
    Browse the repository at this point in the history

Commits on May 24, 2023

  1. fix: use bool instead of uint8/byte in Deberta and SEW-D

    @yuxianq
    yuxianq committed May 24, 2023
    Configuration menu
    Copy the full SHA
    cbb2c9e View commit details
    Browse the repository at this point in the history