Native functions v2

tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java

@@ -19,6 +19,8 @@
 
 import java.nio.charset.Charset;
 import java.util.List;
+import java.util.Map;
+import org.tensorflow.ConcreteFunction;
 import org.tensorflow.DeviceSpec;
 import org.tensorflow.EagerSession;
 import org.tensorflow.ExecutionEnvironment;
@@ -87,6 +89,7 @@
 import org.tensorflow.op.core.ExtractVolumePatches;
 import org.tensorflow.op.core.Fill;
 import org.tensorflow.op.core.Fingerprint;
+import org.tensorflow.op.core.Function;
 import org.tensorflow.op.core.Gather;
 import org.tensorflow.op.core.GatherNd;
 import org.tensorflow.op.core.GetSessionHandle;
@@ -1116,6 +1119,31 @@ public Bucketize bucketize(Operand<? extends TNumber> input, List<Float> boundar
  return Bucketize.create(scope, input, boundaries);
  }
 
+ /**
+ * Calls the function in an execution environment, adding its graph as a function if it isn't
+ * already present. Only works for functions with a single input and output.
+ *
+ * @param argument the argument to the call
+ * @return the output of the function
+ * @see ConcreteFunction#call(Ops, Operand)
+ */
+ public Operand<?> call(ConcreteFunction function, Operand<?> argument) {
+ return Function.call(scope, function, argument);
+ }
+
+ /**
+ * Calls the function in an execution environment, adding its graph as a function if it isn't
+ * already present. The inputs and outputs are keyed by the names set in the {@code Signature}.
+ *
+ * @param arguments the arguments to the call
+ * @return the outputs of the function
+ * @see ConcreteFunction#call(Ops, Map)
+ */
+ public Map<String, Operand<?>> call(ConcreteFunction function,
+ Map<String, Operand<?>> arguments) {
+ return Function.call(scope, function, arguments);
+ }
+
  /**
  * Clips tensor values to a specified min and max.
  * Given a tensor {@code t}, this operation returns a tensor of the same type and

tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Function.java

@@ -13,7 +13,7 @@
 // Once created and added to graphs, functions can be invoked by creating an
 // operation whose operation type matches the function name.
 @Opaque @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class)
-public class TF_Function extends Pointer {
+public class TF_Function extends org.tensorflow.internal.c_api.AbstractTF_Function {
  /** Empty constructor. Calls {@code super((Pointer)null)}. */
  public TF_Function() { super((Pointer)null); }
  /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */