public CumProd(SameDiff sameDiff, SDVariable x, int... dimension) { super(null, sameDiff, new SDVariable[]{x}); this.sameDiff = sameDiff; this.dimensions = dimension; addArgs(); }
public CumProd(SameDiff sameDiff, SDVariable x, boolean exclusive, boolean reverse, int... dimension) { super(null, sameDiff, new SDVariable[]{x}); this.sameDiff = sameDiff; this.dimensions = dimension; this.exclusive = exclusive; this.reverse = reverse; addArgs(); }
@Override public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String, AttrValue> attributesForNode, GraphDef graph) { TFGraphMapper.getInstance().initFunctionFromProperties(nodeDef.getOp(), this, attributesForNode, nodeDef, graph); addArgs(); }