Searched refs:HostTransferManager (Results 1 – 1 of 1) sorted by relevance
65 class HostTransferManager { class67 explicit HostTransferManager(TpuNodeContext*, xla::Backend*) {} in HostTransferManager() function in tensorflow::__anon5955583b0111::HostTransferManager78 TF_DISALLOW_COPY_AND_ASSIGN(HostTransferManager);81 xla::StatusOr<HostTransferManager::HostCommmandHandler>82 HostTransferManager::Initialize(const TPUHostTransferInfoProto& program, in Initialize()361 std::shared_ptr<HostTransferManager> host_transfer_manager) { in UnregisterCancellation()431 std::shared_ptr<HostTransferManager> host_transfer_manager = in TPUExecute()432 std::make_shared<HostTransferManager>(node_context, backend); in TPUExecute()433 TF_ASSIGN_OR_RETURN(HostTransferManager::HostCommmandHandler handler, in TPUExecute()