o e@sTdZddlmZddlmZddlmZedgdddZed gdd 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_tensorrOD:\Projects\ConvertPro\env\Lib\site-packages\tensorflow/python/dlpack/dlpack.py to_dlpacks rzexperimental.dlpack.from_dlpackcCstt|tjS)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_FromDlpackCapsule_handle)Z dlcapsulerrr from_dlpack,s r N) __doc__Ztensorflow.pythonrZtensorflow.python.eagerrZ tensorflow.python.util.tf_exportrrr rrrrs