Searched refs:TransferFromDevice (Results 1 – 6 of 6) sorted by relevance
209 virtual std::shared_ptr<Event> TransferFromDevice(
294 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anone63593b20111::RecordingTpuDriver302 std::make_shared<RecordingEvent>(driver_->TransferFromDevice( in TransferFromDevice()
191 std::shared_ptr<Event> TransferFromDevice(const BufferHandle* src, void* dst,377 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver381 return stream->TransferFromDevice(src, dst, wait_for); in TransferFromDevice()798 std::shared_ptr<Event> GrpcTpuStream::TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
488 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon6b014a980111::PodTpuDriver504 return core_to_driver_[core_id]->TransferFromDevice( in TransferFromDevice()552 auto src_event = TransferFromDevice(src, host_buf, wait_for); in TransferFromDeviceToDevice()
351 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon9d833a900111::DirectTpuDriver
311 transfer_events.push_back(client_->driver()->TransferFromDevice( in CopyToHostAsync()317 transfer_events.push_back(client_->driver()->TransferFromDevice( in CopyToHostAsync()