We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
util
__dlpack__
xp2tensorflow
1 parent fba3bf0 commit fc323e1Copy full SHA for fc323e1
thinc/util.py
@@ -404,7 +404,8 @@ def xp2tensorflow(
404
dlpack_tensor = xp_tensor.toDlpack() # type: ignore
405
tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor)
406
elif hasattr(xp_tensor, "__dlpack__"):
407
- tf_tensor = tf.experimental.dlpack.from_dlpack(xp_tensor)
+ dlpack_tensor = xp_tensor.__dlpack__() # type: ignore
408
+ tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor)
409
else:
410
tf_tensor = tf.convert_to_tensor(xp_tensor)
411
if as_variable:
0 commit comments