BVhEldZddlmZddlmZddlmZedgdZedgd Zy ) zDLPack modules for Tensorflow.) pywrap_tfe)context) tf_exportzexperimental.dlpack.to_dlpack)v1c,tj|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) tf_tensors O/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/dlpack/dlpack.py to_dlpackr s&  ' ' 22zexperimental.dlpack.from_dlpackctjjtj|tjjS)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) dlcapsules r from_dlpackr,s7$ //&&(  ) ))W__5F5N5N OOr N) __doc__tensorflow.pythonrtensorflow.python.eagerr tensorflow.python.util.tf_exportrr rr r rsM%(+6 *r2333* ,4P5Pr