o
    پi                     @   s8   d Z ddlmZ ddlmZ dd Zdd Zd	d
 ZdS )zDECORD tensorflow bridge    )absolute_import   )_from_dlpackc               
   C   s2   zt ddgdW S  ty }  ztdd} ~ ww )zTry to import tensorflow dlpack at runtime.

    Returns
    -------
    tensorflow dlpack module if found. Raise ImportError otherwise
    ztensorflow.experimental.dlpack )fromlistz tensorflow >= 2.2.0 is required.N)
__import__ImportError)e r
   D/home/ubuntu/.local/lib/python3.10/site-packages/decord/bridge/tf.pytry_import_tfdl   s   r   c                 C   s@   t  }ddlm} ddlm} | }|  ||  |jS )z"from decord to tensorflow, no copyr   )
pywrap_tfe)context)	r   tensorflow.pythonr   tensorflow.python.eagerr   ensure_initializedTFE_FromDlpackCapsule	to_dlpack_handle)
decord_arrtfdlr   r   ctxr
   r
   r   to_tensorflow   s   r   c                 C   s   t  }t|| S )z"from tensorflow to decord, no copy)r   r   r   )	tf_tensorr   r
   r
   r   from_tensorflow   s   r   N)__doc__
__future__r   _ffi._ctypes.ndarrayr   r   r   r   r
   r
   r
   r   <module>   s    	