TF代码研读-Device(RemoteDevice)

最新推荐文章于 2022-04-06 18:23:28 发布

橘子都吃不起！

最新推荐文章于 2022-04-06 18:23:28 发布

阅读量383

点赞数

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

本文链接：https://blog.csdn.net/weixin_42754237/article/details/113470895

版权

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

10 篇文章 10 订阅 ¥29.90 ¥99.00

订阅专栏

RemoteDevice是TensorFlow中用于处理远程设备的类，它通过RPC服务与本地设备通信。该类提供了与本地Device相同的操作接口，但实际计算在RPC通道上进行。在remote_device.cc中定义，GetAllocator返回nullptr，IsLocal表明非本地设备。AsRemoteDevices和NewRemoteDevices负责创建和管理远程设备实例，前者检查是否存在，后者通过WorkerCacheInterface创建新的远程设备。

摘要由CSDN通过智能技术生成

RemoteDevice类为Device子类，专门针对远程设备提供接口。远程设备与本地的通信主要通过RPC服务进行的。对于本地进程来说，RemoteDevice类提供了与Device类一致的接口，但在底层运算上，通过RPC通道传输操作指令与数据。

RemoteDevice类定义在tensorflow/core/distributed_runtime/remote_device.cc中，注意不是在头文件里，其头文件中定义了两个创建RemoteDevice实例的函数:

typedef std::function<Status(StringPiece name, Device** device)>
    LookupLocalDevice;//定义了一个函数指针
// 创建一个远程设备（需要已知attributes）
void AsRemoteDevices(
    Env* env,
    const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
    LookupLocalDevice lookup_local_device,
    std::vector<std::unique_ptr<Device>>* remote_devices);

//通过RPC创建指定worker_name的远程设备，如果成功返回NewRemoteDevicesDone done
typedef std::function<

了解本专栏