diff options
| author |  Jacques Pienaar <jpienaar@google.com> | 2017-01-11 15:32:45 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-11 16:01:34 -0800 |
| commit | 99e1b19ceba32b8354dddc2841b81864c9ba96bb (patch) | |
| tree | bc94a344d7a67629f521aa0922ca7e9da9a9c869 /tensorflow/compiler/xla/service/backend.h | |
| parent | ca5ba46fb51f25eef39c4b4bfdf0abb75704dba6 (diff) | |
Clarify ResetDevice operates on all devices associated with backend.
Change: 144258290
Diffstat (limited to 'tensorflow/compiler/xla/service/backend.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/backend.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/backend.h b/tensorflow/compiler/xla/service/backend.h index 17c53d299e..db482c09ae 100644 --- a/tensorflow/compiler/xla/service/backend.h +++ b/tensorflow/compiler/xla/service/backend.h @@ -149,6 +149,9 @@ class Backend { // used for scheduling work. For other platforms, returns NULL. const Eigen::ThreadPoolDevice* eigen_intra_op_thread_pool_device() const; + // Resets the devices associated with this backend. + Status ResetDevices(); + private: struct EigenThreadPoolWrapper; Backend(int64 replica_count, perftools::gputools::Platform* platform, |