experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

address comments

Browse Source
This commit is contained in:
Zhoulong Jiang 2020-11-18 10:28:12 +00:00
parent e87611188f
commit 77cf1eba91
3 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/c/c_api_experimental.cc
View File

@ -777,9 +777,7 @@ TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename,
} else {
status->status =
env->LoadDynamicLibrary(library_filename, &lib_handle->lib_handle);
if (status->status.ok()) {
// Init PluggableDevice Plugin
} else {
if (!status->status.ok()) {
delete lib_handle;
return nullptr;
}

2
tensorflow/core/framework/device_base.h
View File

@ -170,7 +170,6 @@ class DeviceBase {
return device_thread_pool_;
}
// Does not take ownership.
void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d);
@ -204,7 +203,6 @@ class DeviceBase {
virtual const Eigen::ThreadPoolDevice* eigen_cpu_device();
// Caller owns the return value. The OpKernelContext calls this even
// for devices that do not implement an eigen_gpu_device. Overridden
// by GPU devices to return a derived type.

2
tensorflow/python/framework/load_library.py
View File

@ -159,7 +159,7 @@ def load_library(library_location):
library_location)
def load_pluggable_device_library(library_location):
"""Loads a Tensorflow PluggableDevice plugin.
"""Loads 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