message: Only run combined_nms_test for TRT5.1+
diff: @@ -18,6 +18,7 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  +from tensorflow.compiler.tf2tensorrt.wrap_py_utils import get_linked_tensorrt_version
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  @@ -87,5 +88,10 @@ class CombinedNmsTest(trt_test.TfTrtIntegrationTestBase):
  'max_output_size_per_class', 'max_total_size',
  'iou_threshold', 'score_threshold']}
  + def ShouldRunTest(self, run_params):
  + # Only run for TRT 5.1 and above.
  + ver = get_linked_tensorrt_version()
  + return ver[0] > 5 or (ver[0] == 5 and ver[1] >= 1)
  +
  if __name__ == '__main__':
  test.main()
  