Searched refs:per_device_pybuffer (Results 1 – 1 of 1) sorted by relevance
159 for (py::handle per_device_pybuffer : per_device_pybuffers) { in ShardArg() local161 xla::PyBuffer::AsPyBuffer(per_device_pybuffer).ValueOrDie()->buffer(); in ShardArg()