code: def _check_batch_size(self, tensor, expected):
      if self.use_static_batch_size or tf.executing_eagerly():
          self.assertEqual(expected, tensor.shape[0])
  
imports: import collections
  from absl import logging
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.auto_batching import frontend
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.auto_batching import numpy_backend
  from tensorflow_probability.python.experimental.auto_batching import tf_backend
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the _TestHidingTFBatchSize obj's underlying member method code with example usage: obj._check_batch_size(tensor, expected) without return types