tf.compat.v1.wrap_function