TensorFlow源码分析(10):TensorFlow的Device类与分布式计算集群

文章(7)介绍的DeviceBase类是TensorFlow设备的基类,它提供了CPU,GPU管理须要的基本数据结构。Device类在DeviceBase的基础上,添加了一些新的数据结构,这些数据结构覆盖了内存,显存管理,线程同步,远程调用等功能。粗略说来,DeviceBase专一于单个设备的管理,而Device类也关注多设备协做。python 首先看Device类的构造函数,路径为tensorf
相关文章
相关标签/搜索