TensorFlow源碼分析(10):TensorFlow的Device類與分佈式計算集羣

文章(7)介紹的DeviceBase類是TensorFlow設備的基類,它提供了CPU,GPU管理須要的基本數據結構。Device類在DeviceBase的基礎上,添加了一些新的數據結構,這些數據結構覆蓋了內存,顯存管理,線程同步,遠程調用等功能。粗略說來,DeviceBase專一於單個設備的管理,而Device類也關注多設備協做。python 首先看Device類的構造函數,路徑爲tensorf
相關文章
相關標籤/搜索