Searched refs:host_transfer_manager (Results 1 – 1 of 1) sorted by relevance
361 std::shared_ptr<HostTransferManager> host_transfer_manager) { in UnregisterCancellation() argument376 (void)host_transfer_manager; in UnregisterCancellation()431 std::shared_ptr<HostTransferManager> host_transfer_manager = in TPUExecute() local434 host_transfer_manager->Initialize( in TPUExecute()517 token, host_transfer_manager); in TPUExecute()