tf.size

TensorFlow 1 version View source on GitHub