【TensorFlow源码系列】【一】Session的创建

隐身守侯 提交于 2020-04-07 02:13:01

【一】代码下载

https://github.com/tensorflow/tensorflow/releases/

PS:本次源码分析采用1.11版本

【二】Session简介

在TensorFlow中,session是沟通tf的桥梁,模型的训练、推理,都需要通过session,session持有graph的引用。tf支持单机与分布式,因此session也分为单机版与分布式版。

【三】session类图

Session  ::tensorflow\core\public\session.h

DirectSession  ::tensorflow\core\common_runtime\direct_session.h

GrpcSession  ::tensorflow\core\distributed_runtime\rpc\grpc_session.h

class DirectSession : public Session    --->单机版session

class GrpcSession : public Session      ---> 分布式版session

【四】session创建

tf对外提供了一套C API,负责给client语言使用。在tensorflow\c\c_api.h中,session的创建流程我们从这里开始。

TF_NewSession

    NewSession (tensorflow\core\public\session.h)

1. NewSession实现 tensorflow\core\common_runtime\session.cc

Status NewSession(const SessionOptions& options, Session** out_session) {
  SessionFactory* factory;
  Status s = SessionFactory::GetFactory(options, &factory);   // 通过SessionOptions来选择不同的factory
  s = factory->NewSession(options, out_session);   // 调用对应的factory来创建对应的session,也就是创建DirectSession 还GrpcSession
}

2. 关于SessionOptions

这里看一下该定义,省略掉其他成员,仅仅看看target,注释中说,如果target为空,则创建DirectSession,如果target以 'grpc’开头,则创建GrpcSession

/// Configuration information for a Session.
struct SessionOptions {

  /// \brief The TensorFlow runtime to connect to.
  ///
  /// If 'target' is empty or unspecified, the local TensorFlow runtime
  /// implementation will be used.  Otherwise, the TensorFlow engine
  /// defined by 'target' will be used to perform all computations.
  ///
  /// "target" can be either a single entry or a comma separated list
  /// of entries. Each entry is a resolvable address of the
  /// following format:
  ///   local
  ///   ip:port
  ///   host:port
  ///   ... other system-specific formats to identify tasks and jobs ...
  ///
  /// NOTE: at the moment 'local' maps to an in-process service-based
  /// runtime.
  ///
  /// Upon creation, a single session affines itself to one of the
  /// remote processes, with possible load balancing choices when the
  /// "target" resolves to a list of possible processes.
  ///
  /// If the session disconnects from the remote process during its
  /// lifetime, session calls may fail immediately.
  string target;
};

3. SessionFactory::GetFactory

这里的关键在于:session_factory.second->AcceptsOptions(options)

依据选项来,这里实际上是调用对应的factory方法

DirectSessionFactory

bool AcceptsOptions(const SessionOptions& options) override {
    return options.target.empty();  // target为空则为true
  }

GrpcSessionFactory

bool AcceptsOptions(const SessionOptions& options) override {
    return str_util::StartsWith(options.target, kSchemePrefix); // const char* const kSchemePrefix = "grpc://"; target为grpc开头
  }

 

class GrpcSessionFactory : public SessionFactory

class DirectSessionFactory : public SessionFactory

【五】总结

session的创建通过option(可以理解为配置)来选择性创建,而session的真正创建由sessionfactory来完成,这里采用了工厂设计模式,将各个factory通过注册的方式注册到sessionfactory中,这样在系统启动后,这些factory就可以用了,新增一个factory也很容易,且不影响上层代码。

 

 

    

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!