TF代码研读-Device(DeviceMgr,DeviceSet)

橘子都吃不起！

于 2021-01-31 14:15:48 发布

阅读量539

点赞数 1

分类专栏： Tensorflow源码研读文章标签： c++ tensorflow 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_42754237/article/details/113471913

版权

Tensorflow源码研读专栏收录该内容

10 篇文章 ¥29.90 ¥99.00

订阅专栏

DeviceMgr是TensorFlow的设备管理器，抽象存储设备并提供查找和通信功能；DeviceSet是设备集合，用于图中设备调度，支持增删查排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DeviceMgr类是设备管理器的抽象，以向量的方式存储设备指针，提供查找设备、输出设备参数怒相关方法，主要用于会话、汇合点和进程之间通信等场合。

//设备管理类
class DeviceMgr {
 public:
  //输入一个包含设备对象指针的列表，初始化DeviceMgr
  explicit DeviceMgr(std::vector<std::unique_ptr<Device>> devices);
  //输入单个设备指针，初始化DeviceMgr
  explicit DeviceMgr(std::unique_ptr<Device> device);
  ~DeviceMgr();
  //放回所有设备的属性参数，放在一个列表中
  void ListDeviceAttributes(std::vector<DeviceAttributes>* devices) const;
  //获取设备列表指针
  std::vector<Device*> ListDevices() const;
  string DebugString() const;
  //放回设备对应的描述字符串？
  string DeviceMappingString() const;

  // Assigns *device with pointer to Device of the given name.
  // Accepts either a full device name, or just the replica

了解本专栏