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

函數(shù)：tf.load_file_system_library

load_file_system_library(library_filename)

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

參見指南：構(gòu)建圖>實(shí)用程序功能

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

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

參數(shù)：

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

返回：

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

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

• RuntimeError：當(dāng)無法加載庫時(shí).

函數(shù)：tf.load_op_library

load_op_library ( library_filename )

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

參見指南：構(gòu)建圖>實(shí)用程序功能

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

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

參數(shù)：

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

返回值：

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

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

• RuntimeError：當(dāng)無法加載庫或獲取 python 包裝時(shí).