Ops exported for sameDiff

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/functions/DifferentialFunctionFactory.java

@@ -33,6 +33,7 @@ import org.nd4j.linalg.api.blas.params.MMulTranspose;
 import org.nd4j.linalg.api.buffer.DataType;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.NoOp;
+import org.nd4j.linalg.api.ops.custom.*;
 import org.nd4j.linalg.api.ops.impl.broadcast.BiasAdd;
 import org.nd4j.linalg.api.ops.impl.broadcast.BiasAddGrad;
 import org.nd4j.linalg.api.ops.impl.controlflow.compat.Enter;
@@ -2649,6 +2650,33 @@ public class DifferentialFunctionFactory {
         return new NextIteration(sameDiff, x).outputVariable();
     }
 
+    public SDVariable adjustContrast(SDVariable in, SDVariable factor) {
+        return new AdjustContrast(sameDiff, in, factor).outputVariable();
+    }
+
+    public SDVariable adjustContrastV2(SDVariable in, SDVariable factor) {
+        return new AdjustContrastV2(sameDiff, in, factor).outputVariable();
+    }
+
+    public SDVariable bitCast(SDVariable in, SDVariable dataType) {
+        return new BitCast(sameDiff, in, dataType).outputVariable();
+    }
+
+    public SDVariable compareAndBitpack(SDVariable threshold) {
+        return new CompareAndBitpack(sameDiff, threshold).outputVariable();
+    }
+
+    public SDVariable divideNoNan(SDVariable in1, SDVariable in2) {
+        return new DivideNoNan(sameDiff, in1, in2).outputVariable();
+    }
+
+    public SDVariable drawBoundingBoxes(SDVariable boxes, SDVariable colors) {
+        return new DrawBoundingBoxes(sameDiff, boxes, colors).outputVariable();
+    }
+
+    public SDVariable fakeQuantWithMinMaxVarsPerChannel(SDVariable x, SDVariable min, SDVariable max) {
+        return new FakeQuantWithMinMaxVarsPerChannel(sameDiff,x,min,max).outputVariable();
+    }
 
     public String toString() {
         return "DifferentialFunctionFactory{methodNames=" + methodNames + "}";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/AdjustContrast.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.base.Preconditions;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
@@ -12,6 +14,10 @@ public class AdjustContrast extends BaseAdjustContrast {
         super(in, factor, out);
     }
 
+    public AdjustContrast(SameDiff sameDiff, SDVariable in, SDVariable factor) {
+        super(sameDiff,new SDVariable[]{in,factor});
+    }
+
     @Override
     public String opName() {
         return "adjust_contrast";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/AdjustContrastV2.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.base.Preconditions;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
@@ -12,6 +14,10 @@ public class AdjustContrastV2 extends BaseAdjustContrast {
         super(in, factor, out);
     }
 
+    public AdjustContrastV2(SameDiff sameDiff, SDVariable in, SDVariable factor) {
+        super( sameDiff,new SDVariable[]{in,factor});
+    }
+
     @Override
     public String opName() {
         return "adjust_contrast_v2";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/BaseAdjustContrast.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.base.Preconditions;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
@@ -16,4 +18,12 @@ public abstract class BaseAdjustContrast extends DynamicCustomOp {
 
         addTArgument(factor);
     }
+
+    public BaseAdjustContrast(SameDiff sameDiff, SDVariable[] vars) {
+        super("", sameDiff, vars);
+    }
+
+    public BaseAdjustContrast(SameDiff sameDiff, SDVariable in, SDVariable factor, SDVariable out) {
+        super(null, sameDiff, new SDVariable[]{in, factor, out});
+    }
 }

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/BitCast.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.linalg.api.buffer.DataType;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
@@ -14,6 +16,10 @@ public class BitCast extends DynamicCustomOp {
         iArguments.add(Long.valueOf(dataType));
     }
 
+    public BitCast(SameDiff sameDiff, SDVariable in, SDVariable dataType) {
+        super("", sameDiff, new SDVariable[]{in, dataType});
+    }
+
     @Override
     public String opName() {
         return "bitcast";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/CompareAndBitpack.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
 import org.nd4j.linalg.factory.Nd4j;
@@ -13,6 +15,10 @@ public class CompareAndBitpack extends DynamicCustomOp {
         outputArguments.add(out);
     }
 
+    public CompareAndBitpack(SameDiff sameDiff, SDVariable threshold) {
+        super("", sameDiff, new SDVariable[]{threshold});
+    }
+
     @Override
     public String opName() {
         return "compare_and_bitpack";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/DivideNoNan.java

@@ -1,6 +1,8 @@
 package org.nd4j.linalg.api.ops.custom;
 
 import org.apache.commons.math3.analysis.function.Divide;
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
 
@@ -14,6 +16,10 @@ public class DivideNoNan extends DynamicCustomOp {
         outputArguments.add(out);
     }
 
+    public DivideNoNan(SameDiff sameDiff, SDVariable in1, SDVariable in2) {
+        super("", sameDiff, new SDVariable[]{in1, in2});
+    }
+
     @Override
     public String opName() {
         return "divide_no_nan";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/DrawBoundingBoxes.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
 
@@ -14,6 +16,10 @@ public class DrawBoundingBoxes extends DynamicCustomOp {
         outputArguments.add(output);
     }
 
+    public DrawBoundingBoxes(SameDiff sameDiff, SDVariable boxes, SDVariable colors) {
+        super("", sameDiff, new SDVariable[]{boxes, colors});
+    }
+
     @Override
     public String opName() {
         return "draw_bounding_boxes";

nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/custom/FakeQuantWithMinMaxVarsPerChannel.java

@@ -1,5 +1,7 @@
 package org.nd4j.linalg.api.ops.custom;
 
+import org.nd4j.autodiff.samediff.SDVariable;
+import org.nd4j.autodiff.samediff.SameDiff;
 import org.nd4j.base.Preconditions;
 import org.nd4j.linalg.api.ndarray.INDArray;
 import org.nd4j.linalg.api.ops.DynamicCustomOp;
@@ -18,6 +20,10 @@ public class FakeQuantWithMinMaxVarsPerChannel extends DynamicCustomOp {
         outputArguments.add(output);
     }
 
+    public FakeQuantWithMinMaxVarsPerChannel(SameDiff sameDiff, SDVariable x, SDVariable min, SDVariable max) {
+        super("", sameDiff, new SDVariable[]{x, min, max});
+    }
+
     @Override
     public String opName() {
         return "fake_quant_with_min_max_vars_per_channel";