aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime/session_factory.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/common_runtime/session_factory.h')
-rw-r--r--tensorflow/core/common_runtime/session_factory.h25
1 files changed, 25 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/session_factory.h b/tensorflow/core/common_runtime/session_factory.h
new file mode 100644
index 0000000000..f770ba93ff
--- /dev/null
+++ b/tensorflow/core/common_runtime/session_factory.h
@@ -0,0 +1,25 @@
+#ifndef TENSORFLOW_COMMON_RUNTIME_SESSION_FACTORY_H_
+#define TENSORFLOW_COMMON_RUNTIME_SESSION_FACTORY_H_
+
+#include <string>
+
+#include "tensorflow/core/lib/gtl/array_slice.h"
+#include "tensorflow/core/platform/port.h"
+#include "tensorflow/core/public/status.h"
+
+namespace tensorflow {
+
+class Session;
+class SessionOptions;
+
+class SessionFactory {
+ public:
+ virtual Session* NewSession(const SessionOptions& options) = 0;
+ virtual ~SessionFactory() {}
+ static void Register(const string& runtime_type, SessionFactory* factory);
+ static SessionFactory* GetFactory(const string& runtime_type);
+};
+
+} // namespace tensorflow
+
+#endif // TENSORFLOW_COMMON_RUNTIME_SESSION_FACTORY_H_