o Õ2úh?ã@s<dZddlmZddlmZddlmZdd„Zdd„Z d S) z*Functions for querying registered kernels.é)Úkernel_def_pb2)Úpywrap_tf_session)ÚcompatcCs.t ¡}t |¡}t ¡}| t |¡¡|S)z8Returns a KernelList proto of all registered kernels. )Úc_apiÚTF_GetAllRegisteredKernelsÚ TF_GetBufferrÚ KernelListÚParseFromStringrÚas_bytes)ÚbufÚdataÚ kernel_list©rú^/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/framework/kernels.pyÚget_all_registered_kernelss  rcCs0t |¡}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ÚTF_GetRegisteredKernelsForOprrrr rr )Únamer r r rrrÚget_registered_kernels_for_op s  rN) Ú__doc__Útensorflow.core.frameworkrÚtensorflow.python.clientrrÚtensorflow.python.utilrrrrrrrÚs