Searched refs:child_buffers (Results 1 – 3 of 3) sorted by relevance
198 std::vector<std::unique_ptr<PyTpuBuffer>> child_buffers; in FromLiterals() local199 child_buffers.reserve(leaves.size()); in FromLiterals()217 child_buffers.push_back(std::move(child_buffer)); in FromLiterals()267 std::vector<std::shared_ptr<TpuSharedBuffer>> child_buffers, in PyTpuBuffer() argument273 child_buffers_(std::move(child_buffers)) {} in PyTpuBuffer()451 std::vector<std::unique_ptr<PyTpuBuffer>> child_buffers; in AllocateBuffer() local452 child_buffers.reserve(shape.tuple_shapes().size()); in AllocateBuffer()460 child_buffers.push_back(std::move(child_buffer)); in AllocateBuffer()
200 std::vector<std::shared_ptr<TpuSharedBuffer>> child_buffers,
415 std::vector<BufferHandle*> child_buffers; in AllocateTuple() local416 child_buffers.reserve(children_ids.size()); in AllocateTuple()420 child_buffers.push_back(underlying_buffers_[children_ids[i]].get()); in AllocateTuple()426 core_to_driver_core_[core_id], region, child_buffers, {})}); in AllocateTuple()