Searched refs:PodBufferHandle (Results 1 – 1 of 1) sorted by relevance
160 class PodBufferHandle : public BufferHandle { class162 explicit PodBufferHandle(PodTpuDriver* driver, int64_t operation_id, in PodBufferHandle() function in tpu_driver::__anon6b014a980111::PodBufferHandle369 return absl::make_unique<PodBufferHandle>(this, operation_id, num_bytes, in Allocate()391 return absl::make_unique<PodBufferHandle>( in Allocate()405 static_cast<PodBufferHandle* const>(children[i])->operation_id(); in AllocateTuple()431 return absl::make_unique<PodBufferHandle>(this, operation_id, 0, in AllocateTuple()440 deps.insert(static_cast<PodBufferHandle*>(handle.get())->operation_id()); in Deallocate()442 auto op_id = static_cast<PodBufferHandle*>(handle.get())->operation_id(); in Deallocate()443 auto core_id = static_cast<PodBufferHandle*>(handle.get())->core_id(); in Deallocate()468 deps.insert(static_cast<PodBufferHandle*>(dst)->operation_id()); in TransferToDevice()[all …]