tf.math.equal