change cpu conversion for tf to convert-to-tensor (#271)

NVIDIA-Merlin · Apr 4, 2023 · 4561776 · 4561776
1 parent 1cc04b3
commit 4561776
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/merlin/table/tensorflow_column.py b/merlin/table/tensorflow_column.py
@@ -158,7 +158,7 @@ def _register_from_dlpack_cpu_to_tf():
     @_from_dlpack_cpu.register(tf.Tensor)
     @_from_dlpack_cpu.register(eager_tensor_type)
     def _from_dlpack_cpu_to_tf(target_type, array):
-        return tf.experimental.dlpack.from_dlpack(array.__dlpack__())
+        return tf.convert_to_tensor(array)
 
 
 @_from_dlpack_gpu.register_lazy("tensorflow")