BVh""dZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddlmZddlmZedd Zej&dd ed g dZdZeddZdZeddZy)z(Function for loading TensorFlow plugins.N)pywrap_tf_session)context)_pywrap_python_op_gen) deprecation) tf_exportload_op_librarycZtj|} tjtj|}tj |t j|j}|tjvrtj|Stjj|d}tjj|}t!||j"t%|dd|tj|<|S#tj |wxYw)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. N_IS_TENSORFLOW_PLUGINT)py_tfTF_LoadLibraryrGetPythonWrappers TF_GetOpListTF_DeleteLibraryHandlehashlibsha1 hexdigestsysmodules importlib machinery ModuleSpecutilmodule_from_specexec__dict__setattr)library_filename lib_handlewrappers module_name module_specmodules X/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/load_library.pyrrs.##$45*-$66 :&(H    , X&002+CKK ;;{ ####..{DA+ >> * *; 7&x! &)40##++k -   ,s (DD*zUse `tf.load_library` instead.)date instructionsload_file_system_library)v1c.tj|y)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)r r )rs r#r&r&Ms('(cPtjdk(r?|jdry|jd}|dk(ry||dzj Stjdk(r|jd Stjd k(r|jd Sy) zECheck the file to see if it is a shared object, only using extension.Linuxz.soTz.so.FDarwinz.dylibWindowsz.dll)platformsystemendswithrfind isdecimal)filenameindexs r#_is_shared_objectr7ds __'! nnV$e " ",,..H$   X &&I%   V $$ r) load_libraryctjj|rtjj|rNtj|}|Dcgc]-}t |r tjj ||/}}n|g}|D]}tj|yttjd|cc}w)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. z@The file or folder to load kernel libraries from does not exist.N) ospathexistsisdirlistdirr7joinr r OSErrorerrnoENOENT)library_locationdirectory_contentsfkernel_librarieslibs r#r8r8xs(WW^^$% ww}}%&::&676H#01 q ! '',,' +##++  3   J #s2Cctjj|rtjj|rNtj|}|Dcgc]-}t |r tjj ||/}}n|g}|D]}tj|tjjyttjd|cc}w)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. zJThe file or folder to load pluggable device libraries from does not exist.N)r:r;r<r=r>r7r?r TF_LoadPluggableDeviceLibraryrreinitialize_physical_devicesr@rArB)rCrDrEpluggable_device_librariesrGs r#load_pluggable_device_libraryrLs WW^^$% ww}}%&::&67&$ q ! '',,' +$ $ %5#5 )/ ))#./ OO335   " $$$s2C*z'experimental.register_filesystem_pluginctjj|rtj|yt t jd|)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)r:r;r<r TF_RegisterFilesystemPluginr@rArB)plugin_locations r#register_filesystem_pluginrPs=WW^^O$ %%o6 %,,L! ##r))__doc__rArrr:r0rtensorflow.python.clientrr tensorflow.python.eagerrtensorflow.python.frameworkrtensorflow.python.utilr tensorflow.python.util.tf_exportrr deprecatedr&r7r8rLrPr)r#rYs /   ?+=.6 **ZT%EG )*+),G)(( >$$N$$N 45#6#r)