o ?e"@sdZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddlmZddlmZedd dZejdd d ed gd dd ZddZedddZddZedddZdS)z(Function for loading TensorFlow plugins.N)pywrap_tf_session)context)_pywrap_python_op_gen) deprecation) tf_exportload_op_libraryc Cst|}ztt|}Wt|nt|wt|}|t j vr,t j |St j |d}t j|}t||jt|dd|t j |<|S)aCLoads a TensorFlow plugin, containing custom ops and kernels. Pass "library_filename" to a platform-specific mechanism for dynamically loading a library. The rules for determining the exact location of the library are platform-specific and are not documented here. When the library is loaded, ops and kernels registered in the library via the `REGISTER_*` macros are made available in the TensorFlow process. Note that ops with the same name as an existing op are rejected and not registered with the process. Args: library_filename: Path to the plugin. Relative or absolute filesystem path to a dynamic library file. Returns: A python module containing the Python wrappers for Ops defined in the plugin. Raises: RuntimeError: when unable to load the library or get the python wrappers. NZ_IS_TENSORFLOW_PLUGINT)py_tfTF_LoadLibraryrZGetPythonWrappersZ TF_GetOpListZTF_DeleteLibraryHandlehashlibsha1 hexdigestsysmodules importlib machinery ModuleSpecutilmodule_from_specexec__dict__setattr)library_filenameZ lib_handlewrappers module_nameZ module_specmoduleri/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/load_library.pyrs       zUse `tf.load_library` instead.)dateZ instructionsload_file_system_library)v1cCst|dS)aLoads a TensorFlow plugin, containing file system implementation. Pass `library_filename` to a platform-specific mechanism for dynamically loading a library. The rules for determining the exact location of the library are platform-specific and are not documented here. Args: library_filename: Path to the plugin. Relative or absolute filesystem path to a dynamic library file. Returns: None. Raises: RuntimeError: when unable to load the library. N)rr )rrrrrMscCsptdkr |dr dS|d}|dkrdS||dStdkr+|d Std kr6|d SdS) zECheck the file to see if it is a shared object, only using extension.Linuxz.soTz.so.FDarwinz.dylibWindowsz.dll)platformsystemendswithrfind isdecimal)filenameindexrrr_is_shared_objectds       r, load_librarycsbtjr*tjrt}fdd|D}ng}|D]}t|q dSttj d)arLoads a TensorFlow plugin. "library_location" can be a path to a specific shared object, or a folder. If it is a folder, all shared objects that are named "libtfkernel*" will be loaded. When the library is loaded, kernels registered in the library via the `REGISTER_*` macros are made available in the TensorFlow process. Args: library_location: Path to the plugin or the folder of plugins. Relative or absolute filesystem path to a dynamic library file or folder. Returns: None Raises: OSError: When the file to be loaded is not found. RuntimeError: when unable to load the library. c"g|] }t|rtj|qSrr,ospathjoin.0flibrary_locationrr s  z load_library..z@The file or folder to load kernel libraries from does not exist.N) r0r1existsisdirlistdirrr OSErrorerrnoENOENT)r7directory_contentsZkernel_librarieslibrr6rr-xs     csntjr0tjrt}fdd|D}ng}|D]}t|q tdSt t j d)amLoads a TensorFlow PluggableDevice plugin. "library_location" can be a path to a specific shared object, or a folder. If it is a folder, all shared objects will be loaded. when the library is loaded, devices/kernels registered in the library via StreamExecutor C API and Kernel/Op Registration C API are made available in TensorFlow process. Args: library_location: Path to the plugin or folder of plugins. Relative or absolute filesystem path to a dynamic library file or folder. Raises: OSError: When the file to be loaded is not found. RuntimeError: when unable to load the library. cr.rr/r3r6rrr8s  z1load_pluggable_device_library..zJThe file or folder to load pluggable device libraries from does not exist.N) r0r1r9r:r;rZTF_LoadPluggableDeviceLibraryrZreinitialize_physical_devicesr<r=r>)r7r?Zpluggable_device_librariesr@rr6rload_pluggable_device_librarys     rAz'experimental.register_filesystem_plugincCs(tj|r t|dSttjd|)a0Loads a TensorFlow FileSystem plugin. Args: plugin_location: Path to the plugin. Relative or absolute filesystem plugin path to a dynamic library file. Returns: None Raises: OSError: When the file to be loaded is not found. RuntimeError: when unable to load the library. z8The file to load file system plugin from does not exist.N)r0r1r9rZTF_RegisterFilesystemPluginr<r=r>)Zplugin_locationrrrregister_filesystem_plugins rB)__doc__r=r rr0r%r Ztensorflow.python.clientrrZtensorflow.python.eagerrZtensorflow.python.frameworkrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportrr deprecatedrr,r-rArBrrrrs2      -   ''