code: def test_size(self):
      t = tf.TensorShape([2, 1, 4])
      self.assertEqual(3, ps.size(t))
      # TF_FLAG_TF_SHAPE_DEFAULT_INT64=true is set by an environment
      # variable when this test is run.
      self.assertEqual(ps.size(t).dtype, tf.size(t).dtype)
  
file_path: probability/tensorflow_probability/python/internal/prefer_static_shape64_test.py
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the SizeShapeTestInt64 obj's underlying member method code with example usage: obj.test_size() without return types