BladeDISC icon indicating copy to clipboard operation
BladeDISC copied to clipboard

In TF_Blade, we found TRT_SUPPORT_LIST is not match the Onnx-Tensorrt Op Support List.

Open b4b4o opened this issue 2 years ago • 0 comments

In TF_Blade, we found TRT_SUPPORT_LIST is not match the Onnx-Tensorrt Op Support List. In our opinion, this will lead to the TRT Engine Convert Failed. Because some Op(like 'FusedBatchNormV3') is seemed support in TRT_SUPPORT_LIST, so it will pass the graph partition. But in the convertion, it's not in Onnx-Tensorrt Op Support List, so it should't sucessfully converted.

Is our understanding correct? @wyzero @tanyokwok @Yancey1989

TRT_SUPPORT_LIST: https://github.com/alibaba/BladeDISC/blob/main/tensorflow_blade/tf_blade/util/tf_conversion_util.py#L20 Onnx-Tensorrt Op Support List: https://github.com/onnx/onnx-tensorrt/blob/main/docs/operators.md

b4b4o avatar Oct 10 '23 11:10 b4b4o