tf.compat.v1.ones_like