code: def GetParams(self):
      # TODO(laigd): we should test the following cases:
      # - batch size is not changed, other dims are changing
      # - batch size is decreasing, other dims are identical
      # - batch size is decreasing, other dims are changing
      # - batch size is increasing, other dims are identical
      # - batch size is increasing, other dims are changing
      input_dims = [
          [[1, 5, 5, 1]],
          [[10, 5, 5, 1]],
          [[3, 5, 5, 1]],
          [[1, 5, 5, 1]],
          [[1, 3, 1, 1]],
          [[2, 9, 9, 1]],
          [[1, 224, 224, 1]],
          [[1, 128, 224, 1]],
      ]
      expected_output_dims = input_dims
  
      return trt_test.TfTrtIntegrationTestParams(
          graph_fn=self.GraphFn,
          input_specs=[
              tensor_spec.TensorSpec(
                  [None, None, None, 1], dtypes.float32, "input"
              )
          ],
          output_specs=[
              tensor_spec.TensorSpec(
                  [None, None, None, 1], dtypes.float32, "output"
              )
          ],
          input_dims=input_dims,
          expected_output_dims=expected_output_dims,
      )
  
file_path: tensorflow/tensorflow/python/compiler/tensorrt/test/dynamic_input_shapes_test.py
imports: import numpy
  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.framework import tensor_spec
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import nn
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the DynamicInputShapesTest obj's underlying member method code with example usage: obj.GetParams() and returns: trt_test