diff options
Diffstat (limited to 'tensorflow/core/grappler/optimizers/pin_to_host_optimizer.h')
-rw-r--r-- | tensorflow/core/grappler/optimizers/pin_to_host_optimizer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.h b/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.h index bed4a9ef95..d557a03463 100644 --- a/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.h +++ b/tensorflow/core/grappler/optimizers/pin_to_host_optimizer.h @@ -26,8 +26,8 @@ namespace tensorflow { namespace grappler { namespace internal { // Try and find an appropriate Host device in `devices` given `device`. -bool TrySwapToHostDevice(const gtl::FlatSet<string>& devices, - bool has_device_cpu, string* device); +string TryFindHostDevice(const gtl::FlatSet<string>& devices, + bool has_device_cpu, const string& device); } // end namespace internal // Optimize TensorFlow ops that should be swapped into the CPU to avoid |