Converting onnx to trt: [8] No importer registered for op: OneHot

francesco.ciannella · February 1, 2021, 8:45pm

Hi NVES,

thanks for your reply. I am using trtexec.
The onnx model checker works fine:

Python 3.6.9 (default, Oct  8 2020, 12:12:24) 
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import onnx
>>> import sys
>>> filename = "saved_model_dialog_nlu.onnx"
>>> model = onnx.load(filename)
>>> onnx.checker.check_model(model)
>>> 
>>> 
>>> 
>>> exit()

This is a large Bert model that I converted from keras to onnx and the conversion was successful.

You can pull the onnx model here:

Let me know if you need more information.

When I look at the table of supported operations for trtexec conversion I see this:

github.com

onnx/onnx-tensorrt/blob/main/docs/operators.md

<!--- SPDX-License-Identifier: Apache-2.0 -->

# Supported ONNX Operators

TensorRT 8.6 supports operators up to Opset 17. Latest information of ONNX operators can be found [here](https://github.com/onnx/onnx/blob/master/docs/Operators.md)

TensorRT supports the following ONNX data types: DOUBLE, FLOAT32, FLOAT16, INT8, and BOOL

> Note: There is limited support for INT32, INT64, and DOUBLE types. TensorRT will attempt to cast down INT64 to INT32 and DOUBLE down to FLOAT, clamping values to `+-INT_MAX` or `+-FLT_MAX` if necessary.

See below for the support matrix of ONNX operators in ONNX-TensorRT.

## Operator Support Matrix

| Operator                  | Supported  | Supported Types | Restrictions                                                                                                           |
|---------------------------|------------|-----------------|------------------------------------------------------------------------------------------------------------------------|
| Abs                       | Y          | FP32, FP16, INT32 |
| Acos                      | Y          | FP32, FP16 |
| Acosh                     | Y          | FP32, FP16 |
| Add                       | Y          | FP32, FP16, INT32 |

This file has been truncated. show original

Could that be the issue? Any idea how to workaround it?