o
    ®?e?  ć                   @   s<   d Z ddlmZ ddlmZ ddlmZ dd Zdd Z	d	S )
z*Functions for querying registered kernels.é    )Śkernel_def_pb2)Śpywrap_tf_session)Ścompatc                  C   s.   t  ” } t  | ”}t ” }| t |”” |S )z8Returns a KernelList proto of all registered kernels.
  )Śc_apiZTF_GetAllRegisteredKernelsŚTF_GetBufferr   Ś
KernelListŚParseFromStringr   Śas_bytes)ŚbufŚdataŚkernel_list© r   śd/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/kernels.pyŚget_all_registered_kernels   s
   
r   c                 C   s0   t  | ”}t  |”}t ” }| t |”” |S )zReturns a KernelList proto of registered kernels for a given op.

  Args:
    name: A string representing the name of the op whose kernels to retrieve.
  )r   ZTF_GetRegisteredKernelsForOpr   r   r   r   r   r	   )Śnamer
   r   r   r   r   r   Śget_registered_kernels_for_op    s
   

r   N)
Ś__doc__Ztensorflow.core.frameworkr   Ztensorflow.python.clientr   r   Ztensorflow.python.utilr   r   r   r   r   r   r   Ś<module>   s   
