Search in sources :

Example 6 with WeightedSquaredLoss

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

the class QuaternaryOp method constructCPLopsWeightedSquaredLoss.

private void constructCPLopsWeightedSquaredLoss(WeightsType wtype) {
    WeightedSquaredLoss wsloss = new WeightedSquaredLoss(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);
    wsloss.setNumThreads(k);
    setOutputDimensions(wsloss);
    setLineNumbers(wsloss);
    setLops(wsloss);
}
Also used : WeightedSquaredLoss(org.apache.sysml.lops.WeightedSquaredLoss)

Aggregations

WeightedSquaredLoss (org.apache.sysml.lops.WeightedSquaredLoss)6 MultiThreadedHop (org.apache.sysml.hops.Hop.MultiThreadedHop)4 Lop (org.apache.sysml.lops.Lop)4 WeightedSquaredLossR (org.apache.sysml.lops.WeightedSquaredLossR)4 Aggregate (org.apache.sysml.lops.Aggregate)2 DataPartition (org.apache.sysml.lops.DataPartition)2 Group (org.apache.sysml.lops.Group)2 UnaryCP (org.apache.sysml.lops.UnaryCP)2