Searched defs:per_device_buffers (Results 1 – 1 of 1) sorted by relevance
83 std::vector<xla::PjRtBuffer*> per_device_buffers; member125 std::vector<xla::PjRtBuffer*>& per_device_buffers = in ShardArg() local151 std::vector<xla::PjRtBuffer*>& per_device_buffers = result.per_device_buffers; in ShardArg() local493 std::vector<xla::PjRtBuffer*>& per_device_buffers = in Call() local