tf.math.pow