code: def test_dtype(self, tf_dtype, np_dtype):
      ind, _ = convolution_util.im2row_index(
          input_shape=(1, 12, 16, 3), block_shape=(2, 3), dtype=tf_dtype
      )
      self.assertDTypeEqual(ind, np_dtype)
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.nn.util import convolution_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
label_desc: Use image node_id 2 for calling the Im2RowTest obj's underlying member method code with example usage: obj.test_dtype(tf_dtype, np_dtype) without return types