Search in sources :

Example 1 with QuantifyFloatPartParam

use of com.tencent.angel.ml.psf.compress.QuantifyFloatParam.QuantifyFloatPartParam in project angel by Tencent.

the class QuantifyFloatFunc method partitionUpdate.

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

Aggregations

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