Searched defs:ShardArgResult (Results 1 – 1 of 1) sorted by relevance
80 struct ShardArgResult { struct83 std::vector<xla::PjRtBuffer*> per_device_buffers;88 std::vector<std::unique_ptr<xla::PjRtBuffer>> owned_buffers;89 py::object owning_sda;