tf.math.reduce_all