code: def get_batch(self, names):
      cuda_stream = polygraphy.Stream()
      try:
          data = next(self.batches)
  
          cuda_data = []
          for input_tensor in data:
              device_array = polygraphy.DeviceArray(
                  shape=input_tensor.shape, dtype=input_tensor.dtype
              )
              device_array.copy_from(
                  host_buffer=input_tensor, stream=cuda_stream
              )
              cuda_data.append(device_array)
  
          return [input_tensor.ptr for input_tensor in cuda_data]
      except StopIteration:
          return None
  
file_path: nebuly/optimization/nebullvm/nebullvm/operations/optimizations/compilers/quantizations/tensor_rt.py
imports: from typing import List, Tuple
  import numpy
  from nebullvm.core.models import QuantizationType, ModelParams
  from nebullvm.optional_modules.tensor_rt import tensorrt, IInt8EntropyCalibrator2, polygraphy
  from nebullvm.tools.transformations import MultiStageTransformation
label_desc: Use image node_id 2 for calling the TensorRTCalibrator obj's underlying member method code with example usage: obj.get_batch(names) and returns: unknown, None