diff --git a/tensorflow/src/main/java/org/bytedeco/javacpp/presets/tensorflow.java b/tensorflow/src/main/java/org/bytedeco/javacpp/presets/tensorflow.java index b24ecf1f4fd..9af6e44c687 100644 --- a/tensorflow/src/main/java/org/bytedeco/javacpp/presets/tensorflow.java +++ b/tensorflow/src/main/java/org/bytedeco/javacpp/presets/tensorflow.java @@ -158,14 +158,21 @@ public void map(InfoMap infoMap) { .put(new Info("std::vector").pointerTypes("NodeOutVector").define()) .put(new Info("std::vector").pointerTypes("NodeVector").define()) .put(new Info("google::protobuf::Map").pointerTypes("StringAttrValueMap").define()) + .put(new Info("tensorflow::ops::NodeOut").valueTypes("@ByVal NodeBuilder.NodeOut", "Node")) .put(new Info("tensorflow::NodeBuilder::NodeOut").pointerTypes("NodeBuilder.NodeOut")) + .put(new Info("std::vector::iterator").skip()) + .put(new Info("std::vector::const_iterator").skip()) + .put(new Info("tensorflow::ops::Cast").cppTypes("class tensorflow::ops::Cast").pointerTypes("CastOp")) + .put(new Info("tensorflow::ops::Const").cppTypes("class tensorflow::ops::Const").pointerTypes("ConstOp")) + .put(new Info("mode_t").skip()) .put(new Info("tensorflow::gtl::ArraySlice").cast().pointerTypes("StringVector")) .put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("TensorVector")) .put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("TensorProtoVector")) .put(new Info("tensorflow::gtl::ArraySlice").cast().pointerTypes("TensorShapeVector")) + .put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("NodeOutVector")) .put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("NodeVector")) .put(new Info("tensorflow::gtl::iterator_range").pointerTypes("NeighborIterRange").define()) .put(new Info("tensorflow::gtl::iterator_range()").skip()) @@ -312,6 +319,15 @@ public static class EdgeLabelFunction extends FunctionPointer { public native @StdString BytePointer call(@Cast("const tensorflow::Edge*") Pointer node); } + public static class FactoryFn extends FunctionPointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public FactoryFn(Pointer p) { super(p); } + protected FactoryFn() { allocate(); } + private native void allocate(); + public native @Cast("tensorflow::FileSystem*") Pointer call(); + } + public static class ShapeInferenceFn extends FunctionPointer { static { Loader.load(); } /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ diff --git a/tensorflow/src/main/java/org/bytedeco/javacpp/tensorflow.java b/tensorflow/src/main/java/org/bytedeco/javacpp/tensorflow.java index 1b610a1e974..2593605e840 100644 --- a/tensorflow/src/main/java/org/bytedeco/javacpp/tensorflow.java +++ b/tensorflow/src/main/java/org/bytedeco/javacpp/tensorflow.java @@ -12924,15 +12924,15 @@ private native void allocate( @Namespace("tensorflow::ops") public static native Node SourceOp(@StdString String op_name, @Const @ByRef GraphDefBuilder.Options opts); // For adding an Op with one input to a GraphDefBuilder. -@Namespace("tensorflow::ops") public static native Node UnaryOp(@StdString BytePointer op_name, @ByVal @Cast("tensorflow::ops::NodeOut*") NodeBuilder.NodeOut input, +@Namespace("tensorflow::ops") public static native Node UnaryOp(@StdString BytePointer op_name, @ByVal NodeBuilder.NodeOut input, @Const @ByRef GraphDefBuilder.Options opts); -@Namespace("tensorflow::ops") public static native Node UnaryOp(@StdString String op_name, @ByVal @Cast("tensorflow::ops::NodeOut*") NodeBuilder.NodeOut input, +@Namespace("tensorflow::ops") public static native Node UnaryOp(@StdString String op_name, Node input, @Const @ByRef GraphDefBuilder.Options opts); // For adding an Op with two inputs to a GraphDefBuilder. -@Namespace("tensorflow::ops") public static native Node BinaryOp(@StdString BytePointer op_name, @ByVal @Cast("tensorflow::ops::NodeOut*") NodeBuilder.NodeOut a, @ByVal @Cast("tensorflow::ops::NodeOut*") NodeBuilder.NodeOut b, +@Namespace("tensorflow::ops") public static native Node BinaryOp(@StdString BytePointer op_name, @ByVal NodeBuilder.NodeOut a, @ByVal NodeBuilder.NodeOut b, @Const @ByRef GraphDefBuilder.Options opts); -@Namespace("tensorflow::ops") public static native Node BinaryOp(@StdString String op_name, @ByVal @Cast("tensorflow::ops::NodeOut*") NodeBuilder.NodeOut a, @ByVal @Cast("tensorflow::ops::NodeOut*") NodeBuilder.NodeOut b, +@Namespace("tensorflow::ops") public static native Node BinaryOp(@StdString String op_name, Node a, Node b, @Const @ByRef GraphDefBuilder.Options opts); // namespace ops