tf.register_tensor_conversion_function