Searched refs:unwrapped_wait_for (Results 1 – 1 of 1) sorted by relevance
160 auto unwrapped_wait_for = UnwrapWaitFor(wait_for); in Allocate() local164 driver_->Allocate(core_id, region, num_bytes, unwrapped_wait_for); in Allocate()184 auto unwrapped_wait_for = UnwrapWaitFor(wait_for); in Allocate() local187 auto handle = driver_->Allocate(core_id, region, shape, unwrapped_wait_for); in Allocate()208 auto unwrapped_wait_for = UnwrapWaitFor(wait_for); in AllocateTuple() local222 unwrapped_wait_for); in AllocateTuple()245 auto unwrapped_wait_for = UnwrapWaitFor(wait_for); in Deallocate() local251 unwrapped_wait_for); in Deallocate()268 auto unwrapped_wait_for = UnwrapWaitFor(wait_for); in TransferToDevice() local276 unwrapped_wait_for)); in TransferToDevice()[all …]