Searched refs:TransferFromDevice (Results 1 – 6 of 6) sorted by relevance
209 virtual std::shared_ptr<Event> TransferFromDevice(
191 std::shared_ptr<Event> TransferFromDevice(const BufferHandle* src, void* dst,377 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anond0091eda0111::GrpcTpuDriver381 return stream->TransferFromDevice(src, dst, wait_for); in TransferFromDevice()798 std::shared_ptr<Event> GrpcTpuStream::TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anond0091eda0111::GrpcTpuStream
294 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon041ad06b0111::RecordingTpuDriver302 std::make_shared<RecordingEvent>(driver_->TransferFromDevice( in TransferFromDevice()
488 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon2989b0910111::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::__anon66f609e90111::DirectTpuDriver
311 transfer_events.push_back(client_->driver()->TransferFromDevice( in CopyToHostAsync()317 transfer_events.push_back(client_->driver()->TransferFromDevice( in CopyToHostAsync()