a .SicE@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. )rTFE_ToDlpackCapsule)Z tf_tensorr[/var/www/html/django/DPS/env/lib/python3.9/site-packages/tensorflow/python/dlpack/dlpack.py to_dlpacksr zexperimental.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 )rensure_initializedrTFE_FromDlpackCapsule_handle)Z dlcapsulerrr from_dlpack,s r N) __doc__tensorflow.pythonrtensorflow.python.eagerr tensorflow.python.util.tf_exportrr r rrrrs