Search in sources :

Example 6 with WeightedDivMM

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

the class QuaternaryOp method constructCPLopsWeightedDivMM.

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

Aggregations

WeightedDivMM (org.apache.sysml.lops.WeightedDivMM)6 MultiThreadedHop (org.apache.sysml.hops.Hop.MultiThreadedHop)4 Lop (org.apache.sysml.lops.Lop)4 WeightedDivMMR (org.apache.sysml.lops.WeightedDivMMR)4 Aggregate (org.apache.sysml.lops.Aggregate)2 DataPartition (org.apache.sysml.lops.DataPartition)2 Group (org.apache.sysml.lops.Group)2