加载TensorFlow插件的函数

函数：tf.load_file_system_library

load_file_system_library(library_filename)

定义在：tensorflow/python/framework/load_library.py.

参见指南：构建图>实用程序功能

加载包含文件系统实现的 TensorFlow 插件.
将 library_filename 传递到一个特定于平台的机制,以动态加载库.

用于确定库的确切位置的规则是平台特定的,这里没有记录.

参数：

• library_filename：插件的路径,动态库文件的相对或绝对文件系统路径.

返回：

该函数将返回 None.

可能引发的异常：

• RuntimeError：当无法加载库时.

函数：tf.load_op_library

load_op_library ( library_filename )

定义在：tensorflow/python/framework/load_library.py

参见指南：构建图>实用程序功能

加载包含自定义操作和内核的 TensorFlow 插件.

将 “library_filename” 传递给特定于平台的机制,以动态加载库.用于确定库的确切位置的规则是平台特定的,这里没有记录.当加载库时,通过 REGISTER_* 宏在库中注册的操作和内核可在 TensorFlow 进程中使用.请注意,具有与现有操作系统相同名称的操作将被拒绝,并且不会在进程中注册.

参数：

• library_filename：插件的路径.动态库文件的相对或绝对文件系统路径.

返回值：

一个 python 模块,包含插件中定义的 Ops 的 python 包装.

可能引发的异常：

• RuntimeError：当无法加载库或获取 python 包装时.