Searched refs:AllocateTuple (Results 1 – 6 of 6) sorted by relevance
180 virtual std::unique_ptr<BufferHandle> AllocateTuple(
204 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anone63593b20111::RecordingTpuDriver221 auto handle = driver_->AllocateTuple(core_id, region, unwrapped_children, in AllocateTuple()
182 std::unique_ptr<BufferHandle> AllocateTuple(357 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver361 return streams_[core_id]->AllocateTuple(core_id, region, children, in AllocateTuple()751 std::unique_ptr<BufferHandle> GrpcTpuStream::AllocateTuple( in AllocateTuple() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
395 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon6b014a980111::PodTpuDriver425 core_to_driver_[core_id]->AllocateTuple( in AllocateTuple()
303 std::unique_ptr<BufferHandle> AllocateTuple( in AllocateTuple() function in tpu_driver::__anon9d833a900111::DirectTpuDriver
255 client->driver()->AllocateTuple( in MakeTuple()