Search in sources :

Example 1 with QuantifyDoublePartUParam

use of com.tencent.angel.ml.psf.compress.QuantifyDoubleParam.QuantifyDoublePartUParam in project angel by Tencent.

the class QuantifyDoubleFunc method partitionUpdate.

@Override
public void partitionUpdate(PartitionUpdateParam partParam) {
    RowBasedPartition part = (RowBasedPartition) psContext.getMatrixStorageManager().getPart(partParam.getMatrixId(), partParam.getPartKey().getPartitionId());
    if (part != null) {
        QuantifyDoublePartUParam cp = (QuantifyDoublePartUParam) partParam;
        ServerRow row = part.getRow(cp.getRowId());
        if (row != null) {
            update(row, cp.getArraySlice());
        }
    }
}
Also used : ServerRow(com.tencent.angel.ps.storage.vector.ServerRow) QuantifyDoublePartUParam(com.tencent.angel.ml.psf.compress.QuantifyDoubleParam.QuantifyDoublePartUParam) RowBasedPartition(com.tencent.angel.ps.storage.partition.RowBasedPartition)

Aggregations

QuantifyDoublePartUParam (com.tencent.angel.ml.psf.compress.QuantifyDoubleParam.QuantifyDoublePartUParam)1 RowBasedPartition (com.tencent.angel.ps.storage.partition.RowBasedPartition)1 ServerRow (com.tencent.angel.ps.storage.vector.ServerRow)1