diff options
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_pimpl.cc')
-rw-r--r-- | tensorflow/stream_executor/stream_executor_pimpl.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/stream_executor_pimpl.cc b/tensorflow/stream_executor/stream_executor_pimpl.cc index 9bbfe7f04a..9dc1749327 100644 --- a/tensorflow/stream_executor/stream_executor_pimpl.cc +++ b/tensorflow/stream_executor/stream_executor_pimpl.cc @@ -268,6 +268,10 @@ const DeviceDescription &StreamExecutor::GetDeviceDescription() const { return *device_description_; } +int64 StreamExecutor::GetDeviceLoad() const { + return implementation_->GetDeviceLoad(); +} + int StreamExecutor::PlatformDeviceCount() const { return implementation_->PlatformDeviceCount(); } |