code: def tensor_shapes(self):
      return [
          [1, 512, 1, 1],
          [64, 2, 2, 2],
          [32, 4, 2, 2],
          [16, 8, 2, 2],
      ]
  
imports: import unittest
  from tensorflow.core.framework import attr_value_pb2
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the MaxBatchSizesTestBase obj's underlying member method code with example usage: obj.tensor_shapes() and returns: list