Searched refs:GrpcBufferHandle (Results 1 – 1 of 1) sorted by relevance
86 class GrpcBufferHandle : public BufferHandle { class88 explicit GrpcBufferHandle( in GrpcBufferHandle() function in tpu_driver::__anond0091eda0111::GrpcBufferHandle367 auto* stream = static_cast<GrpcBufferHandle*>(handle.get())->stream(); in Deallocate()374 auto* stream = static_cast<GrpcBufferHandle*>(dst)->stream(); in TransferToDevice()380 auto* stream = static_cast<const GrpcBufferHandle*>(src)->stream(); in TransferFromDevice()387 auto* stream = static_cast<const GrpcBufferHandle*>(src)->stream(); in TransferFromDeviceToDevice()731 return absl::make_unique<GrpcBufferHandle>(event->id(), std::move(event), in Allocate()747 return absl::make_unique<GrpcBufferHandle>( in Allocate()761 auto grpc_child = static_cast<GrpcBufferHandle*>(child); in AllocateTuple()767 return absl::make_unique<GrpcBufferHandle>(event->id(), std::move(event), 0); in AllocateTuple()[all …]