code: def _batch(self, dims, batch_size):
      if dims[0] is None:
          dims[0] = batch_size
      return dims
  
file_path: model-optimization/tensorflow_model_optimization/python/core/quantization/keras/quantize_models_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import inspect
  import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.quantization.keras import quantize
  from tensorflow_model_optimization.python.core.quantization.keras import utils
label_desc: Use image node_id 1 for calling the QuantizeModelsTest obj's underlying member method code with example usage: obj._batch(dims, batch_size) and returns: dims