TensorFlow源码分析（10）：TensorFlow的Device类与分布式计算集群

文章（7）介绍的DeviceBase类是TensorFlow设备的基类，它提供了CPU，GPU管理需要的基本数据结构。Device类在DeviceBase的基础上，添加了一些新的数据结构，这些数据结构覆盖了内存，显存管理，线程同步，远程调用等功能。粗略说来，DeviceBase专注于单个设备的管理，而Device类也关注多设备协作。

首先看Device类的构造函数，路径为tensorflow/core/common_runtime/device.cc。

Device::Device(Env* env, const DeviceAttributes& device_attributes)
    : DeviceBase(env), device_attributes_(device_attributes) {
  CHECK(DeviceNameUtils::ParseFullName(name(), &parsed_name_))
      << "Invalid device name: " << name();
  rmgr_ = new ResourceMgr(parsed_name_.job);
}

Device类有两个参数。env定义了一些与操作系统相关的一些函数（如文件操作），device_attributes是一个Proto Buffer对象，在文章（8）和文章（9）中已经重点介绍了。除了保存这两个参数外，构造函数还做了如下两件事情。

• 创建parsed_name_成员变量。这个parse_name_是设备在一个TensorFlow集群中唯一的名字，这个名字在分布式计算中至关重要，我们稍后详细