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