Searched refs:replicated_buffer (Results 1 – 1 of 1) sorted by relevance
325 std::vector<::xla::ScopedShapedBuffer> replicated_buffer; in TransferToLocalServer() local326 replicated_buffer.emplace_back(std::move(shaped_buffer)); in TransferToLocalServer()330 std::move(replicated_buffer), in TransferToLocalServer()