TensorRT version: 8.5.2 With JetPack 5.1.2 in AGX Orin does not support TopK from ONNX

[chenscottus]

TensorRT version: 8.5.2 With JetPack 5.1.2 in AGX Orin does not support TopK from ONNX
:
[11/16/2023-10:31:19] [E] [TRT] ModelImporter.cpp:726: While parsing node number 230 [TopK → “/extractor/TopK_output_0”]:
[11/16/2023-10:31:19] [E] [TRT] ModelImporter.cpp:727: — Begin node —
[11/16/2023-10:31:19] [E] [TRT] ModelImporter.cpp:728: input: “/extractor/GatherND_1_output_0”
input: “/extractor/Reshape_4_output_0”
output: “/extractor/TopK_output_0”
output: “/extractor/TopK_output_1”
name: “/extractor/TopK”
op_type: “TopK”
attribute {
name: “axis”
i: 0
type: INT
}
attribute {
name: “largest”
i: 1
type: INT
}
attribute {
name: “sorted”
i: 1
type: INT
}

The onnx is https://github.com/fabio-sim/LightGlue-ONNX/releases/download/v0.1.3/superpoint_2048_lightglue_end2end.onnx

In Windows, TensorRT 8.6.1.6 has no such problem.

Thanks!

Scott

[henbucuoshanghai]

trt 8.6 version support topk op? i dont thik so

[lix19937]

8.6 support topk.