tf.compat.v2.size

View source on GitHub