加載TensorFlow插件的函數(shù)

函數(shù)：tf.load_file_system_library

load_file_system_library(library_filename)

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

參見指南：構建圖>實用程序功能

加載包含文件系統(tǒng)實現(xiàn)的 TensorFlow 插件.
將 library_filename 傳遞到一個特定于平臺的機制,以動態(tài)加載庫.

用于確定庫的確切位置的規(guī)則是平臺特定的,這里沒有記錄.

參數(shù)：

• library_filename：插件的路徑,動態(tài)庫文件的相對或絕對文件系統(tǒng)路徑.

返回：

該函數(shù)將返回 None.

可能引發(fā)的異常：

• RuntimeError：當無法加載庫時.

函數(shù)：tf.load_op_library

load_op_library ( library_filename )

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

參見指南：構建圖>實用程序功能

加載包含自定義操作和內(nèi)核的 TensorFlow 插件.

將 “l(fā)ibrary_filename” 傳遞給特定于平臺的機制,以動態(tài)加載庫.用于確定庫的確切位置的規(guī)則是平臺特定的,這里沒有記錄.當加載庫時,通過 REGISTER_* 宏在庫中注冊的操作和內(nèi)核可在 TensorFlow 進程中使用.請注意,具有與現(xiàn)有操作系統(tǒng)相同名稱的操作將被拒絕,并且不會在進程中注冊.

參數(shù)：

• library_filename：插件的路徑.動態(tài)庫文件的相對或絕對文件系統(tǒng)路徑.

返回值：

一個 python 模塊,包含插件中定義的 Ops 的 python 包裝.

可能引發(fā)的異常：

• RuntimeError：當無法加載庫或獲取 python 包裝時.