Closed HSQ79815 closed 1 year ago
TensorRT only supports INT32 as input data type. When input data type is INT64, the transform from onnx is unsupported, this pass changes input data type to INT32 and inserts a Cast node afer input node so that the transfom can run sucessfully.
Cast
TensorRT only supports INT32 as input data type. When input data type is INT64, the transform from onnx is unsupported, this pass changes input data type to INT32 and inserts a
Castnode afer input node so that the transfom can run sucessfully.