Searched refs:transfer_events (Results 1 – 2 of 2) sorted by relevance
120 TpuEvent* transfer_events[] = {transfer_ev1, transfer_ev2}; in main() local126 /*eventc=*/2, /*eventv*/transfer_events); in main()
283 std::vector<std::shared_ptr<tpu_driver::Event>> transfer_events; in CopyToHostAsync() local311 transfer_events.push_back(client_->driver()->TransferFromDevice( in CopyToHostAsync()317 transfer_events.push_back(client_->driver()->TransferFromDevice( in CopyToHostAsync()324 for (auto& t : transfer_events) { in CopyToHostAsync()