Search in sources :

Example 51 with DoubleBuffer

use of java.nio.DoubleBuffer in project angel by Tencent.

the class MapInPlace method doUpdate.

@Override
protected void doUpdate(ServerDenseDoubleRow[] rows, Serialize func) {
    MapFunc mapper = (MapFunc) func;
    DoubleBuffer rowData = rows[0].getData();
    int size = rows[0].size();
    for (int i = 0; i < size; i++) {
        rowData.put(i, mapper.call(rowData.get(i)));
    }
}
Also used : DoubleBuffer(java.nio.DoubleBuffer)

Example 52 with DoubleBuffer

use of java.nio.DoubleBuffer in project angel by Tencent.

the class MapWithIndex method doUpdate.

@Override
protected void doUpdate(ServerDenseDoubleRow[] rows, Serialize func) {
    MapWithIndexFunc mapper = (MapWithIndexFunc) func;
    DoubleBuffer from = rows[0].getData();
    DoubleBuffer to = rows[1].getData();
    int size = rows[0].size();
    int startCol = (int) rows[0].getStartCol();
    for (int i = 0; i < size; i++) {
        to.put(i, mapper.call(startCol + i, from.get(i)));
    }
}
Also used : DoubleBuffer(java.nio.DoubleBuffer)

Example 53 with DoubleBuffer

use of java.nio.DoubleBuffer in project angel by Tencent.

the class Zip2MapWithIndex method doUpdate.

@Override
protected void doUpdate(ServerDenseDoubleRow[] rows, Serialize func) {
    Zip2MapWithIndexFunc mapper = (Zip2MapWithIndexFunc) func;
    DoubleBuffer from1 = rows[0].getData();
    DoubleBuffer from2 = rows[1].getData();
    DoubleBuffer to = rows[2].getData();
    int size = rows[0].size();
    int startCol = (int) rows[0].getStartCol();
    for (int i = 0; i < size; i++) {
        to.put(i, mapper.call(startCol + i, from1.get(i), from2.get(i)));
    }
}
Also used : DoubleBuffer(java.nio.DoubleBuffer)

Example 54 with DoubleBuffer

use of java.nio.DoubleBuffer in project angel by Tencent.

the class Zip3MapWithIndex method doUpdate.

@Override
protected void doUpdate(ServerDenseDoubleRow[] rows, Serialize func) {
    Zip3MapWithIndexFunc mapper = (Zip3MapWithIndexFunc) func;
    DoubleBuffer from1 = rows[0].getData();
    DoubleBuffer from2 = rows[1].getData();
    DoubleBuffer from3 = rows[2].getData();
    DoubleBuffer to = rows[3].getData();
    int startCol = (int) rows[0].getStartCol();
    int size = rows[0].size();
    for (int i = 0; i < size; i++) {
        to.put(i, mapper.call(startCol + i, from1.get(i), from2.get(i), from3.get(i)));
    }
}
Also used : DoubleBuffer(java.nio.DoubleBuffer)

Example 55 with DoubleBuffer

use of java.nio.DoubleBuffer in project angel by Tencent.

the class Push method doUpdate.

@Override
protected void doUpdate(ServerDenseDoubleRow row, double[] values) {
    DoubleBuffer data = row.getData();
    int size = row.size();
    for (int i = 0; i < size; i++) {
        data.put(i, values[i]);
    }
}
Also used : DoubleBuffer(java.nio.DoubleBuffer)

Aggregations

DoubleBuffer (java.nio.DoubleBuffer)162 ByteBuffer (java.nio.ByteBuffer)39 FloatBuffer (java.nio.FloatBuffer)26 IntBuffer (java.nio.IntBuffer)25 ShortBuffer (java.nio.ShortBuffer)22 LongBuffer (java.nio.LongBuffer)14 CharBuffer (java.nio.CharBuffer)11 BufferOverflowException (java.nio.BufferOverflowException)8 IOException (java.io.IOException)5 BufferUnderflowException (java.nio.BufferUnderflowException)5 ArrayNode (com.fasterxml.jackson.databind.node.ArrayNode)4 ObjectNode (com.fasterxml.jackson.databind.node.ObjectNode)4 ServerDenseDoubleRow (com.tencent.angel.ps.impl.matrix.ServerDenseDoubleRow)4 Test (org.junit.Test)4 InvalidMarkException (java.nio.InvalidMarkException)3 Random (java.util.Random)3 BytePointer (org.bytedeco.javacpp.BytePointer)3 DoublePointer (org.bytedeco.javacpp.DoublePointer)3 FloatPointer (org.bytedeco.javacpp.FloatPointer)3 IntPointer (org.bytedeco.javacpp.IntPointer)3