remove unnecessary changes

This commit is contained in:
Zhoulong Jiang 2020-11-17 15:40:45 +00:00
parent b8c1ec7503
commit e87611188f
2 changed files with 2 additions and 1 deletions

View File

@ -170,6 +170,7 @@ class DeviceBase {
return device_thread_pool_;
}
// Does not take ownership.
void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d);
@ -203,6 +204,7 @@ 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.

View File

@ -219,4 +219,3 @@ def register_filesystem_plugin(plugin_location):
raise OSError(errno.ENOENT,
'The file to load file system plugin from does not exist.',
plugin_location)