Searched refs:TransferInfo (Results 1 – 2 of 2) sorted by relevance
227 struct TransferInfo { struct in tpu_driver::__anond0091eda0111::GrpcTpuStream228 explicit TransferInfo(void* dst, int64_t num_bytes) in TransferInfo() argument306 absl::flat_hash_map<EventId, TransferInfo> transfers_808 TransferInfo info(dst, const_cast<BufferHandle*>(src)->size_in_bytes()); in TransferFromDevice()
522 struct TransferInfo struct535 void copyToCPU (const DeviceInterface& vkd, TransferInfo* transferInfo) in copyToCPU()690 TransferInfo transferInfo; in generateWork()