Examples with CSRStorage com.tencent.angel.ps.storage.partition.storage.CSRStorage used on opensource projects

Example 1 with CSRStorage

use of com.tencent.angel.ps.storage.partition.storage.CSRStorage in project angel by Tencent.

the class BinaryCSRFormat method load.

@Override
public void load(ServerPartition part, MatrixPartitionMeta partMeta, PSMatrixLoadContext loadContext, DataInputStream input) throws IOException {
    CSRPartition csrPart = (CSRPartition) part;
    CSRStorage storage = csrPart.getStorage();
    if (storage instanceof IntCSRStorage) {
        load((IntCSRStorage) storage, input);
    }
}

Example 2 with CSRStorage

use of com.tencent.angel.ps.storage.partition.storage.CSRStorage in project angel by Tencent.

the class BinaryCSRFormat method save.

@Override
public void save(ServerPartition part, MatrixPartitionMeta partMeta, PSMatrixSaveContext saveContext, DataOutputStream output) throws IOException {
    CSRPartition csrPart = (CSRPartition) part;
    CSRStorage storage = csrPart.getStorage();
    if (storage instanceof IntCSRStorage) {
        save((IntCSRStorage) storage, output);
    }
}