o ®Ÿ?eEć@sTdZddlmZddlmZddlmZedgddd„ƒZed gdd d „ƒZd S) zDLPack modules for Tensorflow.é)Ś pywrap_tfe)Ścontext)Ś tf_exportzexperimental.dlpack.to_dlpack)Śv1cCs t |”S)aReturns the dlpack capsule representing the tensor. This operation ensures the underlying data memory is ready when returns. ```python a = tf.tensor([1, 10]) dlcapsule = tf.experimental.dlpack.to_dlpack(a) # dlcapsule represents the dlpack data structure ``` Args: tf_tensor: Tensorflow eager tensor, to be converted to dlpack capsule. Returns: A PyCapsule named as dltensor, which shares the underlying memory to other framework. This PyCapsule can be consumed only once. )rZTFE_ToDlpackCapsule)Z tf_tensor©rś`/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/dlpack/dlpack.pyŚ to_dlpacks rzexperimental.dlpack.from_dlpackcCst ” ”t |t ”j”S)aXReturns the Tensorflow eager tensor. The returned tensor uses the memory shared by dlpack capsules from other framework. ```python a = tf.experimental.dlpack.from_dlpack(dlcapsule) # `a` uses the memory shared by dlpack ``` Args: dlcapsule: A PyCapsule named as dltensor Returns: A Tensorflow eager tensor )rZensure_initializedrZTFE_FromDlpackCapsuleZ_handle)Z dlcapsulerrrŚ from_dlpack,s r N) Ś__doc__Ztensorflow.pythonrZtensorflow.python.eagerrZ tensorflow.python.util.tf_exportrrr rrrrŚs