Search in sources :

Example 6 with WeightedSigmoid

use of org.apache.sysml.lops.WeightedSigmoid in project systemml by apache.

the class QuaternaryOp method constructCPLopsWeightedSigmoid.

private void constructCPLopsWeightedSigmoid(WSigmoidType wtype) {
    WeightedSigmoid wsig = new WeightedSigmoid(getInput().get(0).constructLops(), getInput().get(1).constructLops(), getInput().get(2).constructLops(), getDataType(), getValueType(), wtype, ExecType.CP);
    // set degree of parallelism
    int k = OptimizerUtils.getConstrainedNumThreads(_maxNumThreads);
    wsig.setNumThreads(k);
    setOutputDimensions(wsig);
    setLineNumbers(wsig);
    setLops(wsig);
}
Also used : WeightedSigmoid(org.apache.sysml.lops.WeightedSigmoid)

Aggregations

WeightedSigmoid (org.apache.sysml.lops.WeightedSigmoid)6 MultiThreadedHop (org.apache.sysml.hops.Hop.MultiThreadedHop)4 Lop (org.apache.sysml.lops.Lop)4 WeightedSigmoidR (org.apache.sysml.lops.WeightedSigmoidR)4 DataPartition (org.apache.sysml.lops.DataPartition)2 Group (org.apache.sysml.lops.Group)2