Searched refs:GenericTransferManager (Results 1 – 8 of 8) sorted by relevance
35 GenericTransferManager::GenericTransferManager(se::Platform::Id platform_id, in GenericTransferManager() function in xla::GenericTransferManager39 se::Platform::Id GenericTransferManager::PlatformId() const { in PlatformId()43 Status GenericTransferManager::WriteSingleTupleIndexTable( in WriteSingleTupleIndexTable()58 void GenericTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice()71 Status GenericTransferManager::TransferLiteralFromDeviceInternal( in TransferLiteralFromDeviceInternal()99 Status GenericTransferManager::TransferLiteralToDeviceAsync( in TransferLiteralToDeviceAsync()154 Status GenericTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed()159 Status GenericTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed()165 Status GenericTransferManager::ResetDevices( in ResetDevices()172 int64 GenericTransferManager::GetByteSizeRequirement(const Shape& shape) const { in GetByteSizeRequirement()
36 class GenericTransferManager : public TransferManager {38 GenericTransferManager(se::Platform::Id platform_id, size_t pointer_size);39 ~GenericTransferManager() override {} in ~GenericTransferManager()79 TF_DISALLOW_COPY_AND_ASSIGN(GenericTransferManager);
25 class InterpreterTransferManager : public GenericTransferManager {
27 : GenericTransferManager(se::interpreter::kXlaInterpreterPlatformId, in InterpreterTransferManager()
36 class GpuTransferManager : public GenericTransferManager {
46 : GenericTransferManager(id, pointer_size) {} in GpuTransferManager()
36 class CpuTransferManager : public GenericTransferManager {
90 : GenericTransferManager(se::host::kHostPlatformId, in CpuTransferManager()