Java Code Examples for org.nd4j.autodiff.samediff.SameDiff#addArgsFor()

The following examples show how to use org.nd4j.autodiff.samediff.SameDiff#addArgsFor() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: BaseIndexAccumulation.java From nd4j with Apache License 2.0

6 votes

public BaseIndexAccumulation(SameDiff sameDiff,
                             SDVariable i_v,
                             int[] dimensions) {
    super(sameDiff,new Object[]{dimensions});
    if (i_v != null) {
        this.dimensions = dimensions;
        f().validateDifferentialFunctionsameDiff(i_v);
        sameDiff.addArgsFor(new SDVariable[]{i_v},this);
        if(Shape.isPlaceholderShape(i_v.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v.getVarName());
        }

        this.xVertexId = i_v.getVarName();
    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }
}

Example 2

Source File: BaseScalarBoolOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseScalarBoolOp(SameDiff sameDiff,
                        SDVariable i_v,
                        Number scalar,
                        boolean inPlace,
                        Object[] extraArgs) {
    super(sameDiff,inPlace,extraArgs);
    this.scalarValue = Nd4j.scalar(i_v.dataType(), scalar);
    if (i_v != null) {
        this.xVertexId = i_v.name();
        sameDiff.addArgsFor(new String[]{xVertexId},this);
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v, this);
    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }

}

Example 3

Source File: BaseBroadcastBoolOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseBroadcastBoolOp(SameDiff sameDiff,
                           SDVariable i_v1,
                           SDVariable i_v2,
                           boolean inPlace,
                           int[] dimension) {
    super(sameDiff, inPlace, new Object[]{i_v2});
    if (i_v1 != null && i_v2 != null) {
        this.sameDiff = sameDiff;
        this.inPlace = inPlace;
        this.dimension = dimension;
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);

    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }
}

Example 4

Source File: BaseScalarOp.java From nd4j with Apache License 2.0

6 votes

public BaseScalarOp(SameDiff sameDiff,
                    SDVariable i_v,
                    Number scalar,
                    boolean inPlace,
                    Object[] extraArgs) {
    super(sameDiff,inPlace,extraArgs);
    this.scalarValue = scalar;
    if (i_v != null) {
        this.xVertexId = i_v.getVarName();
        sameDiff.addArgsFor(new String[]{xVertexId},this);
        if(Shape.isPlaceholderShape(i_v.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v.getVarName());
        }
        f().validateDifferentialFunctionsameDiff(i_v);
    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }

}

Example 5

Source File: BaseAccumulation.java From nd4j with Apache License 2.0

6 votes

public BaseAccumulation(SameDiff sameDiff,
                        SDVariable i_v,
                        SDVariable i_v2,
                        int[] dimensions,boolean keepDims) {
    super(sameDiff,new Object[]{dimensions});
    if (i_v != null) {
        if(dimensions == null || dimensions.length < 1)
            dimensions = new int[] {Integer.MAX_VALUE};

        this.dimensions = dimensions;

        this.xVertexId = i_v.getVarName();
        this.yVertexId = i_v2.getVarName();
        f().validateDifferentialFunctionsameDiff(i_v);
        f().validateDifferentialFunctionsameDiff(i_v2);
        this.keepDims = keepDims;
        sameDiff.addArgsFor(new String[]{xVertexId,yVertexId},this);

    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }

}

Example 6

Source File: BaseBroadcastBoolOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseBroadcastBoolOp(SameDiff sameDiff,
                           SDVariable i_v,
                           long[] shape,
                           boolean inPlace,
                           int[] dimension,
                           Object[] extraArgs) {
    super(sameDiff, inPlace, extraArgs);
    this.dimension = dimension;
    if (i_v != null) {
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v, this);
        sameDiff.addArgsFor(new SDVariable[]{i_v},this);


    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }


}

Example 7

Source File: BaseBroadcastBoolOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseBroadcastBoolOp(SameDiff sameDiff,
                           SDVariable i_v1,
                           SDVariable i_v2,
                           int[] dimension,
                           Object[] extraArgs) {
    super(sameDiff, extraArgs);
    this.dimension = dimension;
    if (i_v1 != null && i_v2 != null) {
        this.sameDiff = sameDiff;
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);

    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }


}

Example 8

Source File: BaseIndexAccumulation.java From deeplearning4j with Apache License 2.0

6 votes

public BaseIndexAccumulation(SameDiff sameDiff,
                             SDVariable i_v,
                             SDVariable i_v2,
                             boolean keepDims,
                             int[] dimensions) {
    super(sameDiff,null);
    if (i_v != null) {
        this.dimensions = dimensions;
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v, this);
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v2, this);
        this.xVertexId = i_v.name();
        this.yVertexId = i_v2.name();
        sameDiff.addArgsFor(new SDVariable[]{i_v,i_v2},this);
    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }
    this.keepDims = keepDims;
    defineDimensions(dimensions);
}

Example 9

Source File: BaseBroadcastOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseBroadcastOp(SameDiff sameDiff,
                       SDVariable i_v1,
                       SDVariable i_v2,
                       int[] dimension,
                       Object[] extraArgs) {
    super(sameDiff, extraArgs);
    this.dimension = dimension;
    if (i_v1 != null && i_v2 != null) {
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v1, this);
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v2, this);

        this.sameDiff = sameDiff;
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);

    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }


}

Example 10

Source File: BaseTransformOp.java From nd4j with Apache License 2.0

6 votes

public BaseTransformOp(SameDiff sameDiff,
                       SDVariable i_v,
                       long[] shape,
                       boolean inPlace,
                       Object[] extraArgs) {
    super(sameDiff,inPlace,extraArgs);

    if (i_v != null) {
        f().validateDifferentialFunctionsameDiff(i_v);
        this.xVertexId = i_v.getVarName();
        sameDiff.addArgsFor(new SDVariable[]{i_v},this);
        if(i_v.getShape() != null) {
            this.n = ArrayUtil.prod(i_v.getShape());
        }

        if(Shape.isPlaceholderShape(i_v.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v.getVarName());
        }


    } else {
        throw new IllegalArgumentException("Input must not null variable.");
    }

}

Example 11

Source File: BaseTransformOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseTransformOp(SameDiff sameDiff,
                       SDVariable i_v1,
                       SDVariable i_v2,
                       boolean inPlace) {
    super(sameDiff,inPlace,new Object[] {i_v2});
    if (i_v1 != null && i_v2 != null) {
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v1, this);
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v2, this);
        this.sameDiff = sameDiff;
        this.inPlace = inPlace;
        this.xVertexId = i_v1.name();
        this.yVertexId = i_v2.name();
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);
    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }


}

Example 12

Source File: BaseBroadcastOp.java From nd4j with Apache License 2.0

6 votes

public BaseBroadcastOp(SameDiff sameDiff,
                       SDVariable i_v,
                       long[] shape,
                       boolean inPlace,
                       int[] dimension,
                       Object[] extraArgs) {
    super(sameDiff, inPlace, extraArgs);
    this.dimension = dimension;
    if (i_v != null) {
        f().validateDifferentialFunctionsameDiff(i_v);
        sameDiff.addArgsFor(new SDVariable[]{i_v},this);


    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }


}

Example 13

Source File: BaseTransformOp.java From deeplearning4j with Apache License 2.0

6 votes

public BaseTransformOp(SameDiff sameDiff,
                       SDVariable i_v1,
                       SDVariable i_v2,
                       Object[] extraArgs) {
    super(sameDiff,extraArgs);
    if (i_v1 != null && i_v2 != null) {

        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v1, this);
        SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v2, this);
        this.sameDiff = sameDiff;
        this.xVertexId = i_v1.name();
        this.yVertexId = i_v2.name();
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);
    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }

}

Example 14

Source File: BaseBroadcastOp.java From nd4j with Apache License 2.0

6 votes

public BaseBroadcastOp(SameDiff sameDiff,
                       SDVariable i_v1,
                       SDVariable i_v2,
                       boolean inPlace,
                       int[] dimension) {
    super(sameDiff, inPlace, new Object[]{i_v2});
    if (i_v1 != null && i_v2 != null) {
        f().validateDifferentialFunctionsameDiff(i_v1);
        f().validateDifferentialFunctionsameDiff(i_v2);
        this.sameDiff = sameDiff;
        this.inPlace = inPlace;
        this.dimension = dimension;
        if(Shape.isPlaceholderShape(i_v1.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v1.getVarName());
        }

        if(Shape.isPlaceholderShape(i_v2.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v2.getVarName());
        }
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);

    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }

}

Example 15

Source File: BaseIndexAccumulation.java From nd4j with Apache License 2.0

6 votes

public BaseIndexAccumulation(SameDiff sameDiff,
                             SDVariable i_v,
                             SDVariable i_v2,
                             int[] dimensions) {
    super(sameDiff,new Object[]{dimensions});
    if (i_v != null) {
        this.dimensions = dimensions;
        f().validateDifferentialFunctionsameDiff(i_v);
        f().validateDifferentialFunctionsameDiff(i_v2);
        this.xVertexId = i_v.getVarName();
        this.yVertexId = i_v2.getVarName();
        sameDiff.addArgsFor(new SDVariable[]{i_v,i_v2},this);

        if(Shape.isPlaceholderShape(i_v.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v.getVarName());
        }

        if(Shape.isPlaceholderShape(i_v2.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v2.getVarName());
        }
    } else {
        throw new IllegalArgumentException("Input not null variable.");
    }
}

Example 16

Source File: BaseScalarOp.java From deeplearning4j with Apache License 2.0

5 votes

public BaseScalarOp(SameDiff sameDiff,
                    @NonNull SDVariable i_v,
                    Number scalar,
                    boolean inPlace,
                    Object[] extraArgs) {
    super(sameDiff,inPlace,extraArgs);
    this.scalarValue = Nd4j.scalar(i_v.dataType(), scalar);
    this.xVertexId = i_v.name();
    sameDiff.addArgsFor(new String[]{xVertexId},this);
    SameDiffUtils.validateDifferentialFunctionSameDiff(sameDiff, i_v, this);
}

Example 17

Source File: DepthwiseConv2D.java From nd4j with Apache License 2.0

5 votes

@Builder(builderMethodName = "builder")
public DepthwiseConv2D(SameDiff sameDiff,
                       SDVariable[] inputFunctions,
                       INDArray[] inputArrays, INDArray[] outputs,
                       Conv2DConfig config) {
    super(null, inputArrays, outputs);
    this.sameDiff = sameDiff;
    this.config = config;
    addArgs();
    sameDiff.putFunctionForId(this.getOwnName(), this);    //Normally called in DynamicCustomOp constructor, via setInstanceId - but sameDiff field is null at that point
    sameDiff.addArgsFor(inputFunctions, this);
}

Example 18

Source File: BaseTransformOp.java From nd4j with Apache License 2.0

5 votes

public BaseTransformOp(SameDiff sameDiff,
                       SDVariable i_v1,
                       SDVariable i_v2,
                       boolean inPlace) {
    super(sameDiff,inPlace,new Object[] {i_v2});
    if (i_v1 != null && i_v2 != null) {
        f().validateDifferentialFunctionsameDiff(i_v1);
        f().validateDifferentialFunctionsameDiff(i_v2);
        this.sameDiff = sameDiff;
        this.inPlace = inPlace;
        this.xVertexId = i_v1.getVarName();
        this.yVertexId = i_v2.getVarName();
        sameDiff.addArgsFor(new SDVariable[]{i_v1,i_v2},this);
        if(Shape.isPlaceholderShape(i_v1.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v1.getVarName());
        }

        if(Shape.isPlaceholderShape(i_v2.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v2.getVarName());
        }
        if(i_v1.getShape() != null)
            this.n = ArrayUtil.prod(i_v1.getShape());


    } else {
        throw new IllegalArgumentException("Input not null variables.");
    }


}

Example 19

Source File: BaseRandomOp.java From nd4j with Apache License 2.0

5 votes

public BaseRandomOp(SameDiff sameDiff,
                        SDVariable i_v) {
    if (i_v != null) {
        this.sameDiff = sameDiff;
        this.xVertexId = i_v.getVarName();
        sameDiff.addArgsFor(new String[]{xVertexId},this);
        if(Shape.isPlaceholderShape(i_v.getShape())) {
            sameDiff.addPropertyToResolve(this,i_v.getVarName());
        }
    } else {
        throw new IllegalArgumentException("Input can't be null.");
    }
}

Example 20

Source File: DifferentialFunction.java From nd4j with Apache License 2.0

5 votes

/**
 * Add the various arguments for
 * this function
 * @param sameDiff
 * @param inPlace
 * @param args
 */
public DifferentialFunction(SameDiff sameDiff, boolean inPlace, SDVariable[] args) {
    this.sameDiff = sameDiff;
    this.inPlace = inPlace;
    setInstanceId();
    if(sameDiff != null) {
        sameDiff.addArgsFor(args, this);
        for (int i = 0; i < args.length; i++) {
            if (args[i].isPlaceHolder()) {
                sameDiff.addPropertyToResolve(this, args[i].getVarName());
            }
        }
    }
}