diff options
Diffstat (limited to 'tensorflow/stream_executor/lib/threadpool.h')
-rw-r--r-- | tensorflow/stream_executor/lib/threadpool.h | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/lib/threadpool.h b/tensorflow/stream_executor/lib/threadpool.h new file mode 100644 index 0000000000..3cf297d57b --- /dev/null +++ b/tensorflow/stream_executor/lib/threadpool.h @@ -0,0 +1,19 @@ +#ifndef TENSORFLOW_STREAM_EXECUTOR_LIB_THREADPOOL_H_ +#define TENSORFLOW_STREAM_EXECUTOR_LIB_THREADPOOL_H_ + +#include "tensorflow/core/lib/core/threadpool.h" +#include "tensorflow/stream_executor/lib/env.h" +#include "tensorflow/stream_executor/lib/notification.h" +#include "tensorflow/stream_executor/lib/thread_options.h" + +namespace perftools { +namespace gputools { +namespace port { + +using tensorflow::thread::ThreadPool; + +} // namespace port +} // namespace gputools +} // namespace perftools + +#endif // TENSORFLOW_STREAM_EXECUTOR_LIB_THREADPOOL_H_ |