Searched refs:py_buffers (Results 1 – 1 of 1) sorted by relevance
1482 absl::Span<PjRtBuffer* const> py_buffers, in MakeTupleHelper() argument1487 host_shapes.reserve(py_buffers.size()); in MakeTupleHelper()1488 device_shapes.reserve(py_buffers.size()); in MakeTupleHelper()1489 for (const PjRtBuffer* buffer : py_buffers) { in MakeTupleHelper()