TF代码研读-Device(LocalDevice,ThreadPoolDevice,BaseGPUDevice)

LocalDevice类根据底层执行具体计算设备的不同，可以进一步派生子类：SYCLDevice(异构)、BaseGPUDevice(Cuda显卡)、ThreadPoolDevice(CPU设备)、XlaDevice(支持xla的设备？)。

 

其中ThreadPoolDevice是针对CPU设备的类，主要重写了allocator与张量拷贝创建相关操作，其下还有GPUCompatibleCPUDevice的子类，主要用于GPU兼容CPU的设备（配有GPU的PC？），

 

BaseGPUDevice针对cuda设备(Nvida显卡)重写了Compute相关方法，派生子类GPUDevice类，与

GPUCompatibleCPUDevice类类似，明确获取GPUAllocator。

 

SYCLDevice为针对opencl设备(异构设备),

XlaDevice针对支持xlaDevice的设备.....

 

ThreadPoolDevice类，重写了Allocator与tensor拷贝创建相关的方法

 

// CPU device implementation.
class ThreadPoolDevice : public LocalDevice {
 public:
  ThreadPoolDevice(const Session