aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime/threadpool_device_factory.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/common_runtime/threadpool_device_factory.cc')
-rw-r--r--tensorflow/core/common_runtime/threadpool_device_factory.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/threadpool_device_factory.cc b/tensorflow/core/common_runtime/threadpool_device_factory.cc
index 98656600b3..6d1c482afe 100644
--- a/tensorflow/core/common_runtime/threadpool_device_factory.cc
+++ b/tensorflow/core/common_runtime/threadpool_device_factory.cc
@@ -16,6 +16,7 @@ limitations under the License.
// Register a factory that provides CPU devices.
#include "tensorflow/core/common_runtime/threadpool_device.h"
+#include <vector>
#include "tensorflow/core/common_runtime/device_factory.h"
#include "tensorflow/core/framework/allocator.h"
#include "tensorflow/core/public/session_options.h"