加载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 包装时.
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部