class_docstr: Test for CombinedNMS TopK op in TF-TRT.
class_docstr_tok: ["Test","for","CombinedNMS","TopK","op","in","TF-TRT","."]
code: def GetParams(self):
      # Parameters
      batch_size = 1
      max_detetion_points = 2048
      num_classes = 90
      max_boxes_to_draw = 30
  
      # Inputs
      pre_nms_boxes_shape = [batch_size, max_detetion_points, 1, 4]
      pre_nms_scores_shape = [
          batch_size,
          max_detetion_points,
          num_classes,
      ]
  
      # Outputs
      nmsed_boxes_shape = [batch_size, max_boxes_to_draw, 4]
      nmsed_scores_shape = [batch_size, max_boxes_to_draw]
      nmsed_classes_shape = [batch_size, max_boxes_to_draw]
      valid_detections_shape = [batch_size]
  
      def _get_graph_fn(x, y):
          return self.GraphFn(
              x,
              y,
              max_boxes_to_draw=max_boxes_to_draw,
              max_detetion_points=max_detetion_points,
          )
  
      return self.BuildParams(
          _get_graph_fn,
          dtypes.float32,
          [pre_nms_boxes_shape, pre_nms_scores_shape],
          [
              nmsed_boxes_shape,
              nmsed_scores_shape,
              nmsed_classes_shape,
              valid_detections_shape,
          ],
      )
  
imports: import os
  from tensorflow.python.compiler.tensorrt import utils
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import image_ops_impl
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the CombinedNmsTestTopK obj's underlying member method code with example usage: obj.GetParams() and returns: self, self