c++ - tensorflow c++ 代码 SessionFactory::GetFactory 如何选择使用哪个 session ？Direct 或 Grpc Session？

Question

在tensorflow c++代码中，有两种session类型，

tensorflow/core/common_runtime/direct_session.cctensorflow/core/distributed_runtime/rpc/grpc_session.cc

我可以发现 session 是通过方法初始化的

状态NewSession(const SessionOptions& options, Session out_session)

在/tensorflow/core/common_runtime/session.cc,

问题是，不知道下面代码中的*session_factories是什么时候填上“DIRECT_SESSION”的:

for (const auto& session_factory : *session_factories()) {
   if (session_factory.second->AcceptsOptions(options)) {
       VLOG(0) << "SessionFactory type " << session_factory.first
          << " accepts target: " << options.target;
  candidate_factories.push_back(session_factory);
} else {
  VLOG(0) << "SessionFactory type " << session_factory.first
          << " does not accept target: " << options.target;
}

/tensorflow/core/common_runtime/session_factory.cc,

Answer 1

session_factories() 映射通过调用 SessionFactory::Register() 来填充，实现了 here , 并在小类打电话 DirectSessionRegistrar和 GrpcSessionRegistrar .请注意，DirectSessionRegistrar 和 GrpcSessionRegistrar 在各自的源文件中都被定义为 static 变量。这可确保在第一次调用 NewSession() 之前，在静态初始化时初始化工厂。