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);
}
Aggregations