Searched refs:TpuTransferManager (Results 1 – 3 of 3) sorted by relevance
40 TpuTransferManager::TpuTransferManager() { in TpuTransferManager() function in tensorflow::tpu::TpuTransferManager44 TpuTransferManager::~TpuTransferManager() { in ~TpuTransferManager()48 stream_executor::Platform::Id TpuTransferManager::PlatformId() const { in PlatformId()52 xla::Shape TpuTransferManager::HostShapeToDeviceShape( in HostShapeToDeviceShape()67 Status TpuTransferManager::TransferLiteralToDeviceAsync( in TransferLiteralToDeviceAsync()89 Status TpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed()105 Status TpuTransferManager::TransferBuffersToInfeed( in TransferBuffersToInfeed()129 Status TpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed()150 Status TpuTransferManager::ResetDevices( in ResetDevices()190 void TpuTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice()[all …]
30 class TpuTransferManager : public xla::TpuTransferManagerInterface {32 TpuTransferManager();33 ~TpuTransferManager() override;
27 return std::make_unique<TpuTransferManager>(); in CreateTpuTransferManager()