Searched defs:CopyDeviceTensorToCPU (Results 1 – 4 of 4) sorted by relevance
92 void XlaTransferManager::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::XlaTransferManager137 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::XlaDeviceContext
32 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::GPUDeviceContext
102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::SYCLDeviceContext
87 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function