diff --git a/thinc/util.py b/thinc/util.py index 7add0aaf7..43fb115d2 100644 --- a/thinc/util.py +++ b/thinc/util.py @@ -404,7 +404,8 @@ def xp2tensorflow( dlpack_tensor = xp_tensor.toDlpack() # type: ignore tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor) elif hasattr(xp_tensor, "__dlpack__"): - tf_tensor = tf.experimental.dlpack.from_dlpack(xp_tensor) + dlpack_tensor = xp_tensor.__dlpack__() # type: ignore + tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor) else: tf_tensor = tf.convert_to_tensor(xp_tensor) if as_variable: