diff --git a/tensorflow-core/tensorflow-core-api/WORKSPACE b/tensorflow-core/tensorflow-core-api/WORKSPACE index f6aa07115ed..013338cfbba 100644 --- a/tensorflow-core/tensorflow-core-api/WORKSPACE +++ b/tensorflow-core/tensorflow-core-api/WORKSPACE @@ -12,6 +12,8 @@ http_archive( # ":tensorflow-macosx.patch", # ":tensorflow-windows.patch", # https://github.com/tensorflow/tensorflow/issues/25213 ":tensorflow-proto.patch", + ":custom-grad-helpers.patch", + ":custom-grad-symbols.patch", ], patch_tool = "patch", patch_args = ["-p1"], diff --git a/tensorflow-core/tensorflow-core-api/external/custom-grad-helpers.patch b/tensorflow-core/tensorflow-core-api/external/custom-grad-helpers.patch new file mode 100644 index 00000000000..aedf53e2e56 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/external/custom-grad-helpers.patch @@ -0,0 +1,57 @@ +diff --git a/tensorflow/c/c_api.cc b/tensorflow/c/c_api.cc +index f3bf7b98a1e6b..c9194c36c116b 100644 +--- a/tensorflow/c/c_api.cc ++++ b/tensorflow/c/c_api.cc +@@ -782,9 +782,9 @@ void TF_GraphGetTensorShape(TF_Graph* graph, TF_Output output, int64_t* dims, + + extern "C" { + +-static TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, +- const char* op_type, +- const char* oper_name) ++TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, ++ const char* op_type, ++ const char* oper_name) + TF_EXCLUSIVE_LOCKS_REQUIRED(graph->mu) { + return new TF_OperationDescription(graph, op_type, oper_name); + } +@@ -1041,8 +1041,8 @@ void TF_SetAttrValueProto(TF_OperationDescription* desc, const char* attr_name, + status->status = Status::OK(); + } + +-static TF_Operation* TF_FinishOperationLocked(TF_OperationDescription* desc, +- TF_Status* status) ++TF_Operation* TF_FinishOperationLocked(TF_OperationDescription* desc, ++ TF_Status* status) + TF_EXCLUSIVE_LOCKS_REQUIRED(desc->graph->mu) { + Node* ret = nullptr; + +diff --git a/tensorflow/c/c_api.h b/tensorflow/c/c_api.h +index 705cf85e0512f..fb746dd4af94f 100644 +--- a/tensorflow/c/c_api.h ++++ b/tensorflow/c/c_api.h +@@ -255,6 +255,12 @@ TF_CAPI_EXPORT extern void TF_GraphGetTensorShape(TF_Graph* graph, + int64_t* dims, int num_dims, + TF_Status* status); + ++// TF_NewOperation, but without locking the graph. ++// Should prefer TF_NewOperation when possible. ++TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperationLocked(TF_Graph* graph, ++ const char* op_type, ++ const char* oper_name); ++ + // Operation will only be added to *graph when TF_FinishOperation() is + // called (assuming TF_FinishOperation() does not return an error). + // *graph must not be deleted until after TF_FinishOperation() is +@@ -406,6 +412,11 @@ TF_CAPI_EXPORT extern void TF_SetAttrValueProto(TF_OperationDescription* desc, + size_t proto_len, + TF_Status* status); + ++// TF_FinishOperation, but without locking the graph. ++// TF_FinishOperation should be preferred when possible. ++TF_CAPI_EXPORT extern TF_Operation* TF_FinishOperationLocked(TF_OperationDescription* desc, ++ TF_Status* status); ++ + // If this function succeeds: + // * *status is set to an OK value, + // * a TF_Operation is added to the graph, diff --git a/tensorflow-core/tensorflow-core-api/external/custom-grad-symbols.patch b/tensorflow-core/tensorflow-core-api/external/custom-grad-symbols.patch new file mode 100644 index 00000000000..c47b9da0127 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/external/custom-grad-symbols.patch @@ -0,0 +1,151 @@ +Index: tensorflow/tools/def_file_filter/BUILD +IDEA additional info: +Subsystem: com.intellij.openapi.diff.impl.patch.CharsetEP +<+>UTF-8 +=================================================================== +diff --git a/tensorflow/tools/def_file_filter/BUILD b/tensorflow/tools/def_file_filter/BUILD +--- a/tensorflow/tools/def_file_filter/BUILD (revision 5e5cc35b4c0f629a1e092b540fdf2b63367aa5ad) ++++ b/tensorflow/tools/def_file_filter/BUILD (date 1629063191558) +@@ -12,3 +12,8 @@ + name = "symbols_pybind", + srcs = ["symbols_pybind.txt"], + ) ++ ++filegroup( ++ name = "symbols_java", ++ srcs = ["symbols_java.txt"], ++) +Index: tensorflow/BUILD +IDEA additional info: +Subsystem: com.intellij.openapi.diff.impl.patch.CharsetEP +<+>UTF-8 +=================================================================== +diff --git a/tensorflow/BUILD b/tensorflow/BUILD +--- a/tensorflow/BUILD (revision 5e5cc35b4c0f629a1e092b540fdf2b63367aa5ad) ++++ b/tensorflow/BUILD (date 1629063361078) +@@ -1069,13 +1069,20 @@ + # the dynamic libraries of custom ops can find it at runtime. + genrule( + name = "tensorflow_filtered_def_file", +- srcs = [":tensorflow_def_file"], ++ srcs = [ ++ ":tensorflow_def_file", ++ ":java_symbol_target_libs_file", ++ ":win_lib_files_for_java_exported_symbols", ++ "//tensorflow/tools/def_file_filter:symbols_java", ++ ], + outs = ["tensorflow_filtered_def_file.def"], + cmd = select({ + "//tensorflow:windows": """ + $(location @local_config_def_file_filter//:def_file_filter) \\ + --input $(location :tensorflow_def_file) \\ +- --output $@ ++ --output $@ \\ ++ --symbols $(location //tensorflow/tools/def_file_filter:symbols_java) \\ ++ --lib_paths_file $(location :java_symbol_target_libs_file) + """, + "//conditions:default": "touch $@", # Just a placeholder for Unix platforms + }), +@@ -1083,6 +1090,34 @@ + visibility = ["//visibility:public"], + ) + ++# Write to a file a list of all cc_library targets that we need for exporting symbols on Windows. ++genrule( ++ name = "java_symbol_target_libs_file", ++ srcs = [":win_lib_files_for_java_exported_symbols"], ++ outs = ["java_symbol_target_libs_file.txt"], ++ cmd = select({ ++ "//tensorflow:windows": """ ++ for SRC in $(SRCS); do ++ echo $$SRC | sed 's/third_party\\///g' >> $@ ++ done ++ """, ++ "//conditions:default": "touch $@", # Just a placeholder for Unix platforms ++ }), ++ visibility = ["//visibility:public"], ++) ++ ++filegroup( ++ name = "win_lib_files_for_java_exported_symbols", ++ srcs = [ ++ "//tensorflow/cc:scope", ++ "//tensorflow/cc:grad_op_registry", ++ "//tensorflow/c:tf_status_helper", ++ "//tensorflow/cc:ops" ++ ], ++ visibility = ["//visibility:private"], ++) ++ ++ + # The interface library (tensorflow.dll.if.lib) for linking tensorflow DLL library (tensorflow.dll) on Windows. + # To learn more about import library (called interface library in Bazel): + # https://docs.microsoft.com/en-us/cpp/build/linking-an-executable-to-a-dll?view=vs-2017#linking-implicitly +Index: tensorflow/tools/def_file_filter/BUILD.tpl +IDEA additional info: +Subsystem: com.intellij.openapi.diff.impl.patch.CharsetEP +<+>UTF-8 +=================================================================== +diff --git a/tensorflow/tools/def_file_filter/BUILD.tpl b/tensorflow/tools/def_file_filter/BUILD.tpl +--- a/tensorflow/tools/def_file_filter/BUILD.tpl (revision 5e5cc35b4c0f629a1e092b540fdf2b63367aa5ad) ++++ b/tensorflow/tools/def_file_filter/BUILD.tpl (date 1629063191583) +@@ -18,3 +18,8 @@ + name = "symbols_pybind", + srcs = ["symbols_pybind.txt"], + ) ++ ++filegroup( ++ name = "symbols_java", ++ srcs = ["symbols_java.txt"], ++) +Index: tensorflow/tools/def_file_filter/symbols_java.txt +IDEA additional info: +Subsystem: com.intellij.openapi.diff.impl.patch.CharsetEP +<+>UTF-8 +=================================================================== +diff --git a/tensorflow/tools/def_file_filter/symbols_java.txt b/tensorflow/tools/def_file_filter/symbols_java.txt +new file mode 100644 +--- /dev/null (date 1629063607794) ++++ b/tensorflow/tools/def_file_filter/symbols_java.txt (date 1629063607794) +@@ -0,0 +1,26 @@ ++[//tensorflow/cc:scope] # scope ++tensorflow::Scope::graph ++tensorflow::Scope::ok ++tensorflow::Scope::UpdateBuilder ++tensorflow::Scope::GetUniqueNameForOp ++tensorflow::Scope::ExitOnError ++tensorflow::Scope::WithDevice ++tensorflow::Scope::WithNoControlDependencies ++tensorflow::Scope::WithControlDependencies ++tensorflow::Scope::NewSubScope ++tensorflow::Scope::NewRootScope ++tensorflow::Scope::operator= ++tensorflow::Scope::~Scope ++tensorflow::Scope::Scope ++ ++[//tensorflow/cc:ops] ++tensorflow::Operation::Operation ++ ++[//tensorflow/cc:grad_op_registry] # custom gradients for graph ++tensorflow::ops::GradOpRegistry::Global ++tensorflow::ops::GradOpRegistry::Lookup ++tensorflow::ops::GradOpRegistry::Register ++ ++[//tensorflow/c:tf_status_helper] # status helpers ++tensorflow::Set_TF_Status_from_Status ++tensorflow::StatusFromTF_Status +=================================================================== +diff --git a/tensorflow/tools/def_file_filter/def_file_filter.py.tpl b/tensorflow/tools/def_file_filter/def_file_filter.py.tpl +--- a/tensorflow/tools/def_file_filter/def_file_filter.py.tpl (revision 919f693420e35d00c8d0a42100837ae3718f7927) ++++ b/tensorflow/tools/def_file_filter/def_file_filter.py.tpl (date 1632048268359) +@@ -143,8 +143,8 @@ + re_filter_comp = re.compile(r"{}".format(re_filter)) + + # Filter out symbol from the split line (`sym_split` in the for loop below). +- sym_line_filter = r".*\s+\| (.*) \(.*" +- sym_line_filter_anomaly = r".*\s+\| (.*)" ++ sym_line_filter = r".*\s+\| (.*?) \(.*" ++ sym_line_filter_anomaly = r".*\s+\| (.*?)" + + for sym_line in sym_split: + if re_filter_comp.search(sym_line): diff --git a/tensorflow-core/tensorflow-core-api/pom.xml b/tensorflow-core/tensorflow-core-api/pom.xml index 9f23757e83d..5c35a6e2c62 100644 --- a/tensorflow-core/tensorflow-core-api/pom.xml +++ b/tensorflow-core/tensorflow-core-api/pom.xml @@ -143,6 +143,19 @@ + + maven-resources-plugin + 3.1.0 + + + javacpp-parser + generate-sources + + resources + + + + maven-compiler-plugin 3.8.0 @@ -212,6 +225,11 @@ ${project.basedir}/ ${project.basedir}/bazel-${project.artifactId}/external/org_tensorflow/ + ${project.basedir}/bazel-bin/external/org_tensorflow/ + ${project.basedir}/bazel-${project.artifactId}/external/com_google_absl/ + ${project.basedir}/bazel-${project.artifactId}/external/eigen_archive/ + ${project.basedir}/bazel-${project.artifactId}/external/com_google_protobuf/src/ + ${project.basedir}/target/classes/org/tensorflow/internal/c_api/include/ ${project.basedir}/bazel-bin/external/llvm_openmp/ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java index 8da481a6c72..4397483d606 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java @@ -367,10 +367,10 @@ public final class Ops { public final SparseOps sparse; - public final BitwiseOps bitwise; - public final TpuOps tpu; + public final BitwiseOps bitwise; + public final MathOps math; public final AudioOps audio; @@ -383,7 +383,7 @@ public final class Ops { private final Scope scope; - private Ops(Scope scope) { + Ops(Scope scope) { this.scope = scope; nn = new NnOps(this); summary = new SummaryOps(this); @@ -398,8 +398,8 @@ private Ops(Scope scope) { random = new RandomOps(this); strings = new StringsOps(this); sparse = new SparseOps(this); - bitwise = new BitwiseOps(this); tpu = new TpuOps(this); + bitwise = new BitwiseOps(this); math = new MathOps(this); audio = new AudioOps(this); signal = new SignalOps(this); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/GradFunc.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/GradFunc.java new file mode 100644 index 00000000000..df5ceb98746 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/GradFunc.java @@ -0,0 +1,25 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + + +/** GradFunc is the signature for all gradient functions in GradOpRegistry. + * Implementations should add operations to compute the gradient outputs of + * 'op' (returned in 'grad_outputs') using 'scope' and 'grad_inputs'. */ +@Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class GradFunc extends FunctionPointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public GradFunc(Pointer p) { super(p); } + protected GradFunc() { allocate(); } + private native void allocate(); + public native @ByVal NativeStatus call(@Const @ByRef TF_Scope scope, @Const @ByRef NativeOperation op, + @Const @ByRef NativeOutputVector grad_inputs, + NativeOutputVector grad_outputs); +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/GradOpRegistry.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/GradOpRegistry.java new file mode 100644 index 00000000000..7a1d69bafbd --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/GradOpRegistry.java @@ -0,0 +1,48 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + + +/** GradOpRegistry maintains a static registry of gradient functions. + * Gradient functions are indexed in the registry by the forward op name (i.e. + * "MatMul" -> MatMulGrad func). */ +@Namespace("tensorflow::ops") @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class GradOpRegistry extends Pointer { + static { Loader.load(); } + /** Default native constructor. */ + public GradOpRegistry() { super((Pointer)null); allocate(); } + /** Native array allocator. Access with {@link Pointer#position(long)}. */ + public GradOpRegistry(long size) { super((Pointer)null); allocateArray(size); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public GradOpRegistry(Pointer p) { super(p); } + private native void allocate(); + private native void allocateArray(long size); + @Override public GradOpRegistry position(long position) { + return (GradOpRegistry)super.position(position); + } + @Override public GradOpRegistry getPointer(long i) { + return new GradOpRegistry((Pointer)this).offsetAddress(i); + } + + /** Registers 'func' as the gradient function for 'op'. + * Returns true if registration was successful, check fails otherwise. */ + public native @Cast("bool") boolean Register(@StdString BytePointer op, GradFunc func); + public native @Cast("bool") boolean Register(@StdString String op, GradFunc func); + + /** Sets 'func' to the gradient function for 'op' and returns Status OK if + * the gradient function for 'op' exists in the registry. + * Note that 'func' can be null for ops that have registered no-gradient with + * the registry. + * Returns error status otherwise. */ + public native @ByVal NativeStatus Lookup(@StdString BytePointer op, @ByPtrPtr GradFunc func); + public native @ByVal NativeStatus Lookup(@StdString String op, @ByPtrPtr GradFunc func); + + /** Returns a pointer to the global gradient function registry. */ + public static native GradOpRegistry Global(); +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NameMap.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NameMap.java new file mode 100644 index 00000000000..0be7fab2798 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NameMap.java @@ -0,0 +1,41 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + +@Name("std::unordered_map") @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NameMap extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NameMap(Pointer p) { super(p); } + public NameMap() { allocate(); } + private native void allocate(); + public native @Name("operator =") @ByRef NameMap put(@ByRef NameMap x); + + public boolean empty() { return size() == 0; } + public native long size(); + + @Index public native Node get(@StdString BytePointer i); + public native NameMap put(@StdString BytePointer i, Node value); + + public native void erase(@ByVal Iterator pos); + public native @ByVal Iterator begin(); + public native @ByVal Iterator end(); + @NoOffset @Name("iterator") public static class Iterator extends Pointer { + public Iterator(Pointer p) { super(p); } + public Iterator() { } + + public native @Name("operator ++") @ByRef Iterator increment(); + public native @Name("operator ==") boolean equals(@ByRef Iterator it); + public native @Name("operator *().first") @MemberGetter @StdString BytePointer first(); + public native @Name("operator *().second") @MemberGetter @Const Node second(); + } + + public native long erase(@StdString BytePointer key); +} + diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeGraphPointer.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeGraphPointer.java new file mode 100644 index 00000000000..b947a4b322f --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeGraphPointer.java @@ -0,0 +1,18 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + + +@Name("tensorflow::Graph") @Opaque @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NativeGraphPointer extends Pointer { + /** Empty constructor. Calls {@code super((Pointer)null)}. */ + public NativeGraphPointer() { super((Pointer)null); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NativeGraphPointer(Pointer p) { super(p); } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOperation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOperation.java new file mode 100644 index 00000000000..af771a0aa12 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOperation.java @@ -0,0 +1,49 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + + +/** \addtogroup core + * \{ +

+ * Represents a node in the computation graph. */ +@Name("tensorflow::Operation") @NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NativeOperation extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NativeOperation(Pointer p) { super(p); } + /** Native array allocator. Access with {@link Pointer#position(long)}. */ + public NativeOperation(long size) { super((Pointer)null); allocateArray(size); } + private native void allocateArray(long size); + @Override public NativeOperation position(long position) { + return (NativeOperation)super.position(position); + } + @Override public NativeOperation getPointer(long i) { + return new NativeOperation((Pointer)this).offsetAddress(i); + } + + public NativeOperation() { super((Pointer)null); allocate(); } + private native void allocate(); + public NativeOperation(Node n) { super((Pointer)null); allocate(n); } + private native void allocate(Node n); + + + + + + + + + + public native Node node(); + + + + public native @Cast("bool") @Name("operator ==") boolean equals(@Const @ByRef NativeOperation other); +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOutput.java new file mode 100644 index 00000000000..55bcac60f5b --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOutput.java @@ -0,0 +1,45 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + + +/** Represents a tensor value produced by an Operation. */ +@Name("tensorflow::Output") @NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NativeOutput extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NativeOutput(Pointer p) { super(p); } + /** Native array allocator. Access with {@link Pointer#position(long)}. */ + public NativeOutput(long size) { super((Pointer)null); allocateArray(size); } + private native void allocateArray(long size); + @Override public NativeOutput position(long position) { + return (NativeOutput)super.position(position); + } + @Override public NativeOutput getPointer(long i) { + return new NativeOutput((Pointer)this).offsetAddress(i); + } + + public NativeOutput() { super((Pointer)null); allocate(); } + private native void allocate(); + public NativeOutput(Node n) { super((Pointer)null); allocate(n); } + private native void allocate(Node n); + public NativeOutput(Node n, @Cast("tensorflow::int32") int index) { super((Pointer)null); allocate(n, index); } + private native void allocate(Node n, @Cast("tensorflow::int32") int index); + public NativeOutput(@Const @ByRef NativeOperation op, @Cast("tensorflow::int32") int index) { super((Pointer)null); allocate(op, index); } + private native void allocate(@Const @ByRef NativeOperation op, @Cast("tensorflow::int32") int index); + + public native @ByVal NativeOperation op(); + public native Node node(); + public native @Cast("tensorflow::int32") int index(); + + public native @StdString BytePointer name(); + public native @Cast("bool") @Name("operator ==") boolean equals(@Const @ByRef NativeOutput other); + + +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOutputVector.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOutputVector.java new file mode 100644 index 00000000000..9dc9dd5b971 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeOutputVector.java @@ -0,0 +1,79 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + +@Name("std::vector") @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NativeOutputVector extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NativeOutputVector(Pointer p) { super(p); } + public NativeOutputVector(NativeOutput value) { this(1); put(0, value); } + public NativeOutputVector(NativeOutput ... array) { this(array.length); put(array); } + public NativeOutputVector() { allocate(); } + public NativeOutputVector(long n) { allocate(n); } + private native void allocate(); + private native void allocate(@Cast("size_t") long n); + public native @Name("operator =") @ByRef NativeOutputVector put(@ByRef NativeOutputVector x); + + public boolean empty() { return size() == 0; } + public native long size(); + public void clear() { resize(0); } + public native void resize(@Cast("size_t") long n); + + @Index(function = "at") public native @ByRef NativeOutput get(@Cast("size_t") long i); + public native NativeOutputVector put(@Cast("size_t") long i, NativeOutput value); + + public native @ByVal Iterator insert(@ByVal Iterator pos, @ByRef NativeOutput value); + public native @ByVal Iterator erase(@ByVal Iterator pos); + public native @ByVal Iterator begin(); + public native @ByVal Iterator end(); + @NoOffset @Name("iterator") public static class Iterator extends Pointer { + public Iterator(Pointer p) { super(p); } + public Iterator() { } + + public native @Name("operator ++") @ByRef Iterator increment(); + public native @Name("operator ==") boolean equals(@ByRef Iterator it); + public native @Name("operator *") @ByRef @Const NativeOutput get(); + } + + public NativeOutput[] get() { + NativeOutput[] array = new NativeOutput[size() < Integer.MAX_VALUE ? (int)size() : Integer.MAX_VALUE]; + for (int i = 0; i < array.length; i++) { + array[i] = get(i); + } + return array; + } + @Override public String toString() { + return java.util.Arrays.toString(get()); + } + + public NativeOutput pop_back() { + long size = size(); + NativeOutput value = get(size - 1); + resize(size - 1); + return value; + } + public NativeOutputVector push_back(NativeOutput value) { + long size = size(); + resize(size + 1); + return put(size, value); + } + public NativeOutputVector put(NativeOutput value) { + if (size() != 1) { resize(1); } + return put(0, value); + } + public NativeOutputVector put(NativeOutput ... array) { + if (size() != array.length) { resize(array.length); } + for (int i = 0; i < array.length; i++) { + put(i, array[i]); + } + return this; + } +} + diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeStatus.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeStatus.java new file mode 100644 index 00000000000..7e21aae659b --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NativeStatus.java @@ -0,0 +1,94 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + +// #endif + +/** \ingroup core + * Denotes success or failure of a call in Tensorflow. */ +@Name("tensorflow::Status") @NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NativeStatus extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NativeStatus(Pointer p) { super(p); } + + /** Create a success status. */ + + /** \brief Create a status with the specified error code and msg as a + * human-readable string containing more detailed information. */ + + /** \brief Create a status with the specified error code, msg, and stack trace + * as a human-readable string containing more detailed information. */ +// #ifndef SWIG +// #endif + + /** Copy the specified status. */ + public native @ByRef @Name("operator =") NativeStatus put(@Const @ByRef NativeStatus s); +// #ifndef SWIG +// #endif // SWIG + + public static native @ByVal NativeStatus OK(); + + /** Returns true iff the status indicates success. */ + public native @Cast("bool") boolean ok(); + + + + public native @StdString BytePointer error_message(); + + public native @Cast("bool") @Name("operator ==") boolean equals(@Const @ByRef NativeStatus x); + + /// + public native @Cast("bool") @Name("operator !=") boolean notEquals(@Const @ByRef NativeStatus x); + + /** \brief If {@code ok()}, stores {@code new_status} into {@code *this}. If {@code !ok()}, + * preserves the current status, but may augment with additional + * information about {@code new_status}. + * + * Convenient way of keeping track of the first error encountered. + * Instead of: + * {@code if (overall_status.ok()) overall_status = new_status} + * Use: + * {@code overall_status.Update(new_status);} */ + public native void Update(@Const @ByRef NativeStatus new_status); + + /** \brief Return a string representation of this status suitable for + * printing. Returns the string {@code "OK"} for success. */ + public native @StdString BytePointer ToString(); + + // Ignores any errors. This method does nothing except potentially suppress + // complaints from any tools that are checking that errors are not dropped on + // the floor. + public native void IgnoreError(); + + // The Payload-related APIs are cloned from absl::Status. + // + // Returns the payload of a status given its unique `type_url` key, if + // present. Returns an empty StringPiece if the status is ok, or if the key is + // not present. + + + // Sets the payload for a non-ok status using a `type_url` key, overwriting + // any existing payload for that `type_url`. + // + // This function does nothing if the Status is ok. + + + // Erases the payload corresponding to the `type_url` key. Returns `true` if + // the payload was present. + + + // Returns all the payload information. + // Returns an empty result if status is ok. + + + // Copies all the payloads using the input and discards existing payloads. + // Does nothing if status is ok or 'payloads' is empty. + +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/Node.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/Node.java new file mode 100644 index 00000000000..f3baf914963 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/Node.java @@ -0,0 +1,155 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + +// Parsed from tensorflow/core/graph/graph.h + +@Name("tensorflow::Node") @NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class Node extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public Node(Pointer p) { super(p); } + + public native @StdString BytePointer DebugString(); + public native int id(); + public native int cost_id(); + public native @StdString BytePointer name(); + public native void set_name(@StdString BytePointer name); + public native void set_name(@StdString String name); + public native @StdString BytePointer type_string(); + + // def() provides the NodeDef the user supplied, but the specifics + // of this Node may have changed due to placement, optimization, etc. + // In particular: + // * def().name() will match name(); + // * def().op() will match type_string() and op_def().name(); + // * def().input() is not reliable, use "in_edges()" below instead; + // * def().device() is the "user's requested device" and may not match + // the actual assigned device, see assigned_device_name() below; + // * def().attr() is authoritative. + // TODO(irving): Replace with NodeInfo. + + // TODO(mdan): This is only used by control_flow_deps_o_chains. Remove? + + // input and output types + public native @Cast("tensorflow::int32") int num_inputs(); + + public native @Cast("tensorflow::int32") int num_outputs(); + + // The device requested by the user. For the actual assigned device, + // use assigned_device_name() below. + public native @StdString BytePointer requested_device(); + + // This changes the user requested device but not necessarily the device that + // on which the operation will run. + public native void set_requested_device(@StdString BytePointer device); + public native void set_requested_device(@StdString String device); + + // This gives the device the runtime has assigned this node to. If + // you want the device the user requested, use def().device() instead. + // TODO(josh11b): Validate that the assigned_device, if not empty: + // fully specifies a device, and satisfies def().device(). + // TODO(josh11b): Move assigned_device_name outside of Node into a + // NodeId->DeviceName map. + public native @StdString BytePointer assigned_device_name(); + public native void set_assigned_device_name(@StdString BytePointer device_name); + public native void set_assigned_device_name(@StdString String device_name); + public native @Cast("bool") boolean has_assigned_device_name(); + public native int assigned_device_name_index(); + public native void set_assigned_device_name_index(int index); + + // Sets 'original_node_names' field of this node's DebugInfo proto to + // 'names'. + + + // Read only access to attributes + + // Inputs requested by the NodeDef. For the actual inputs, use in_edges. + + // Get the neighboring nodes via edges either in or out of this node. This + // includes control edges. + + // Node type helpers. + public native @Cast("bool") boolean IsSource(); + public native @Cast("bool") boolean IsSink(); + // Anything other than the special Source & Sink nodes. + public native @Cast("bool") boolean IsOp(); + + // Node class helpers + public native @Cast("bool") boolean IsSwitch(); + public native @Cast("bool") boolean IsMerge(); + public native @Cast("bool") boolean IsEnter(); + public native @Cast("bool") boolean IsExit(); + public native @Cast("bool") boolean IsNextIteration(); + public native @Cast("bool") boolean IsLoopCond(); + public native @Cast("bool") boolean IsControlTrigger(); + public native @Cast("bool") boolean IsSend(); + public native @Cast("bool") boolean IsRecv(); + public native @Cast("bool") boolean IsConstant(); + public native @Cast("bool") boolean IsVariable(); + public native @Cast("bool") boolean IsIdentity(); + public native @Cast("bool") boolean IsGetSessionHandle(); + public native @Cast("bool") boolean IsGetSessionTensor(); + public native @Cast("bool") boolean IsDeleteSessionTensor(); + public native @Cast("bool") boolean IsControlFlow(); + public native @Cast("bool") boolean IsHostSend(); + public native @Cast("bool") boolean IsHostRecv(); + public native @Cast("bool") boolean IsScopedAllocator(); + public native @Cast("bool") boolean IsCollective(); + + public native @Cast("bool") boolean IsMetadata(); + public native @Cast("bool") boolean IsFakeParam(); + public native @Cast("bool") boolean IsPartitionedCall(); + + // Returns true if this node is any kind of function call node. + // + // NOTE: "function call nodes" include partitioned call ops, symbolic gradient + // ops, and ops whose type_string is the name of a function ("function ops"). + public native @Cast("bool") boolean IsFunctionCall(); + + public native @Cast("bool") boolean IsIfNode(); + public native @Cast("bool") boolean IsWhileNode(); + public native @Cast("bool") boolean IsCaseNode(); + // Is this node a function input + public native @Cast("bool") boolean IsArg(); + // Is this node a function output + public native @Cast("bool") boolean IsRetval(); + + public native @Cast("bool") boolean IsDistributedCommunication(); + + + + + + + + // Returns into '*e' the edge connecting to the 'idx' input of this Node. + + // Returns into '*edges' the input data edges of this Node, indexed by input + // number. Does not return control edges. + + // Returns into '*n' the node that has an output connected to the + // 'idx' input of this Node. + + + + // Returns into '*t' the idx-th input tensor of this node, represented as the + // output tensor of input_node(idx). + + // Sets the stack trace for the node. Assumes that getting and setting the + // stack trace for a given node will not race. + + + // Get the stack trace for when the node was instantiated. + + + // Called after an attr has changed. Decides whether we need to update some + // property of the node (stored in props_). + public native void UpdateProperties(); +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NodeBuilder.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NodeBuilder.java new file mode 100644 index 00000000000..5922ff38b0a --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/NodeBuilder.java @@ -0,0 +1,17 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + +@Namespace("tensorflow") @Opaque @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class NodeBuilder extends Pointer { + /** Empty constructor. Calls {@code super((Pointer)null)}. */ + public NodeBuilder() { super((Pointer)null); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public NodeBuilder(Pointer p) { super(p); } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Graph.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Graph.java index f4521d36625..c4d88baf176 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Graph.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Graph.java @@ -8,19 +8,42 @@ import static org.tensorflow.internal.c_api.global.tensorflow.*; +// Parsed from tensorflow/c/c_api_internal.h -// TODO(jeff,sanjay): -// - export functions to set Config fields - -// -------------------------------------------------------------------------- -// The new graph construction API, still under development. - -// Represents a computation graph. Graphs may be shared between sessions. -// Graphs are thread-safe when used as directed below. -@Opaque @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +@NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) public class TF_Graph extends org.tensorflow.internal.c_api.AbstractTF_Graph { - /** Empty constructor. Calls {@code super((Pointer)null)}. */ - public TF_Graph() { super((Pointer)null); } + static { Loader.load(); } /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ public TF_Graph(Pointer p) { super(p); } + + + + public native @MemberGetter @ByRef NativeGraphPointer graph(); + + // Runs shape inference. + + + // Maps from name of an operation to the Node* in 'graph'. + public native @ByRef NameMap name_map(); public native TF_Graph name_map(NameMap setter); + + // The keys of this map are all the active sessions using this graph. Each + // value records whether the graph has been mutated since the corresponding + // session has been run (this is detected in RecordMutation function). If the + // string is empty, no mutation has occurred. Otherwise the string is a + // description of the mutation suitable for returning to the user. + // + // Sessions are added to this map in TF_NewSession, and removed in + // TF_DeleteSession. + // TF_Graph may only / must be deleted when + // sessions.size() == 0 && delete_requested == true + // + // TODO(b/74949947): mutations currently trigger a warning instead of a bad + // status, this should be reverted when possible. + + // set true by TF_DeleteGraph + + // Used to link graphs contained in TF_WhileParams to the parent graph that + // will eventually contain the full while loop. + public native TF_Graph parent(); public native TF_Graph parent(TF_Graph setter); + public native TF_Output parent_inputs(); public native TF_Graph parent_inputs(TF_Output setter); } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Operation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Operation.java index 4daad4f8a2a..96e5ef47b38 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Operation.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Operation.java @@ -8,14 +8,11 @@ import static org.tensorflow.internal.c_api.global.tensorflow.*; - -// Operation that has been added to the graph. Valid until the graph is -// deleted -- in particular adding a new operation to the graph does not -// invalidate old TF_Operation* pointers. -@Opaque @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +@Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) public class TF_Operation extends Pointer { - /** Empty constructor. Calls {@code super((Pointer)null)}. */ - public TF_Operation() { super((Pointer)null); } + static { Loader.load(); } /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ public TF_Operation(Pointer p) { super(p); } + + public native @MemberGetter @ByRef Node node(); } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_OperationDescription.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_OperationDescription.java index 490ca238753..71738f4ac02 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_OperationDescription.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_OperationDescription.java @@ -8,12 +8,14 @@ import static org.tensorflow.internal.c_api.global.tensorflow.*; - -// Operation being built. The underlying graph must outlive this. -@Opaque @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +@NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) public class TF_OperationDescription extends Pointer { - /** Empty constructor. Calls {@code super((Pointer)null)}. */ - public TF_OperationDescription() { super((Pointer)null); } + static { Loader.load(); } /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ public TF_OperationDescription(Pointer p) { super(p); } + + + public native @ByRef NodeBuilder node_builder(); public native TF_OperationDescription node_builder(NodeBuilder setter); + public native TF_Graph graph(); public native TF_OperationDescription graph(TF_Graph setter); + } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Scope.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Scope.java new file mode 100644 index 00000000000..bea7ec98a25 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/TF_Scope.java @@ -0,0 +1,216 @@ +// Targeted by JavaCPP version 1.5.6: DO NOT EDIT THIS FILE + +package org.tensorflow.internal.c_api; + +import java.nio.*; +import org.bytedeco.javacpp.*; +import org.bytedeco.javacpp.annotation.*; + +import static org.tensorflow.internal.c_api.global.tensorflow.*; + + +/** \addtogroup core + * \{ +

+ * A {@code Scope} object represents a set of related TensorFlow ops that have the + * same properties such as a common name prefix. + * + * A Scope object is a container for TensorFlow Op properties. Op constructors + * get a Scope object as a mandatory first argument and the constructed op + * acquires the properties in the object. + * + * A simple example: + * + * using namespace ops; + * Scope root = Scope::NewRootScope(); + * auto c1 = Const(root, { {1, 1} }); + * auto m = MatMul(root, c1, { {41}, {1} }); + * GraphDef gdef; + * Status s = root.ToGraphDef(&gdef); + * if (!s.ok()) { ... } + * + * Scope hierarchy: + * + * The Scope class provides various With<> functions that create a new scope. + * The new scope typically has one property changed while other properties are + * inherited from the parent scope. + * NewSubScope(name) method appends {@code name} to the prefix of names for ops + * created within the scope, and WithOpName() changes the suffix which + * otherwise defaults to the type of the op. + * + * Name examples: + * + * Scope root = Scope::NewRootScope(); + * Scope linear = root.NewSubScope("linear"); + * // W will be named "linear/W" + * auto W = Variable(linear.WithOpName("W"), + * {2, 2}, DT_FLOAT); + * // b will be named "linear/b_3" + * int idx = 3; + * auto b = Variable(linear.WithOpName("b_", idx), + * {2}, DT_FLOAT); + * auto x = Const(linear, {...}); // name: "linear/Const" + * auto m = MatMul(linear, x, W); // name: "linear/MatMul" + * auto r = BiasAdd(linear, m, b); // name: "linear/BiasAdd" + * + * Scope lifetime: + * + * A new scope is created by calling Scope::NewRootScope. This creates some + * resources that are shared by all the child scopes that inherit from this + * scope, directly or transitively. For instance, a new scope creates a new + * Graph object to which operations are added when the new scope or its + * children are used by an Op constructor. The new scope also has a Status + * object which will be used to indicate errors by Op-constructor functions + * called on any child scope. The Op-constructor functions have to check the + * scope's status by calling the ok() method before proceeding to construct the + * op. + * + * Thread safety: + * + * A {@code Scope} object is NOT thread-safe. Threads cannot concurrently call + * op-constructor functions on the same {@code Scope} object. */ +@Name("tensorflow::Scope") @NoOffset @Properties(inherit = org.tensorflow.internal.c_api.presets.tensorflow.class) +public class TF_Scope extends Pointer { + static { Loader.load(); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public TF_Scope(Pointer p) { super(p); } + + public TF_Scope(@Const @ByRef TF_Scope other) { super((Pointer)null); allocate(other); } + private native void allocate(@Const @ByRef TF_Scope other); + public native @ByRef @Name("operator =") TF_Scope put(@Const @ByRef TF_Scope other); + + // The following functions are for users making graphs. They return brand new + // scopes, or scopes derived from an existing scope object. + + /** Return a new scope. + * This creates a new graph and all operations constructed in this graph + * should use the returned object as the "root" scope. */ + public static native @ByVal TF_Scope NewRootScope(); + + /** Return a new scope. Ops created with this scope will have + * {@code name/child_scope_name} as the prefix. The actual name will be unique + * in the current scope. All other properties are inherited from the current + * scope. If {@code child_scope_name} is empty, the {@code /} is elided. */ + public native @ByVal TF_Scope NewSubScope(@StdString BytePointer child_scope_name); + public native @ByVal TF_Scope NewSubScope(@StdString String child_scope_name); + + /** Return a new scope. All ops created within the returned scope will have + * names of the form {@code name/StrCat(fragments...)[_suffix]} */ + + /** Return a new scope. All ops created within the returned scope will have as + * control dependencies the union of operations in the control_deps vector + * and the control dependencies of the current scope. */ + public native @ByVal TF_Scope WithControlDependencies( + @Span NativeOperation control_deps); + /** Same as above, but convenient to add control dependency on the operation + * producing the control_dep output. */ + public native @ByVal TF_Scope WithControlDependencies(@Const @ByRef NativeOutput control_dep); + + /** Return a new scope. All ops created within the returned scope will have no + * control dependencies on other operations. */ + public native @ByVal TF_Scope WithNoControlDependencies(); + + /** Return a new scope. All ops created within the returned scope will have + * the device field set to 'device'. */ + public native @ByVal TF_Scope WithDevice(@StdString BytePointer device); + public native @ByVal TF_Scope WithDevice(@StdString String device); + + /** Returns a new scope. All ops created within the returned scope will have + * their assigned device set to {@code assigned_device}. */ + + + /** Returns a new scope. All ops created within the returned scope will have + * their _XlaCluster attribute set to {@code xla_cluster}. */ + + + /** Return a new scope. All ops created within the returned scope will be + * co-located on the device where op is placed. + * NOTE: This function is intended to be use internal libraries only for + * controlling placement of ops on to devices. Public use is not encouraged + * because the implementation of device placement is subject to change. */ + + /** Convenience function for above. */ + + /** Clear all colocation constraints. */ + + + /** Return a new scope. The op-constructor functions taking the returned scope + * as the scope argument will exit as soon as an error is detected, instead + * of setting the status on the scope. */ + public native @ByVal TF_Scope ExitOnError(); + + /** Return a new scope. All ops created with the new scope will have + * kernel_label as the value for their '_kernel' attribute; */ + + + // The following functions are for scope object consumers. + + /** Return a unique name, using default_name if an op name has not been + * specified. */ + public native @StdString BytePointer GetUniqueNameForOp(@StdString BytePointer default_name); + public native @StdString String GetUniqueNameForOp(@StdString String default_name); + + /** Update the status on this scope. + * Note: The status object is shared between all children of this scope. + * If the resulting status is not Status::OK() and exit_on_error_ is set on + * this scope, this function exits by calling LOG(FATAL). */ + + + // START_SKIP_DOXYGEN + + /** Update the builder with properties accumulated in this scope. Does not set + * status(). */ + // TODO(skyewm): NodeBuilder is not part of public API + public native void UpdateBuilder(NodeBuilder builder); + // END_SKIP_DOXYGEN + + public native @Cast("bool") boolean ok(); + + // TODO(skyewm): Graph is not part of public API + public native NativeGraphPointer graph(); + + // TODO(skyewm): Graph is not part of public API + + + + + /** If status() is Status::OK(), convert the Graph object stored in this scope + * to a GraphDef proto and return Status::OK(). Otherwise, return the error + * status as is without performing GraphDef conversion. */ + + + // START_SKIP_DOXYGEN + + /** If status() is Status::OK(), construct a Graph object using {@code opts} as the + * GraphConstructorOptions, and return Status::OK if graph construction was + * successful. Otherwise, return the error status. */ + // TODO(josh11b, keveman): Make this faster; right now it converts + // Graph->GraphDef->Graph. This cleans up the graph (e.g. adds + // edges from the source and to the sink node, resolves back edges + // by name), and makes sure the resulting graph is valid. + + + // Calls AddNode() using this scope's ShapeRefiner. This exists in the public + // API to prevent custom op wrappers from needing access to shape_refiner.h or + // scope_internal.h. + // TODO(skyewm): remove this from public API + + + // Creates a new root scope that causes all DoShapeInference() calls to return + // Status::OK() (on the returned scope and any subscopes). Used for testing. + // TODO(skyewm): fix tests that still require this and eventually remove, or + // at least remove from public API + + // END_SKIP_DOXYGEN + + + + // START_SKIP_DOXYGEN + @Opaque public static class Impl extends Pointer { + /** Empty constructor. Calls {@code super((Pointer)null)}. */ + public Impl() { super((Pointer)null); } + /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ + public Impl(Pointer p) { super(p); } + } + public native Impl impl(); +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/global/tensorflow.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/global/tensorflow.java index 56ed191ae28..3128389e255 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/global/tensorflow.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/internal/c_api/global/tensorflow.java @@ -11,6 +11,12 @@ public class tensorflow extends org.tensorflow.internal.c_api.presets.tensorflow { static { Loader.load(); } +// Targeting ../NativeOutputVector.java + + +// Targeting ../NameMap.java + + // Parsed from tensorflow/core/platform/ctstring_internal.h /* Copyright 2019 The TensorFlow Authors. All Rights Reserved. @@ -982,9 +988,15 @@ public static native void TF_SetConfig(TF_SessionOptions options, // Destroy an options object. public static native void TF_DeleteSessionOptions(TF_SessionOptions arg0); -// Targeting ../TF_Graph.java +// TODO(jeff,sanjay): +// - export functions to set Config fields + +// -------------------------------------------------------------------------- +// The new graph construction API, still under development. +// Represents a computation graph. Graphs may be shared between sessions. +// Graphs are thread-safe when used as directed below. // Return a new graph object. public static native TF_Graph TF_NewGraph(); @@ -992,12 +1004,12 @@ public static native void TF_SetConfig(TF_SessionOptions options, // Destroy an options object. Graph will be deleted once no more // TFSession's are referencing it. public static native void TF_DeleteGraph(TF_Graph arg0); -// Targeting ../TF_OperationDescription.java - - -// Targeting ../TF_Operation.java +// Operation being built. The underlying graph must outlive this. +// Operation that has been added to the graph. Valid until the graph is +// deleted -- in particular adding a new operation to the graph does not +// invalidate old TF_Operation* pointers. // Targeting ../TF_Input.java @@ -1079,6 +1091,15 @@ public static native void TF_GraphGetTensorShape(TF_Graph graph, @Cast("int64_t*") long[] dims, int num_dims, TF_Status status); +// TF_NewOperation, but without locking the graph. +// Should prefer TF_NewOperation when possible. +public static native TF_OperationDescription TF_NewOperationLocked(TF_Graph graph, + @Cast("const char*") BytePointer op_type, + @Cast("const char*") BytePointer oper_name); +public static native TF_OperationDescription TF_NewOperationLocked(TF_Graph graph, + String op_type, + String oper_name); + // Operation will only be added to *graph when TF_FinishOperation() is // called (assuming TF_FinishOperation() does not return an error). // *graph must not be deleted until after TF_FinishOperation() is @@ -1418,6 +1439,11 @@ public static native void TF_SetAttrValueProto(TF_OperationDescription desc, @Cast("size_t") long proto_len, TF_Status status); +// TF_FinishOperation, but without locking the graph. +// TF_FinishOperation should be preferred when possible. +public static native TF_Operation TF_FinishOperationLocked(TF_OperationDescription desc, + TF_Status status); + // If this function succeeds: // * *status is set to an OK value, // * a TF_Operation is added to the graph, @@ -4084,6 +4110,26 @@ public static native void TF_ShapeInferenceContextConcatenateShapes( // #endif // TENSORFLOW_C_OPS_H_ +// Parsed from tensorflow_adapters.h + +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + http://www.apache.org/licenses/LICENSE-2.0 + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. + ======================================================================= + */ + +// #include "absl/types/span.h" + + + // Parsed from tensorflow/c/eager/c_api.h /* Copyright 2017 The TensorFlow Authors. All Rights Reserved. @@ -4783,4 +4829,306 @@ public static native void TFE_OpSetAttrValueProto(@Const TFE_Op op, public static final int TFE_CUSTOM_DEVICE_VERSION = 4; +// Parsed from tensorflow/cc/framework/scope.h + +/* Copyright 2016 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +// #ifndef TENSORFLOW_CC_FRAMEWORK_SCOPE_H_ +// #define TENSORFLOW_CC_FRAMEWORK_SCOPE_H_ + +// #include +// #include +// #include +// #include +// #include + +// #include "absl/strings/str_cat.h" +// #include "tensorflow/cc/framework/ops.h" +// #include "tensorflow/core/common_runtime/graph_constructor.h" +// #include "tensorflow/core/lib/core/status.h" +// #include "tensorflow/core/lib/gtl/array_slice.h" +// Targeting ../NativeGraphPointer.java + + +// Targeting ../NodeBuilder.java + + +// Targeting ../TF_Scope.java + + + +/** A helper struct to hold the scopes that would be used by a function + * constructing a composite op. */ + +// Creates a node of the given operation, with the given inputs, and assigns the +// result to output. This does not support the ability to add additional +// attributes. + +/** \} */ + + // namespace tensorflow + +// #endif // TENSORFLOW_CC_FRAMEWORK_SCOPE_H_ + + +// Parsed from tensorflow/cc/framework/grad_op_registry.h + +/* Copyright 2016 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +// #ifndef TENSORFLOW_CC_FRAMEWORK_GRAD_OP_REGISTRY_H_ +// #define TENSORFLOW_CC_FRAMEWORK_GRAD_OP_REGISTRY_H_ + +// #include + +// #include "tensorflow/cc/framework/ops.h" +// #include "tensorflow/cc/framework/scope.h" +// Targeting ../GradFunc.java + + +// Targeting ../GradOpRegistry.java + + + + // namespace ops + +// Macros used to define gradient functions for ops. +// #define REGISTER_GRADIENT_OP(name, fn) +// REGISTER_GRADIENT_OP_UNIQ_HELPER(__COUNTER__, name, fn) + +// #define REGISTER_NO_GRADIENT_OP(name) +// REGISTER_GRADIENT_OP_UNIQ_HELPER(__COUNTER__, name, nullptr) + +// #define REGISTER_GRADIENT_OP_UNIQ_HELPER(ctr, name, fn) +// REGISTER_GRADIENT_OP_UNIQ(ctr, name, fn) + +// #define REGISTER_GRADIENT_OP_UNIQ(ctr, name, fn) +// static bool unused_ret_val_##ctr = +// ::tensorflow::ops::GradOpRegistry::Global()->Register(name, fn) + + // namespace tensorflow + +// #endif // TENSORFLOW_CC_FRAMEWORK_GRAD_OP_REGISTRY_H_ + + +// Parsed from tensorflow/core/platform/status.h + +/* Copyright 2015 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +// #ifndef TENSORFLOW_CORE_PLATFORM_STATUS_H_ +// #define TENSORFLOW_CORE_PLATFORM_STATUS_H_ + +// #include +// #include +// #include +// #include +// #include + +// #include "tensorflow/core/platform/logging.h" +// #include "tensorflow/core/platform/macros.h" +// #include "tensorflow/core/platform/stack_frame.h" +// #include "tensorflow/core/platform/stringpiece.h" +// #include "tensorflow/core/platform/types.h" +// #include "tensorflow/core/protobuf/error_codes.pb.h" + +// #if defined(__clang__) +// Only clang supports warn_unused_result as a type annotation. +// Targeting ../NativeStatus.java + + + +// Helper class to manage multiple child status values. + + + + + +// #ifndef SWIG + + + +// #endif // SWIG + + + + + +/** \ingroup core */ +@Namespace("tensorflow") public static native @Cast("std::ostream*") @ByRef @Name("operator <<") Pointer shiftLeft(@Cast("std::ostream*") @ByRef Pointer os, @Const @ByRef NativeStatus x); + +@Namespace("tensorflow") public static native @StdString BytePointer TfCheckOpHelperOutOfLine( + @Const @ByRef NativeStatus v, @Cast("const char*") BytePointer msg); +@Namespace("tensorflow") public static native @StdString BytePointer TfCheckOpHelperOutOfLine( + @Const @ByRef NativeStatus v, String msg); + +@Namespace("tensorflow") public static native @StdString BytePointer TfCheckOpHelper(@ByVal NativeStatus v, + @Cast("const char*") BytePointer msg); +@Namespace("tensorflow") public static native @StdString BytePointer TfCheckOpHelper(@ByVal NativeStatus v, + String msg); + +// #define TF_DO_CHECK_OK(val, level) +// while (auto _result = ::tensorflow::TfCheckOpHelper(val, #val)) +// LOG(level) << *(_result) + +// #define TF_CHECK_OK(val) TF_DO_CHECK_OK(val, FATAL) +// #define TF_QCHECK_OK(val) TF_DO_CHECK_OK(val, QFATAL) + +// DEBUG only version of TF_CHECK_OK. Compiler still parses 'val' even in opt +// mode. +// #ifndef NDEBUG +// #define TF_DCHECK_OK(val) TF_CHECK_OK(val) +// #else +// #define TF_DCHECK_OK(val) +// while (false && (::tensorflow::Status::OK() == (val))) LOG(FATAL) +// #endif + + // namespace tensorflow + +// #endif // TENSORFLOW_CORE_PLATFORM_STATUS_H_ + + +// Targeting ../Node.java + + + +// Stores debug information associated with the Node. + + +// Parsed from tensorflow/c/tf_status_helper.h + +/* Copyright 2016 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +// #ifndef TENSORFLOW_C_TF_STATUS_HELPER_H_ +// #define TENSORFLOW_C_TF_STATUS_HELPER_H_ + +// #include "tensorflow/c/tf_status.h" +// #include "tensorflow/core/platform/status.h" + +// Set the attribute of "tf_status" from the attributes of "status". +@Namespace("tensorflow") public static native void Set_TF_Status_from_Status(TF_Status tf_status, + @Const @ByRef NativeStatus status); + +// Returns a "status" from "tf_status". +@Namespace("tensorflow") public static native @ByVal NativeStatus StatusFromTF_Status(@Const TF_Status tf_status); + // namespace internal + + // namespace tensorflow + +// #endif // TENSORFLOW_C_TF_STATUS_HELPER_H_ + + +// Parsed from tensorflow/cc/framework/ops.h + +/* Copyright 2016 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +// #ifndef TENSORFLOW_CC_FRAMEWORK_OPS_H_ +// #define TENSORFLOW_CC_FRAMEWORK_OPS_H_ + +// #include + +// #include "tensorflow/core/framework/tensor.h" +// #include "tensorflow/core/framework/tensor.pb.h" +// #include "tensorflow/core/graph/graph.h" +// #include "tensorflow/core/lib/hash/hash.h" +// #include "tensorflow/core/lib/strings/strcat.h" + +/** \defgroup core Core Tensorflow API */ +// Targeting ../NativeOperation.java + + +// Targeting ../NativeOutput.java + + + +/** Hash class that can be used for e.g. storing Outputs in an unordered_map */ + +/** Represents a tensor value that can be used as an operand to an Operation. */ + +/** A type for representing the output of ops that produce more than one output, + * or a list of tensors. */ + +/** A type for representing the input to ops that require a list of tensors. */ + +/** \} */ + + // namespace tensorflow + +// #endif // TENSORFLOW_CC_FRAMEWORK_OPS_H_ + + +// Targeting ../TF_Graph.java + + +// Targeting ../TF_OperationDescription.java + + +// Targeting ../TF_Operation.java + + + + } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/AudioSpectrogram.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/AudioSpectrogram.java index 12fca67165e..4bc3f61c197 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/AudioSpectrogram.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/AudioSpectrogram.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; @@ -54,6 +56,10 @@ * tensorflow/examples/wav_to_spectrogram to read in an audio file and save out the * resulting spectrogram as a PNG image. */ +@OpMetadata( + opType = AudioSpectrogram.OP_NAME, + inputsClass = AudioSpectrogram.Inputs.class +) @Operator( group = "audio" ) @@ -65,8 +71,8 @@ public final class AudioSpectrogram extends RawOp implements Operand { private Output spectrogram; - private AudioSpectrogram(Operation operation) { - super(operation); + public AudioSpectrogram(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; spectrogram = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options magnitudeSquared(Boolean magnitudeSquared) { } } + @OpInputsMetadata( + outputsClass = AudioSpectrogram.class + ) public static class Inputs extends RawOpInputs { /** * Float representation of audio data. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/DecodeWav.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/DecodeWav.java index 0caa473a0db..1666d475d57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/DecodeWav.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/DecodeWav.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * number of samples. For example, a ten-sample-long stereo WAV file should give an * output shape of [10, 2]. */ +@OpMetadata( + opType = DecodeWav.OP_NAME, + inputsClass = DecodeWav.Inputs.class +) @Operator( group = "audio" ) @@ -59,8 +65,8 @@ public final class DecodeWav extends RawOp { private Output sampleRate; - private DecodeWav(Operation operation) { - super(operation); + public DecodeWav(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; audio = operation.output(outputIdx++); sampleRate = operation.output(outputIdx++); @@ -165,6 +171,9 @@ public Options desiredSamples(Long desiredSamples) { } } + @OpInputsMetadata( + outputsClass = DecodeWav.class + ) public static class Inputs extends RawOpInputs { /** * The WAV-encoded audio, usually from a file. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/EncodeWav.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/EncodeWav.java index 8e535d1d48b..ecbafeb2d4e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/EncodeWav.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/EncodeWav.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ *

{@code audio} is a 2-D float Tensor of shape {@code [length, channels]}. * {@code sample_rate} is a scalar Tensor holding the rate to use (e.g. 44100). */ +@OpMetadata( + opType = EncodeWav.OP_NAME, + inputsClass = EncodeWav.Inputs.class +) @Operator( group = "audio" ) @@ -52,8 +58,8 @@ public final class EncodeWav extends RawOp implements Operand { private Output contents; - private EncodeWav(Operation operation) { - super(operation); + public EncodeWav(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; contents = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return contents; } + @OpInputsMetadata( + outputsClass = EncodeWav.class + ) public static class Inputs extends RawOpInputs { /** * 2-D with shape {@code [length, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/Mfcc.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/Mfcc.java index b75d638879c..cf1f13ea4ac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/Mfcc.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/audio/Mfcc.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * history in the speech recognition world, and https://en.wikipedia.org/wiki/Mel-frequency_cepstrum * is a good resource to learn more. */ +@OpMetadata( + opType = Mfcc.OP_NAME, + inputsClass = Mfcc.Inputs.class +) @Operator( group = "audio" ) @@ -51,8 +57,8 @@ public final class Mfcc extends RawOp implements Operand { private Output output; - private Mfcc(Operation operation) { - super(operation); + public Mfcc(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -212,6 +218,9 @@ public Options dctCoefficientCount(Long dctCoefficientCount) { } } + @OpInputsMetadata( + outputsClass = Mfcc.class + ) public static class Inputs extends RawOpInputs { /** * Typically produced by the Spectrogram op, with magnitude_squared diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseAnd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseAnd.java index 0a208566534..75e1d6e8a5f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseAnd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseAnd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = BitwiseAnd.OP_NAME, + inputsClass = BitwiseAnd.Inputs.class +) @Operator( group = "bitwise" ) @@ -64,8 +70,8 @@ public final class BitwiseAnd extends RawOp implements Operan private Output z; - private BitwiseAnd(Operation operation) { - super(operation); + public BitwiseAnd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -103,6 +109,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = BitwiseAnd.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseOr.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseOr.java index ba212ec5596..28865bc0cc8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseOr.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseOr.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = BitwiseOr.OP_NAME, + inputsClass = BitwiseOr.Inputs.class +) @Operator( group = "bitwise" ) @@ -64,8 +70,8 @@ public final class BitwiseOr extends RawOp implements Operand private Output z; - private BitwiseOr(Operation operation) { - super(operation); + public BitwiseOr(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -103,6 +109,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = BitwiseOr.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseXor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseXor.java index 57248c5cc6d..e4655d95f8f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseXor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/BitwiseXor.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = BitwiseXor.OP_NAME, + inputsClass = BitwiseXor.Inputs.class +) @Operator( group = "bitwise" ) @@ -64,8 +70,8 @@ public final class BitwiseXor extends RawOp implements Operan private Output z; - private BitwiseXor(Operation operation) { - super(operation); + public BitwiseXor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -103,6 +109,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = BitwiseXor.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/Invert.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/Invert.java index f5f2765eb33..4dd4375304c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/Invert.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/Invert.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -74,6 +76,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Invert.OP_NAME, + inputsClass = Invert.Inputs.class +) @Operator( group = "bitwise" ) @@ -85,8 +91,8 @@ public final class Invert extends RawOp implements Operand private Output y; - private Invert(Operation operation) { - super(operation); + public Invert(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -122,6 +128,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Invert.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/LeftShift.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/LeftShift.java index 6420dcc7f65..0173cdc694d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/LeftShift.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/LeftShift.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -64,6 +66,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = LeftShift.OP_NAME, + inputsClass = LeftShift.Inputs.class +) @Operator( group = "bitwise" ) @@ -75,8 +81,8 @@ public final class LeftShift extends RawOp implements Operand private Output z; - private LeftShift(Operation operation) { - super(operation); + public LeftShift(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -114,6 +120,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = LeftShift.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/RightShift.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/RightShift.java index 5fa84781682..00990db0ce0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/RightShift.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/bitwise/RightShift.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -66,6 +68,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RightShift.OP_NAME, + inputsClass = RightShift.Inputs.class +) @Operator( group = "bitwise" ) @@ -77,8 +83,8 @@ public final class RightShift extends RawOp implements Operan private Output z; - private RightShift(Operation operation) { - super(operation); + public RightShift(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -116,6 +122,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RightShift.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java index 39eb7f70ce5..7c2432bcc34 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -37,6 +39,10 @@ * of the k-MC^2 algorithm and returns the index of one candidate point to be added * as an additional cluster center. */ +@OpMetadata( + opType = KMC2ChainInitialization.OP_NAME, + inputsClass = KMC2ChainInitialization.Inputs.class +) public final class KMC2ChainInitialization extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class KMC2ChainInitialization extends RawOp implements Operand index; - private KMC2ChainInitialization(Operation operation) { - super(operation); + public KMC2ChainInitialization(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; index = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return index; } + @OpInputsMetadata( + outputsClass = KMC2ChainInitialization.class + ) public static class Inputs extends RawOpInputs { /** * Vector with squared distances to the closest previously sampled cluster center diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java index 65058cc1268..e2d7fc5f79a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -37,6 +39,10 @@ * distance from the nearest row selected thus far till num_to_sample rows have * been sampled. */ +@OpMetadata( + opType = KmeansPlusPlusInitialization.OP_NAME, + inputsClass = KmeansPlusPlusInitialization.Inputs.class +) public final class KmeansPlusPlusInitialization extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class KmeansPlusPlusInitialization extends RawOp implements Operand private Output samples; - private KmeansPlusPlusInitialization(Operation operation) { - super(operation); + public KmeansPlusPlusInitialization(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; samples = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return samples; } + @OpInputsMetadata( + outputsClass = KmeansPlusPlusInitialization.class + ) public static class Inputs extends RawOpInputs { /** * Matrix of shape (n, d). Rows are assumed to be input points. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/AllReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/AllReduce.java index e338973185a..53e72874745 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/AllReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/AllReduce.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @deprecated use {@link org.tensorflow.op.collective.Reduce} instead */ +@OpMetadata( + opType = AllReduce.OP_NAME, + inputsClass = AllReduce.Inputs.class +) @Deprecated public final class AllReduce extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class AllReduce extends RawOp implements Operand private Output data; - private AllReduce(Operation operation) { - super(operation); + public AllReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -218,6 +224,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = AllReduce.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastRecv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastRecv.java index 76cdc37a0c4..94dbc145772 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastRecv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastRecv.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = BroadcastRecv.OP_NAME, + inputsClass = BroadcastRecv.Inputs.class +) public final class BroadcastRecv extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class BroadcastRecv extends RawOp implements Opera private Output data; - private BroadcastRecv(Operation operation) { - super(operation); + public BroadcastRecv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -156,6 +162,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = BroadcastRecv.class + ) public static class Inputs extends RawOpInputs> { /** * The T attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastSend.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastSend.java index 707032af259..cc18c528581 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastSend.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/BroadcastSend.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = BroadcastSend.OP_NAME, + inputsClass = BroadcastSend.Inputs.class +) public final class BroadcastSend extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class BroadcastSend extends RawOp implements Opera private Output data; - private BroadcastSend(Operation operation) { - super(operation); + public BroadcastSend(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -155,6 +161,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = BroadcastSend.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Gather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Gather.java index b058a53afea..9140ca240ca 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Gather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Gather.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = Gather.OP_NAME, + inputsClass = Gather.Inputs.class +) public final class Gather extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class Gather extends RawOp implements Operand private Output data; - private Gather(Operation operation) { - super(operation); + public Gather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -155,6 +161,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = Gather.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/GatherV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/GatherV2.java index 77e32799164..2ee7eda0773 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/GatherV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/GatherV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = GatherV2.OP_NAME, + inputsClass = GatherV2.Inputs.class +) public final class GatherV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class GatherV2 extends RawOp implements Operand< private Output data; - private GatherV2(Operation operation) { - super(operation); + public GatherV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -184,6 +190,9 @@ public Options NorderingToken(Long NorderingToken) { } } + @OpInputsMetadata( + outputsClass = GatherV2.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Reduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Reduce.java index 3b86c960d1a..7d503474d02 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Reduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/Reduce.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = Reduce.OP_NAME, + inputsClass = Reduce.Inputs.class +) public final class Reduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class Reduce extends RawOp implements Operand private Output data; - private Reduce(Operation operation) { - super(operation); + public Reduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -215,6 +221,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = Reduce.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/ReduceV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/ReduceV2.java index da55ef12535..0c09c73f020 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/ReduceV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/ReduceV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = ReduceV2.OP_NAME, + inputsClass = ReduceV2.Inputs.class +) public final class ReduceV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class ReduceV2 extends RawOp implements Operand< private Output data; - private ReduceV2(Operation operation) { - super(operation); + public ReduceV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -215,6 +221,9 @@ public Options maxSubdivsPerDevice(Long maxSubdivsPerDevice) { } } + @OpInputsMetadata( + outputsClass = ReduceV2.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Abort.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Abort.java index 455cb214124..c51158b43dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Abort.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Abort.java @@ -25,6 +25,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; /** @@ -33,6 +35,10 @@ * otherwise it will exit with a SIGABORT signal. *

Returns nothing but an exception. */ +@OpMetadata( + opType = Abort.OP_NAME, + inputsClass = Abort.Inputs.class +) @Operator public final class Abort extends RawOp { /** @@ -40,8 +46,8 @@ public final class Abort extends RawOp { */ public static final String OP_NAME = "Abort"; - private Abort(Operation operation) { - super(operation); + public Abort(Operation operation) { + super(operation, OP_NAME); } /** @@ -123,6 +129,9 @@ public Options exitWithoutError(Boolean exitWithoutError) { } } + @OpInputsMetadata( + outputsClass = Abort.class + ) public static class Inputs extends RawOpInputs { /** * A string which is the message associated with the exception. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/All.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/All.java index ca17e9a1ba0..81d49962fc1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/All.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/All.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ * {@code axis}. If {@code keep_dims} is true, the reduced dimensions are * retained with length 1. */ +@OpMetadata( + opType = All.OP_NAME, + inputsClass = All.Inputs.class +) @Operator public final class All extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class All extends RawOp implements Operand { private Output output; - private All(Operation operation) { - super(operation); + public All(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = All.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Any.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Any.java index c853cf365ea..8ab3c606ad4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Any.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Any.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ * {@code axis}. If {@code keep_dims} is true, the reduced dimensions are * retained with length 1. */ +@OpMetadata( + opType = Any.OP_NAME, + inputsClass = Any.Inputs.class +) @Operator public final class Any extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class Any extends RawOp implements Operand { private Output output; - private Any(Operation operation) { - super(operation); + public Any(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = Any.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssertThat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssertThat.java index fdd9f9628c7..af5c1dd9f9e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssertThat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssertThat.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -36,6 +38,10 @@ * If {@code condition} evaluates to false, print the list of tensors in {@code data}. * {@code summarize} determines how many entries of the tensors to print. */ +@OpMetadata( + opType = AssertThat.OP_NAME, + inputsClass = AssertThat.Inputs.class +) @Operator public final class AssertThat extends RawOp { /** @@ -43,8 +49,8 @@ public final class AssertThat extends RawOp { */ public static final String OP_NAME = "Assert"; - private AssertThat(Operation operation) { - super(operation); + public AssertThat(Operation operation) { + super(operation, OP_NAME); } /** @@ -105,6 +111,9 @@ public Options summarize(Long summarize) { } } + @OpInputsMetadata( + outputsClass = AssertThat.class + ) public static class Inputs extends RawOpInputs { /** * The condition to evaluate. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Assign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Assign.java index 9af134e388e..ba1e413af71 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Assign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Assign.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = Assign.OP_NAME, + inputsClass = Assign.Inputs.class +) @Operator public final class Assign extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class Assign extends RawOp implements Operand { private Output outputRef; - private Assign(Operation operation) { - super(operation); + public Assign(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -159,6 +165,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = Assign.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. May be uninitialized. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAdd.java index b863c0aed92..3a6df4bed6a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = AssignAdd.OP_NAME, + inputsClass = AssignAdd.Inputs.class +) @Operator public final class AssignAdd extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class AssignAdd extends RawOp implements Operand outputRef; - private AssignAdd(Operation operation) { - super(operation); + public AssignAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = AssignAdd.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAddVariableOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAddVariableOp.java index 618f646be84..e7dadaaec95 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAddVariableOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignAddVariableOp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Any ReadVariableOp with a control dependency on this op is guaranteed to * see the incremented value or a subsequent newer one. */ +@OpMetadata( + opType = AssignAddVariableOp.OP_NAME, + inputsClass = AssignAddVariableOp.Inputs.class +) @Operator public final class AssignAddVariableOp extends RawOp { /** @@ -42,8 +48,8 @@ public final class AssignAddVariableOp extends RawOp { */ public static final String OP_NAME = "AssignAddVariableOp"; - private AssignAddVariableOp(Operation operation) { - super(operation); + public AssignAddVariableOp(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static AssignAddVariableOp create(Scope scope, Operand r return new AssignAddVariableOp(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = AssignAddVariableOp.class + ) public static class Inputs extends RawOpInputs { /** * handle to the resource in which to store the variable. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSub.java index a1d6c5bfb3a..8ae31dde667 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = AssignSub.OP_NAME, + inputsClass = AssignSub.Inputs.class +) @Operator public final class AssignSub extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class AssignSub extends RawOp implements Operand outputRef; - private AssignSub(Operation operation) { - super(operation); + public AssignSub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = AssignSub.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSubVariableOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSubVariableOp.java index ddd83dc834e..6de4c159ee1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSubVariableOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignSubVariableOp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Any ReadVariableOp with a control dependency on this op is guaranteed to * see the decremented value or a subsequent newer one. */ +@OpMetadata( + opType = AssignSubVariableOp.OP_NAME, + inputsClass = AssignSubVariableOp.Inputs.class +) @Operator public final class AssignSubVariableOp extends RawOp { /** @@ -42,8 +48,8 @@ public final class AssignSubVariableOp extends RawOp { */ public static final String OP_NAME = "AssignSubVariableOp"; - private AssignSubVariableOp(Operation operation) { - super(operation); + public AssignSubVariableOp(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static AssignSubVariableOp create(Scope scope, Operand r return new AssignSubVariableOp(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = AssignSubVariableOp.class + ) public static class Inputs extends RawOpInputs { /** * handle to the resource in which to store the variable. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignVariableOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignVariableOp.java index 98dd0682d0d..8a158a0f014 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignVariableOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AssignVariableOp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Any ReadVariableOp with a control dependency on this op is guaranteed to return * this value or a subsequent newer value of the variable. */ +@OpMetadata( + opType = AssignVariableOp.OP_NAME, + inputsClass = AssignVariableOp.Inputs.class +) @Operator public final class AssignVariableOp extends RawOp { /** @@ -42,8 +48,8 @@ public final class AssignVariableOp extends RawOp { */ public static final String OP_NAME = "AssignVariableOp"; - private AssignVariableOp(Operation operation) { - super(operation); + public AssignVariableOp(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static AssignVariableOp create(Scope scope, Operand reso return new AssignVariableOp(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = AssignVariableOp.class + ) public static class Inputs extends RawOpInputs { /** * handle to the resource in which to store the variable. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Barrier.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Barrier.java index a722c116dd7..0bb0e71fc7a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Barrier.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Barrier.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -45,6 +47,10 @@ * incomplete element has some undefined components in its value tuple, * and may be updated using BarrierInsertMany. */ +@OpMetadata( + opType = Barrier.OP_NAME, + inputsClass = Barrier.Inputs.class +) @Operator public final class Barrier extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class Barrier extends RawOp implements Operand { private Output handle; - private Barrier(Operation operation) { - super(operation); + public Barrier(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -247,6 +253,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = Barrier.class + ) public static class Inputs extends RawOpInputs { /** * The type of each component in a value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierClose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierClose.java index 509f714fb88..ba5c6c37b2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierClose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierClose.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * continue to succeed if sufficient completed elements remain in the barrier. * Subsequent TakeMany operations that would block will fail immediately. */ +@OpMetadata( + opType = BarrierClose.OP_NAME, + inputsClass = BarrierClose.Inputs.class +) @Operator public final class BarrierClose extends RawOp { /** @@ -45,8 +51,8 @@ public final class BarrierClose extends RawOp { */ public static final String OP_NAME = "BarrierClose"; - private BarrierClose(Operation operation) { - super(operation); + public BarrierClose(Operation operation) { + super(operation, OP_NAME); } /** @@ -108,6 +114,9 @@ public Options cancelPendingEnqueues(Boolean cancelPendingEnqueues) { } } + @OpInputsMetadata( + outputsClass = BarrierClose.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a barrier. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierIncompleteSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierIncompleteSize.java index 1fe4a2be77e..88528da89eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierIncompleteSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierIncompleteSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -34,6 +36,10 @@ /** * Computes the number of incomplete elements in the given barrier. */ +@OpMetadata( + opType = BarrierIncompleteSize.OP_NAME, + inputsClass = BarrierIncompleteSize.Inputs.class +) @Operator public final class BarrierIncompleteSize extends RawOp implements Operand { /** @@ -43,8 +49,8 @@ public final class BarrierIncompleteSize extends RawOp implements Operand output; - private BarrierIncompleteSize(Operation operation) { - super(operation); + public BarrierIncompleteSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BarrierIncompleteSize.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a barrier. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierInsertMany.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierInsertMany.java index b5d28bdc4da..c660005a872 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierInsertMany.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierInsertMany.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * already has a value at component_index, this operation will fail with * INVALID_ARGUMENT, and leave the barrier in an undefined state. */ +@OpMetadata( + opType = BarrierInsertMany.OP_NAME, + inputsClass = BarrierInsertMany.Inputs.class +) @Operator public final class BarrierInsertMany extends RawOp { /** @@ -45,8 +51,8 @@ public final class BarrierInsertMany extends RawOp { */ public static final String OP_NAME = "BarrierInsertMany"; - private BarrierInsertMany(Operation operation) { - super(operation); + public BarrierInsertMany(Operation operation) { + super(operation, OP_NAME); } /** @@ -73,6 +79,9 @@ public static BarrierInsertMany create(Scope scope, Operand handle, return new BarrierInsertMany(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = BarrierInsertMany.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a barrier. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierReadySize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierReadySize.java index f38db0fd368..ded885bc84d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierReadySize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierReadySize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -34,6 +36,10 @@ /** * Computes the number of complete elements in the given barrier. */ +@OpMetadata( + opType = BarrierReadySize.OP_NAME, + inputsClass = BarrierReadySize.Inputs.class +) @Operator public final class BarrierReadySize extends RawOp implements Operand { /** @@ -43,8 +49,8 @@ public final class BarrierReadySize extends RawOp implements Operand { private Output output; - private BarrierReadySize(Operation operation) { - super(operation); + public BarrierReadySize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BarrierReadySize.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a barrier. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierTakeMany.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierTakeMany.java index baf77154ae3..18dacc2b92d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierTakeMany.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BarrierTakeMany.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -45,6 +47,10 @@ * information about the batch in which each element was originally inserted * into the barrier. */ +@OpMetadata( + opType = BarrierTakeMany.OP_NAME, + inputsClass = BarrierTakeMany.Inputs.class +) @Operator public final class BarrierTakeMany extends RawOp { /** @@ -59,8 +65,8 @@ public final class BarrierTakeMany extends RawOp { private List> values; @SuppressWarnings("unchecked") - private BarrierTakeMany(Operation operation) { - super(operation); + public BarrierTakeMany(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; indices = operation.output(outputIdx++); keys = operation.output(outputIdx++); @@ -219,6 +225,9 @@ public Options timeoutMs(Long timeoutMs) { } } + @OpInputsMetadata( + outputsClass = BarrierTakeMany.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a barrier. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Batch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Batch.java index ce2a30e4fce..b6437531c11 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Batch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Batch.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -67,6 +69,10 @@ * empty, the op name will be used as the shared name. * T: the types of tensors to be batched. */ +@OpMetadata( + opType = Batch.OP_NAME, + inputsClass = Batch.Inputs.class +) @Operator public final class Batch extends RawOp { /** @@ -81,8 +87,8 @@ public final class Batch extends RawOp { private Output id; @SuppressWarnings("unchecked") - private Batch(Operation operation) { - super(operation); + public Batch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int batchedTensorsLength = operation.outputListLength("batched_tensors"); batchedTensors = Arrays.asList(operation.outputList(outputIdx, batchedTensorsLength)); @@ -311,6 +317,9 @@ public Options batchingQueue(String batchingQueue) { } } + @OpInputsMetadata( + outputsClass = Batch.class + ) public static class Inputs extends RawOpInputs { /** * The inTensors input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchFunction.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchFunction.java index 25cba54989e..59537a13d93 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchFunction.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchFunction.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -70,6 +72,10 @@ *

SparseTensor is not supported. The return value of the decorated function * must be a Tensor or a list/tuple of Tensors. */ +@OpMetadata( + opType = BatchFunction.OP_NAME, + inputsClass = BatchFunction.Inputs.class +) @Operator public final class BatchFunction extends RawOp implements Iterable> { /** @@ -80,8 +86,8 @@ public final class BatchFunction extends RawOp implements Iterable> outTensors; @SuppressWarnings("unchecked") - private BatchFunction(Operation operation) { - super(operation); + public BatchFunction(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outTensorsLength = operation.outputListLength("out_tensors"); outTensors = Arrays.asList(operation.outputList(outputIdx, outTensorsLength)); @@ -353,6 +359,9 @@ public Options enableLargeBatchSplitting(Boolean enableLargeBatchSplitting) { } } + @OpInputsMetadata( + outputsClass = BatchFunction.class + ) public static class Inputs extends RawOpInputs { /** * The tensors to be batched. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpace.java index 9562debefb7..b4752272b0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpace.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchToSpace.OP_NAME, + inputsClass = BatchToSpace.Inputs.class +) @Operator public final class BatchToSpace extends RawOp implements Operand { /** @@ -52,8 +58,8 @@ public final class BatchToSpace extends RawOp implements Operan private Output output; - private BatchToSpace(Operation operation) { - super(operation); + public BatchToSpace(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -150,6 +156,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchToSpace.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D tensor with shape diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpaceNd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpaceNd.java index f197bf88fe1..c056ee528d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpaceNd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BatchToSpaceNd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchToSpaceNd.OP_NAME, + inputsClass = BatchToSpaceNd.Inputs.class +) @Operator public final class BatchToSpaceNd extends RawOp implements Operand { /** @@ -52,8 +58,8 @@ public final class BatchToSpaceNd extends RawOp implements Oper private Output output; - private BatchToSpaceNd(Operation operation) { - super(operation); + public BatchToSpaceNd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -185,6 +191,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchToSpaceNd.class + ) public static class Inputs extends RawOpInputs> { /** * N-D with shape {@code input_shape = [batch] + spatial_shape + remaining_shape}, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bitcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bitcast.java index e0982cfbd92..7a191514e42 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bitcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bitcast.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -95,6 +97,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Bitcast.OP_NAME, + inputsClass = Bitcast.Inputs.class +) @Operator public final class Bitcast extends RawOp implements Operand { /** @@ -104,8 +110,8 @@ public final class Bitcast extends RawOp implements Operand private Output output; - private Bitcast(Operation operation) { - super(operation); + public Bitcast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Bitcast.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastDynamicShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastDynamicShape.java index 55d084bf691..90df4598114 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastDynamicShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastDynamicShape.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code r0} output */ +@OpMetadata( + opType = BroadcastDynamicShape.OP_NAME, + inputsClass = BroadcastDynamicShape.Inputs.class +) @Operator public final class BroadcastDynamicShape extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class BroadcastDynamicShape extends RawOp implem private Output r0; - private BroadcastDynamicShape(Operation operation) { - super(operation); + public BroadcastDynamicShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; r0 = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return r0; } + @OpInputsMetadata( + outputsClass = BroadcastDynamicShape.class + ) public static class Inputs extends RawOpInputs> { /** * The s0 input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java index 807cf34adb8..c70e82caeb9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code r0} output */ +@OpMetadata( + opType = BroadcastGradientArgs.OP_NAME, + inputsClass = BroadcastGradientArgs.Inputs.class +) public final class BroadcastGradientArgs extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class BroadcastGradientArgs extends RawOp { private Output r1; - private BroadcastGradientArgs(Operation operation) { - super(operation); + public BroadcastGradientArgs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; r0 = operation.output(outputIdx++); r1 = operation.output(outputIdx++); @@ -91,6 +97,9 @@ public Output r1() { return r1; } + @OpInputsMetadata( + outputsClass = BroadcastGradientArgs.class + ) public static class Inputs extends RawOpInputs> { /** * The s0 input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastTo.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastTo.java index 1761611649e..02f7187275f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastTo.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastTo.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -65,6 +67,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BroadcastTo.OP_NAME, + inputsClass = BroadcastTo.Inputs.class +) @Operator public final class BroadcastTo extends RawOp implements Operand { /** @@ -74,8 +80,8 @@ public final class BroadcastTo extends RawOp implements Operand private Output output; - private BroadcastTo(Operation operation) { - super(operation); + public BroadcastTo(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -114,6 +120,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BroadcastTo.class + ) public static class Inputs extends RawOpInputs> { /** * A Tensor to broadcast. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bucketize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bucketize.java index 9ce50864f2b..68c99f15181 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bucketize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Bucketize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -45,6 +47,10 @@ * [3, 2] * [1, 3]] */ +@OpMetadata( + opType = Bucketize.OP_NAME, + inputsClass = Bucketize.Inputs.class +) @Operator public final class Bucketize extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class Bucketize extends RawOp implements Operand { private Output output; - private Bucketize(Operation operation) { - super(operation); + public Bucketize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Bucketize.class + ) public static class Inputs extends RawOpInputs { /** * Any shape of Tensor contains with int or float type. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ClipByValue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ClipByValue.java index 0936dc33ac8..a031889ee23 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ClipByValue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ClipByValue.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ClipByValue.OP_NAME, + inputsClass = ClipByValue.Inputs.class +) @Operator public final class ClipByValue extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class ClipByValue extends RawOp implements Operand private Output output; - private ClipByValue(Operation operation) { - super(operation); + public ClipByValue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ClipByValue.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CollectiveGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CollectiveGather.java index 65d455900d6..d30e8e38019 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CollectiveGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CollectiveGather.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @deprecated use {@link org.tensorflow.op.collective.Gather} instead */ +@OpMetadata( + opType = CollectiveGather.OP_NAME, + inputsClass = CollectiveGather.Inputs.class +) @Deprecated public final class CollectiveGather extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class CollectiveGather extends RawOp implements private Output data; - private CollectiveGather(Operation operation) { - super(operation); + public CollectiveGather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -158,6 +164,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = CollectiveGather.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java index 863f03c3063..05a2295e86a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * Returns a scalar variant tensor containing a single {@code CompositeTensorVariant} * with the specified Tensor components and TypeSpec. */ +@OpMetadata( + opType = CompositeTensorVariantFromComponents.OP_NAME, + inputsClass = CompositeTensorVariantFromComponents.Inputs.class +) public final class CompositeTensorVariantFromComponents extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class CompositeTensorVariantFromComponents extends RawOp implements private Output encoded; @SuppressWarnings("unchecked") - private CompositeTensorVariantFromComponents(Operation operation) { - super(operation); + public CompositeTensorVariantFromComponents(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; encoded = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return (Output) encoded; } + @OpInputsMetadata( + outputsClass = CompositeTensorVariantFromComponents.class + ) public static class Inputs extends RawOpInputs { /** * The component tensors for the extension type value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java index 061ee7bf0be..b1bf76d5de7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ *

Raises an error if {@code type_spec_proto} doesn't match the TypeSpec * in {@code encoded}. */ +@OpMetadata( + opType = CompositeTensorVariantToComponents.OP_NAME, + inputsClass = CompositeTensorVariantToComponents.Inputs.class +) public final class CompositeTensorVariantToComponents extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class CompositeTensorVariantToComponents extends RawOp implements I private List> components; @SuppressWarnings("unchecked") - private CompositeTensorVariantToComponents(Operation operation) { - super(operation); + public CompositeTensorVariantToComponents(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -94,6 +100,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = CompositeTensorVariantToComponents.class + ) public static class Inputs extends RawOpInputs { /** * A scalar {@code variant} Tensor containing an encoded ExtensionType value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Concat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Concat.java index d4b8127f644..4abd12fe0d8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Concat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Concat.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Concat.OP_NAME, + inputsClass = Concat.Inputs.class +) @Operator public final class Concat extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class Concat extends RawOp implements Operand { private Output output; - private Concat(Operation operation) { - super(operation); + public Concat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Concat.class + ) public static class Inputs extends RawOpInputs> { /** * List of {@code N} Tensors to concatenate. Their ranks and types must match, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConsumeMutexLock.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConsumeMutexLock.java index 3f66976d9c6..bb847edc7cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConsumeMutexLock.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConsumeMutexLock.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ *

NOTE: This operation must run on the same device as its input. This may * be enforced via the {@code colocate_with} mechanism. */ +@OpMetadata( + opType = ConsumeMutexLock.OP_NAME, + inputsClass = ConsumeMutexLock.Inputs.class +) @Operator public final class ConsumeMutexLock extends RawOp { /** @@ -45,8 +51,8 @@ public final class ConsumeMutexLock extends RawOp { */ public static final String OP_NAME = "ConsumeMutexLock"; - private ConsumeMutexLock(Operation operation) { - super(operation); + public ConsumeMutexLock(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static ConsumeMutexLock create(Scope scope, Operand mute return new ConsumeMutexLock(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ConsumeMutexLock.class + ) public static class Inputs extends RawOpInputs { /** * A tensor returned by {@code MutexLock}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ControlTrigger.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ControlTrigger.java index 03a2c6485d2..a17d125586f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ControlTrigger.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ControlTrigger.java @@ -25,12 +25,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; /** * Does nothing. Serves as a control trigger for scheduling. * Only useful as a placeholder for control edges. */ +@OpMetadata( + opType = ControlTrigger.OP_NAME, + inputsClass = ControlTrigger.Inputs.class +) @Operator public final class ControlTrigger extends RawOp { /** @@ -38,8 +44,8 @@ public final class ControlTrigger extends RawOp { */ public static final String OP_NAME = "ControlTrigger"; - private ControlTrigger(Operation operation) { - super(operation); + public ControlTrigger(Operation operation) { + super(operation, OP_NAME); } /** @@ -56,6 +62,9 @@ public static ControlTrigger create(Scope scope) { return new ControlTrigger(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ControlTrigger.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new ControlTrigger(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Copy.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Copy.java index 4138011fac8..4fb2dbf1269 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Copy.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Copy.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Copy.OP_NAME, + inputsClass = Copy.Inputs.class +) public final class Copy extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -51,8 +57,8 @@ public final class Copy extends RawOp implements Operand { private Output output; - private Copy(Operation operation) { - super(operation); + public Copy(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -195,6 +201,9 @@ public Options debugOpsSpec(String... debugOpsSpec) { } } + @OpInputsMetadata( + outputsClass = Copy.class + ) public static class Inputs extends RawOpInputs> { /** * Input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CopyHost.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CopyHost.java index d35bd8e9b81..0af9c1ac794 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CopyHost.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CopyHost.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CopyHost.OP_NAME, + inputsClass = CopyHost.Inputs.class +) public final class CopyHost extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class CopyHost extends RawOp implements Operand private Output output; - private CopyHost(Operation operation) { - super(operation); + public CopyHost(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -193,6 +199,9 @@ public Options debugOpsSpec(String... debugOpsSpec) { } } + @OpInputsMetadata( + outputsClass = CopyHost.class + ) public static class Inputs extends RawOpInputs> { /** * Input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CountUpTo.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CountUpTo.java index 94ca8305dd7..7ce1ece5a75 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CountUpTo.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CountUpTo.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CountUpTo.OP_NAME, + inputsClass = CountUpTo.Inputs.class +) @Operator public final class CountUpTo extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class CountUpTo extends RawOp implements Operand private Output output; - private CountUpTo(Operation operation) { - super(operation); + public CountUpTo(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = CountUpTo.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a scalar {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DecodeProto.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DecodeProto.java index 1011cac53b7..7c1818a97eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DecodeProto.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DecodeProto.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -89,6 +91,10 @@ * * */ +@OpMetadata( + opType = DecodeProto.OP_NAME, + inputsClass = DecodeProto.Inputs.class +) @Operator public final class DecodeProto extends RawOp { /** @@ -101,8 +107,8 @@ public final class DecodeProto extends RawOp { private List> values; @SuppressWarnings("unchecked") - private DecodeProto(Operation operation) { - super(operation); + public DecodeProto(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sizes = operation.output(outputIdx++); int valuesLength = operation.outputListLength("values"); @@ -253,6 +259,9 @@ public Options sanitize(Boolean sanitize) { } } + @OpInputsMetadata( + outputsClass = DecodeProto.class + ) public static class Inputs extends RawOpInputs { /** * Tensor of serialized protos with shape {@code batch_shape}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeepCopy.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeepCopy.java index ea55c769019..01d71313c42 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeepCopy.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeepCopy.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = DeepCopy.OP_NAME, + inputsClass = DeepCopy.Inputs.class +) @Operator public final class DeepCopy extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class DeepCopy extends RawOp implements Operand private Output y; - private DeepCopy(Operation operation) { - super(operation); + public DeepCopy(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = DeepCopy.class + ) public static class Inputs extends RawOpInputs> { /** * The source tensor of type {@code T}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeleteSessionTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeleteSessionTensor.java index 47e499eba70..de259a8599b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeleteSessionTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeleteSessionTensor.java @@ -26,12 +26,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** * Delete the tensor specified by its handle in the session. */ +@OpMetadata( + opType = DeleteSessionTensor.OP_NAME, + inputsClass = DeleteSessionTensor.Inputs.class +) @Operator public final class DeleteSessionTensor extends RawOp { /** @@ -39,8 +45,8 @@ public final class DeleteSessionTensor extends RawOp { */ public static final String OP_NAME = "DeleteSessionTensor"; - private DeleteSessionTensor(Operation operation) { - super(operation); + public DeleteSessionTensor(Operation operation) { + super(operation, OP_NAME); } /** @@ -59,6 +65,9 @@ public static DeleteSessionTensor create(Scope scope, Operand handle) { return new DeleteSessionTensor(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DeleteSessionTensor.class + ) public static class Inputs extends RawOpInputs { /** * The handle for a tensor stored in the session state. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyResourceOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyResourceOp.java index 040c054f085..b4baf185453 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyResourceOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyResourceOp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ * All subsequent operations using the resource will result in a NotFound * error status. */ +@OpMetadata( + opType = DestroyResourceOp.OP_NAME, + inputsClass = DestroyResourceOp.Inputs.class +) @Operator public final class DestroyResourceOp extends RawOp { /** @@ -41,8 +47,8 @@ public final class DestroyResourceOp extends RawOp { */ public static final String OP_NAME = "DestroyResourceOp"; - private DestroyResourceOp(Operation operation) { - super(operation); + public DestroyResourceOp(Operation operation) { + super(operation, OP_NAME); } /** @@ -103,6 +109,9 @@ public Options ignoreLookupError(Boolean ignoreLookupError) { } } + @OpInputsMetadata( + outputsClass = DestroyResourceOp.class + ) public static class Inputs extends RawOpInputs { /** * handle to the resource to delete. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyTemporaryVariable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyTemporaryVariable.java index 7d828ee733b..527f7179901 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyTemporaryVariable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DestroyTemporaryVariable.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code value} output */ +@OpMetadata( + opType = DestroyTemporaryVariable.OP_NAME, + inputsClass = DestroyTemporaryVariable.Inputs.class +) @Operator public final class DestroyTemporaryVariable extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class DestroyTemporaryVariable extends RawOp imple private Output value; - private DestroyTemporaryVariable(Operation operation) { - super(operation); + public DestroyTemporaryVariable(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; value = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return value; } + @OpInputsMetadata( + outputsClass = DestroyTemporaryVariable.class + ) public static class Inputs extends RawOpInputs> { /** * A reference to the temporary variable tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java index a06f456040c..fdb7d9c0b10 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; /** @@ -37,6 +39,10 @@ * (1) Device does not exist in the given device list. * (2) It is in XLA compilation. */ +@OpMetadata( + opType = DeviceIndex.OP_NAME, + inputsClass = DeviceIndex.Inputs.class +) public final class DeviceIndex extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class DeviceIndex extends RawOp implements Operand { private Output index; - private DeviceIndex(Operation operation) { - super(operation); + public DeviceIndex(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; index = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return index; } + @OpInputsMetadata( + outputsClass = DeviceIndex.class + ) public static class Inputs extends RawOpInputs { /** * The deviceNames attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java index 3415b800196..3be75c630c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The DummyMemoryCache operation */ +@OpMetadata( + opType = DummyMemoryCache.OP_NAME, + inputsClass = DummyMemoryCache.Inputs.class +) public final class DummyMemoryCache extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class DummyMemoryCache extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private DummyMemoryCache(Operation operation) { - super(operation); + public DummyMemoryCache(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -76,6 +82,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DummyMemoryCache.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new DummyMemoryCache(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicPartition.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicPartition.java index 5c65adef756..cebd26df319 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicPartition.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicPartition.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -70,6 +72,10 @@ * * @param data type for {@code outputs} output */ +@OpMetadata( + opType = DynamicPartition.OP_NAME, + inputsClass = DynamicPartition.Inputs.class +) @Operator public final class DynamicPartition extends RawOp implements Iterable> { /** @@ -80,8 +86,8 @@ public final class DynamicPartition extends RawOp implements It private List> outputs; @SuppressWarnings("unchecked") - private DynamicPartition(Operation operation) { - super(operation); + public DynamicPartition(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList((Output[]) operation.outputList(outputIdx, outputsLength)); @@ -125,6 +131,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = DynamicPartition.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicStitch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicStitch.java index da206083909..012d1231a37 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicStitch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DynamicStitch.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -91,6 +93,10 @@ * * @param data type for {@code merged} output */ +@OpMetadata( + opType = DynamicStitch.OP_NAME, + inputsClass = DynamicStitch.Inputs.class +) @Operator public final class DynamicStitch extends RawOp implements Operand { /** @@ -100,8 +106,8 @@ public final class DynamicStitch extends RawOp implements Opera private Output merged; - private DynamicStitch(Operation operation) { - super(operation); + public DynamicStitch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; merged = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Output asOutput() { return merged; } + @OpInputsMetadata( + outputsClass = DynamicStitch.class + ) public static class Inputs extends RawOpInputs> { /** * The indices input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EditDistance.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EditDistance.java index 63ab6da64a9..8e3557d5022 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EditDistance.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EditDistance.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -41,6 +43,10 @@ * (truth_indices, truth_values, truth_shape). *

The inputs are: */ +@OpMetadata( + opType = EditDistance.OP_NAME, + inputsClass = EditDistance.Inputs.class +) @Operator public final class EditDistance extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class EditDistance extends RawOp implements Operand { private Output output; - private EditDistance(Operation operation) { - super(operation); + public EditDistance(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -175,6 +181,9 @@ public Options normalize(Boolean normalize) { } } + @OpInputsMetadata( + outputsClass = EditDistance.class + ) public static class Inputs extends RawOpInputs { /** * The indices of the hypothesis list SparseTensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Empty.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Empty.java index 0a06df36054..6aea00cf250 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Empty.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Empty.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Empty.OP_NAME, + inputsClass = Empty.Inputs.class +) @Operator public final class Empty extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class Empty extends RawOp implements Operand { private Output output; - private Empty(Operation operation) { - super(operation); + public Empty(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options init(Boolean init) { } } + @OpInputsMetadata( + outputsClass = Empty.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. Represents the shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorList.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorList.java index 25fc298a92a..1ebd1112e2e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorList.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorList.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * element_dtype: the type of elements in the list. * element_shape: a shape compatible with that of elements in the list. */ +@OpMetadata( + opType = EmptyTensorList.OP_NAME, + inputsClass = EmptyTensorList.Inputs.class +) @Operator public final class EmptyTensorList extends RawOp implements Operand { /** @@ -52,8 +58,8 @@ public final class EmptyTensorList extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private EmptyTensorList(Operation operation) { - super(operation); + public EmptyTensorList(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = EmptyTensorList.class + ) public static class Inputs extends RawOpInputs { /** * The elementShape input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorMap.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorMap.java index ffdf712e26e..a6fb97ceda1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorMap.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EmptyTensorMap.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ * Creates and returns an empty tensor map. * handle: an empty tensor map */ +@OpMetadata( + opType = EmptyTensorMap.OP_NAME, + inputsClass = EmptyTensorMap.Inputs.class +) @Operator public final class EmptyTensorMap extends RawOp implements Operand { /** @@ -44,8 +50,8 @@ public final class EmptyTensorMap extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private EmptyTensorMap(Operation operation) { - super(operation); + public EmptyTensorMap(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -79,6 +85,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = EmptyTensorMap.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new EmptyTensorMap(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EncodeProto.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EncodeProto.java index 3beb517b637..2eb65960230 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EncodeProto.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EncodeProto.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -82,6 +84,10 @@ * * */ +@OpMetadata( + opType = EncodeProto.OP_NAME, + inputsClass = EncodeProto.Inputs.class +) @Operator public final class EncodeProto extends RawOp implements Operand { /** @@ -91,8 +97,8 @@ public final class EncodeProto extends RawOp implements Operand { private Output bytes; - private EncodeProto(Operation operation) { - super(operation); + public EncodeProto(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; bytes = operation.output(outputIdx++); } @@ -177,6 +183,9 @@ public Options descriptorSource(String descriptorSource) { } } + @OpInputsMetadata( + outputsClass = EncodeProto.class + ) public static class Inputs extends RawOpInputs { /** * Tensor of int32 with shape {@code [batch_shape, len(field_names)]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EnsureShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EnsureShape.java index 3c7443c40b7..13bdd8f7e94 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EnsureShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/EnsureShape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = EnsureShape.OP_NAME, + inputsClass = EnsureShape.Inputs.class +) @Operator public final class EnsureShape extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class EnsureShape extends RawOp implements Operand private Output output; - private EnsureShape(Operation operation) { - super(operation); + public EnsureShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = EnsureShape.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor, whose shape is to be validated. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java index 17562be32ee..82e1831b5f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Enter.OP_NAME, + inputsClass = Enter.Inputs.class +) public final class Enter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class Enter extends RawOp implements Operand { private Output output; - private Enter(Operation operation) { - super(operation); + public Enter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -153,6 +159,9 @@ public Options parallelIterations(Long parallelIterations) { } } + @OpInputsMetadata( + outputsClass = Enter.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be made available to the child frame. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java index 2942db758b4..ee8dcfb8ceb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Exit.OP_NAME, + inputsClass = Exit.Inputs.class +) public final class Exit extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class Exit extends RawOp implements Operand { private Output output; - private Exit(Operation operation) { - super(operation); + public Exit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Exit.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be made available to the parent frame. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExpandDims.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExpandDims.java index 0086c4259b3..4d7c1eea3cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExpandDims.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExpandDims.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -60,6 +62,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ExpandDims.OP_NAME, + inputsClass = ExpandDims.Inputs.class +) @Operator public final class ExpandDims extends RawOp implements Operand { /** @@ -69,8 +75,8 @@ public final class ExpandDims extends RawOp implements Operand< private Output output; - private ExpandDims(Operation operation) { - super(operation); + public ExpandDims(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -112,6 +118,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ExpandDims.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExtractVolumePatches.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExtractVolumePatches.java index 9db29df2224..669b81d1ae0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExtractVolumePatches.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ExtractVolumePatches.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code patches} output */ +@OpMetadata( + opType = ExtractVolumePatches.OP_NAME, + inputsClass = ExtractVolumePatches.Inputs.class +) @Operator public final class ExtractVolumePatches extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class ExtractVolumePatches extends RawOp impleme private Output patches; - private ExtractVolumePatches(Operation operation) { - super(operation); + public ExtractVolumePatches(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; patches = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return patches; } + @OpInputsMetadata( + outputsClass = ExtractVolumePatches.class + ) public static class Inputs extends RawOpInputs> { /** * 5-D Tensor with shape {@code [batch, in_planes, in_rows, in_cols, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fill.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fill.java index 1df3bb32e39..3041b796443 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fill.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fill.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -54,6 +56,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Fill.OP_NAME, + inputsClass = Fill.Inputs.class +) @Operator public final class Fill extends RawOp implements Operand { /** @@ -63,8 +69,8 @@ public final class Fill extends RawOp implements Operand { private Output output; - private Fill(Operation operation) { - super(operation); + public Fill(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Fill.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. Represents the shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fingerprint.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fingerprint.java index 805a0ad1025..d6ef4704e45 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fingerprint.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Fingerprint.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -59,6 +61,10 @@ * *

For string data, one should expect {@code Fingerprint(data) != Fingerprint(ReduceJoin(data))} in general. */ +@OpMetadata( + opType = Fingerprint.OP_NAME, + inputsClass = Fingerprint.Inputs.class +) @Operator public final class Fingerprint extends RawOp implements Operand { /** @@ -68,8 +74,8 @@ public final class Fingerprint extends RawOp implements Operand { private Output fingerprint; - private Fingerprint(Operation operation) { - super(operation); + public Fingerprint(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; fingerprint = operation.output(outputIdx++); } @@ -110,6 +116,9 @@ public Output asOutput() { return fingerprint; } + @OpInputsMetadata( + outputsClass = Fingerprint.class + ) public static class Inputs extends RawOpInputs { /** * Must have rank 1 or higher. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/For.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/For.java index 8089e05d434..e7778c59f31 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/For.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/For.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -43,6 +45,10 @@ * output = body(i, output); * */ +@OpMetadata( + opType = For.OP_NAME, + inputsClass = For.Inputs.class +) @Operator public final class For extends RawOp implements Iterable> { /** @@ -53,8 +59,8 @@ public final class For extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private For(Operation operation) { - super(operation); + public For(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -105,6 +111,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = For.class + ) public static class Inputs extends RawOpInputs { /** * The lower bound. An int32 diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Gather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Gather.java index 1507b1ca9ca..2d0275f26f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Gather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Gather.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -59,6 +61,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Gather.OP_NAME, + inputsClass = Gather.Inputs.class +) @Operator public final class Gather extends RawOp implements Operand { /** @@ -68,8 +74,8 @@ public final class Gather extends RawOp implements Operand { private Output output; - private Gather(Operation operation) { - super(operation); + public Gather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -152,6 +158,9 @@ public Options batchDims(Long batchDims) { } } + @OpInputsMetadata( + outputsClass = Gather.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor from which to gather values. Must be at least rank diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GatherNd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GatherNd.java index 0b4ee538a18..0528f5d3fbe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GatherNd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GatherNd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -126,6 +128,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = GatherNd.OP_NAME, + inputsClass = GatherNd.Inputs.class +) @Operator public final class GatherNd extends RawOp implements Operand { /** @@ -135,8 +141,8 @@ public final class GatherNd extends RawOp implements Operand private Output output; - private GatherNd(Operation operation) { - super(operation); + public GatherNd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -176,6 +182,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = GatherNd.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor from which to gather values. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionHandle.java index 60a52b39aaa..0538620d314 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionHandle.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Store the input tensor in the state of the current session. */ +@OpMetadata( + opType = GetSessionHandle.OP_NAME, + inputsClass = GetSessionHandle.Inputs.class +) @Operator public final class GetSessionHandle extends RawOp implements Operand { /** @@ -44,8 +50,8 @@ public final class GetSessionHandle extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private GetSessionHandle(Operation operation) { - super(operation); + public GetSessionHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = GetSessionHandle.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to be stored. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionTensor.java index c8a85d0b714..417eba05f1d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetSessionTensor.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * * @param data type for {@code value} output */ +@OpMetadata( + opType = GetSessionTensor.OP_NAME, + inputsClass = GetSessionTensor.Inputs.class +) @Operator public final class GetSessionTensor extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class GetSessionTensor extends RawOp implements Op private Output value; - private GetSessionTensor(Operation operation) { - super(operation); + public GetSessionTensor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; value = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return value; } + @OpInputsMetadata( + outputsClass = GetSessionTensor.class + ) public static class Inputs extends RawOpInputs> { /** * The handle for a tensor stored in the session state. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GuaranteeConst.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GuaranteeConst.java index 2f6e8c9dda2..4a2a67691b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GuaranteeConst.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GuaranteeConst.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = GuaranteeConst.OP_NAME, + inputsClass = GuaranteeConst.Inputs.class +) @Operator public final class GuaranteeConst extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class GuaranteeConst extends RawOp implements Oper private Output output; - private GuaranteeConst(Operation operation) { - super(operation); + public GuaranteeConst(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = GuaranteeConst.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HashTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HashTable.java index f81970293ed..d2791779391 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HashTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HashTable.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * Before using the table you will have to initialize it. After initialization the * table will be immutable. */ +@OpMetadata( + opType = HashTable.OP_NAME, + inputsClass = HashTable.Inputs.class +) @Operator public final class HashTable extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class HashTable extends RawOp implements Operand { private Output tableHandle; @SuppressWarnings("unchecked") - private HashTable(Operation operation) { - super(operation); + public HashTable(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tableHandle = operation.output(outputIdx++); } @@ -187,6 +193,9 @@ public Options useNodeNameSharing(Boolean useNodeNameSharing) { } } + @OpInputsMetadata( + outputsClass = HashTable.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this table is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HistogramFixedWidth.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HistogramFixedWidth.java index fc7ac24dbaf..13011fdd573 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HistogramFixedWidth.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HistogramFixedWidth.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -52,6 +54,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = HistogramFixedWidth.OP_NAME, + inputsClass = HistogramFixedWidth.Inputs.class +) @Operator public final class HistogramFixedWidth extends RawOp implements Operand { /** @@ -61,8 +67,8 @@ public final class HistogramFixedWidth extends RawOp implemen private Output out; - private HistogramFixedWidth(Operation operation) { - super(operation); + public HistogramFixedWidth(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -128,6 +134,9 @@ public Output asOutput() { return out; } + @OpInputsMetadata( + outputsClass = HistogramFixedWidth.class + ) public static class Inputs extends RawOpInputs> { /** * Numeric {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Identity.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Identity.java index a07329e1b19..e8b8f336869 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Identity.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Identity.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Identity.OP_NAME, + inputsClass = Identity.Inputs.class +) @Operator public final class Identity extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class Identity extends RawOp implements Operand private Output output; - private Identity(Operation operation) { - super(operation); + public Identity(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Identity.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IdentityN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IdentityN.java index ec6b7c51e61..fc06bb10126 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IdentityN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IdentityN.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -50,6 +52,10 @@ * return [None, g(dy)] # Do not backprop to f(x). * */ +@OpMetadata( + opType = IdentityN.OP_NAME, + inputsClass = IdentityN.Inputs.class +) @Operator public final class IdentityN extends RawOp implements Iterable> { /** @@ -60,8 +66,8 @@ public final class IdentityN extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private IdentityN(Operation operation) { - super(operation); + public IdentityN(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -99,6 +105,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = IdentityN.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ImmutableConst.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ImmutableConst.java index 8dbe85ffc6e..beacc9d4de7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ImmutableConst.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ImmutableConst.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = ImmutableConst.OP_NAME, + inputsClass = ImmutableConst.Inputs.class +) @Operator public final class ImmutableConst extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class ImmutableConst extends RawOp implements Oper private Output tensor; - private ImmutableConst(Operation operation) { - super(operation); + public ImmutableConst(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tensor = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return tensor; } + @OpInputsMetadata( + outputsClass = ImmutableConst.class + ) public static class Inputs extends RawOpInputs> { /** * Type of the returned tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTable.java index fa23039a392..2350327dfa7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTable.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -33,6 +35,10 @@ /** * Table initializer that takes two tensors for keys and values respectively. */ +@OpMetadata( + opType = InitializeTable.OP_NAME, + inputsClass = InitializeTable.Inputs.class +) @Operator public final class InitializeTable extends RawOp { /** @@ -40,8 +46,8 @@ public final class InitializeTable extends RawOp { */ public static final String OP_NAME = "InitializeTableV2"; - private InitializeTable(Operation operation) { - super(operation); + public InitializeTable(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static InitializeTable create(Scope scope, Operand table return new InitializeTable(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = InitializeTable.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a table which will be initialized. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTableFromTextFile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTableFromTextFile.java index 96f0274ddf7..f01af12b7bf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTableFromTextFile.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InitializeTableFromTextFile.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * on {@code delimiter}. * */ +@OpMetadata( + opType = InitializeTableFromTextFile.OP_NAME, + inputsClass = InitializeTableFromTextFile.Inputs.class +) @Operator public final class InitializeTableFromTextFile extends RawOp { /** @@ -51,8 +57,8 @@ public final class InitializeTableFromTextFile extends RawOp { */ public static final String OP_NAME = "InitializeTableFromTextFileV2"; - private InitializeTableFromTextFile(Operation operation) { - super(operation); + public InitializeTableFromTextFile(Operation operation) { + super(operation, OP_NAME); } /** @@ -171,6 +177,9 @@ public Options offset(Long offset) { } } + @OpInputsMetadata( + outputsClass = InitializeTableFromTextFile.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a table which will be initialized. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceAdd.java index b0f8ad1113c..71fcbe428d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = InplaceAdd.OP_NAME, + inputsClass = InplaceAdd.Inputs.class +) @Operator public final class InplaceAdd extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class InplaceAdd extends RawOp implements Operand< private Output y; - private InplaceAdd(Operation operation) { - super(operation); + public InplaceAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = InplaceAdd.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceSub.java index 87329d1405f..596a281f92c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceSub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = InplaceSub.OP_NAME, + inputsClass = InplaceSub.Inputs.class +) @Operator public final class InplaceSub extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class InplaceSub extends RawOp implements Operand< private Output y; - private InplaceSub(Operation operation) { - super(operation); + public InplaceSub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = InplaceSub.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceUpdate.java index b3cc7b8ca35..72237006fe3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/InplaceUpdate.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = InplaceUpdate.OP_NAME, + inputsClass = InplaceUpdate.Inputs.class +) @Operator public final class InplaceUpdate extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class InplaceUpdate extends RawOp implements Opera private Output y; - private InplaceUpdate(Operation operation) { - super(operation); + public InplaceUpdate(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = InplaceUpdate.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor of type {@code T}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IsVariableInitialized.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IsVariableInitialized.java index 2d386193212..b838486c19f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IsVariableInitialized.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/IsVariableInitialized.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -36,6 +38,10 @@ * Checks whether a tensor has been initialized. * Outputs boolean scalar indicating whether the tensor has been initialized. */ +@OpMetadata( + opType = IsVariableInitialized.OP_NAME, + inputsClass = IsVariableInitialized.Inputs.class +) @Operator public final class IsVariableInitialized extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class IsVariableInitialized extends RawOp implements Operand private Output isInitialized; - private IsVariableInitialized(Operation operation) { - super(operation); + public IsVariableInitialized(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; isInitialized = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return isInitialized; } + @OpInputsMetadata( + outputsClass = IsVariableInitialized.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. May be uninitialized. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/KthOrderStatistic.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/KthOrderStatistic.java index d8d7a3c2411..fe0b69c2112 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/KthOrderStatistic.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/KthOrderStatistic.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; @@ -47,6 +49,10 @@ * equal to the Kth order statistic. The semantics are not the same as * top_k_unique. */ +@OpMetadata( + opType = KthOrderStatistic.OP_NAME, + inputsClass = KthOrderStatistic.Inputs.class +) @Operator public final class KthOrderStatistic extends RawOp implements Operand { /** @@ -56,8 +62,8 @@ public final class KthOrderStatistic extends RawOp implements Operand private Output output; - private KthOrderStatistic(Operation operation) { - super(operation); + public KthOrderStatistic(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = KthOrderStatistic.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java index b8d1000d67e..085181e808e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = LinSpace.OP_NAME, + inputsClass = LinSpace.Inputs.class +) public final class LinSpace extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class LinSpace extends RawOp implements Operand< private Output output; - private LinSpace(Operation operation) { - super(operation); + public LinSpace(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = LinSpace.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D tensor. First entry in the range. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableExport.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableExport.java index d36982e1128..bfc4a05a26c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableExport.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableExport.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = LookupTableExport.OP_NAME, + inputsClass = LookupTableExport.Inputs.class +) @Operator public final class LookupTableExport extends RawOp { /** @@ -50,8 +56,8 @@ public final class LookupTableExport extends R private Output values; - private LookupTableExport(Operation operation) { - super(operation); + public LookupTableExport(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; keys = operation.output(outputIdx++); values = operation.output(outputIdx++); @@ -98,6 +104,9 @@ public Output values() { return values; } + @OpInputsMetadata( + outputsClass = LookupTableExport.class + ) public static class Inputs extends RawOpInputs> { /** * Handle to the table. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableFind.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableFind.java index fb9508e2d30..c9c192867a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableFind.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableFind.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = LookupTableFind.OP_NAME, + inputsClass = LookupTableFind.Inputs.class +) @Operator public final class LookupTableFind extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class LookupTableFind extends RawOp implements Ope private Output values; - private LookupTableFind(Operation operation) { - super(operation); + public LookupTableFind(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; values = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return values; } + @OpInputsMetadata( + outputsClass = LookupTableFind.class + ) public static class Inputs extends RawOpInputs> { /** * Handle to the table. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableImport.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableImport.java index b2483f12b25..0223c4370a8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableImport.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableImport.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * The tensor {@code keys} must be of the same type as the keys of the table. * The tensor {@code values} must be of the type of the table values. */ +@OpMetadata( + opType = LookupTableImport.OP_NAME, + inputsClass = LookupTableImport.Inputs.class +) @Operator public final class LookupTableImport extends RawOp { /** @@ -42,8 +48,8 @@ public final class LookupTableImport extends RawOp { */ public static final String OP_NAME = "LookupTableImportV2"; - private LookupTableImport(Operation operation) { - super(operation); + public LookupTableImport(Operation operation) { + super(operation, OP_NAME); } /** @@ -67,6 +73,9 @@ public static LookupTableImport create(Scope scope, Operand tab return new LookupTableImport(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = LookupTableImport.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the table. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableInsert.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableInsert.java index 143032f960a..a3b7207c594 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableInsert.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableInsert.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * The tensor {@code keys} must be of the same type as the keys of the table. * The tensor {@code values} must be of the type of the table values. */ +@OpMetadata( + opType = LookupTableInsert.OP_NAME, + inputsClass = LookupTableInsert.Inputs.class +) @Operator public final class LookupTableInsert extends RawOp { /** @@ -42,8 +48,8 @@ public final class LookupTableInsert extends RawOp { */ public static final String OP_NAME = "LookupTableInsertV2"; - private LookupTableInsert(Operation operation) { - super(operation); + public LookupTableInsert(Operation operation) { + super(operation, OP_NAME); } /** @@ -67,6 +73,9 @@ public static LookupTableInsert create(Scope scope, Operand tab return new LookupTableInsert(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = LookupTableInsert.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the table. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java index 1e6a0864a29..26b91e332cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * The tensor {@code keys} must of the same type as the keys of the table. Keys not * already in the table are silently ignored. */ +@OpMetadata( + opType = LookupTableRemove.OP_NAME, + inputsClass = LookupTableRemove.Inputs.class +) public final class LookupTableRemove extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LookupTableRemoveV2"; - private LookupTableRemove(Operation operation) { - super(operation); + public LookupTableRemove(Operation operation) { + super(operation, OP_NAME); } /** @@ -63,6 +69,9 @@ public static LookupTableRemove create(Scope scope, Operand tab return new LookupTableRemove(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = LookupTableRemove.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the table. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableSize.java index e437e22fa0e..406e4a7c8f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Computes the number of elements in the given table. */ +@OpMetadata( + opType = LookupTableSize.OP_NAME, + inputsClass = LookupTableSize.Inputs.class +) @Operator public final class LookupTableSize extends RawOp implements Operand { /** @@ -43,8 +49,8 @@ public final class LookupTableSize extends RawOp implements Operand { private Output output; - private LookupTableSize(Operation operation) { - super(operation); + public LookupTableSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -79,6 +85,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = LookupTableSize.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the table. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LoopCond.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LoopCond.java index 1322b0c6e8f..48a363cda0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LoopCond.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LoopCond.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; @@ -35,6 +37,10 @@ * This operator represents the loop termination condition used by the * "pivot" switches of a loop. */ +@OpMetadata( + opType = LoopCond.OP_NAME, + inputsClass = LoopCond.Inputs.class +) @Operator public final class LoopCond extends RawOp implements Operand { /** @@ -44,8 +50,8 @@ public final class LoopCond extends RawOp implements Operand { private Output output; - private LoopCond(Operation operation) { - super(operation); + public LoopCond(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = LoopCond.class + ) public static class Inputs extends RawOpInputs { /** * A boolean scalar, representing the branch predicate of the Switch op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java index 53a0a5b7028..7950ad2686b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = LowerBound.OP_NAME, + inputsClass = LowerBound.Inputs.class +) public final class LowerBound extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -59,8 +65,8 @@ public final class LowerBound extends RawOp implements Operan private Output output; - private LowerBound(Operation operation) { - super(operation); + public LowerBound(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -123,6 +129,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = LowerBound.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D Tensor where each row is ordered. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MakeUnique.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MakeUnique.java index 6e4ba6ab699..bb738fe065b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MakeUnique.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MakeUnique.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; @@ -37,6 +39,10 @@ * of the corresponding output element. Behavior for infinite elements is * undefined. Behavior for subnormal elements is undefined. */ +@OpMetadata( + opType = MakeUnique.OP_NAME, + inputsClass = MakeUnique.Inputs.class +) @Operator public final class MakeUnique extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class MakeUnique extends RawOp implements Operand { private Output output; - private MakeUnique(Operation operation) { - super(operation); + public MakeUnique(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = MakeUnique.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapClear.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapClear.java index 035faa1eac7..25616fab77d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapClear.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapClear.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Op removes all elements in the underlying container. */ +@OpMetadata( + opType = MapClear.OP_NAME, + inputsClass = MapClear.Inputs.class +) @Operator public final class MapClear extends RawOp { /** @@ -41,8 +47,8 @@ public final class MapClear extends RawOp { */ public static final String OP_NAME = "MapClear"; - private MapClear(Operation operation) { - super(operation); + public MapClear(Operation operation) { + super(operation, OP_NAME); } /** @@ -179,6 +185,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = MapClear.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java index f7762d6dbce..1be93e14437 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java @@ -32,6 +32,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -47,6 +49,10 @@ *

Note that this op is not exposed to users directly, but is invoked in tf.data * rewrites. */ +@OpMetadata( + opType = MapDefun.OP_NAME, + inputsClass = MapDefun.Inputs.class +) public final class MapDefun extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -56,8 +62,8 @@ public final class MapDefun extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private MapDefun(Operation operation) { - super(operation); + public MapDefun(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -158,6 +164,9 @@ public Options maxIntraOpParallelism(Long maxIntraOpParallelism) { } } + @OpInputsMetadata( + outputsClass = MapDefun.class + ) public static class Inputs extends RawOpInputs { /** *

diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapIncompleteSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapIncompleteSize.java
index 1f93fd8fae3..01f5a5db918 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapIncompleteSize.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapIncompleteSize.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -37,6 +39,10 @@
 /**
  * Op returns the number of incomplete elements in the underlying container.
  */
+@OpMetadata(
+    opType = MapIncompleteSize.OP_NAME,
+    inputsClass = MapIncompleteSize.Inputs.class
+)
 @Operator
 public final class MapIncompleteSize extends RawOp implements Operand {
   /**
@@ -46,8 +52,8 @@ public final class MapIncompleteSize extends RawOp implements Operand {
 
   private Output output;
 
-  private MapIncompleteSize(Operation operation) {
-    super(operation);
+  public MapIncompleteSize(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -200,6 +206,9 @@ public Options sharedName(String sharedName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = MapIncompleteSize.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The capacity attribute
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapPeek.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapPeek.java
index 6c0a543d536..651f1fad54c 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapPeek.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapPeek.java
@@ -30,6 +30,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -41,6 +43,10 @@
  * underlying container does not contain this key
  * this op will block until it does.
  */
+@OpMetadata(
+    opType = MapPeek.OP_NAME,
+    inputsClass = MapPeek.Inputs.class
+)
 @Operator
 public final class MapPeek extends RawOp implements Iterable> {
   /**
@@ -51,8 +57,8 @@ public final class MapPeek extends RawOp implements Iterable> {
   private List> values;
 
   @SuppressWarnings("unchecked")
-  private MapPeek(Operation operation) {
-    super(operation);
+  public MapPeek(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     int valuesLength = operation.outputListLength("values");
     values = Arrays.asList(operation.outputList(outputIdx, valuesLength));
@@ -212,6 +218,9 @@ public Options sharedName(String sharedName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = MapPeek.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The key input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapSize.java
index 30beae4600a..df150021bc0 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapSize.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapSize.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -37,6 +39,10 @@
 /**
  * Op returns the number of elements in the underlying container.
  */
+@OpMetadata(
+    opType = MapSize.OP_NAME,
+    inputsClass = MapSize.Inputs.class
+)
 @Operator
 public final class MapSize extends RawOp implements Operand {
   /**
@@ -46,8 +52,8 @@ public final class MapSize extends RawOp implements Operand {
 
   private Output output;
 
-  private MapSize(Operation operation) {
-    super(operation);
+  public MapSize(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -200,6 +206,9 @@ public Options sharedName(String sharedName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = MapSize.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The capacity attribute
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapStage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapStage.java
index 2a23b4711f3..f6638c34873 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapStage.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapStage.java
@@ -28,6 +28,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -37,6 +39,10 @@
 /**
  * Stage (key, values) in the underlying container which behaves like a hashtable.
  */
+@OpMetadata(
+    opType = MapStage.OP_NAME,
+    inputsClass = MapStage.Inputs.class
+)
 @Operator
 public final class MapStage extends RawOp {
   /**
@@ -44,8 +50,8 @@ public final class MapStage extends RawOp {
    */
   public static final String OP_NAME = "MapStage";
 
-  private MapStage(Operation operation) {
-    super(operation);
+  public MapStage(Operation operation) {
+    super(operation, OP_NAME);
   }
 
   /**
@@ -193,6 +199,9 @@ public Options sharedName(String sharedName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = MapStage.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * int64
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstage.java
index 7eb18247706..f5d26ae1171 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstage.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstage.java
@@ -30,6 +30,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -41,6 +43,10 @@
  * from the underlying container.   If the underlying container
  * does not contain this key, the op will block until it does.
  */
+@OpMetadata(
+    opType = MapUnstage.OP_NAME,
+    inputsClass = MapUnstage.Inputs.class
+)
 @Operator
 public final class MapUnstage extends RawOp implements Iterable> {
   /**
@@ -51,8 +57,8 @@ public final class MapUnstage extends RawOp implements Iterable>
   private List> values;
 
   @SuppressWarnings("unchecked")
-  private MapUnstage(Operation operation) {
-    super(operation);
+  public MapUnstage(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     int valuesLength = operation.outputListLength("values");
     values = Arrays.asList(operation.outputList(outputIdx, valuesLength));
@@ -212,6 +218,9 @@ public Options sharedName(String sharedName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = MapUnstage.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The key input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstageNoKey.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstageNoKey.java
index 753817a5d3b..e1975a4a4e2 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstageNoKey.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapUnstageNoKey.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -40,6 +42,10 @@
  * from the underlying container.   If the underlying container
  * does not contain elements, the op will block until it does.
  */
+@OpMetadata(
+    opType = MapUnstageNoKey.OP_NAME,
+    inputsClass = MapUnstageNoKey.Inputs.class
+)
 @Operator
 public final class MapUnstageNoKey extends RawOp {
   /**
@@ -52,8 +58,8 @@ public final class MapUnstageNoKey extends RawOp {
   private List> values;
 
   @SuppressWarnings("unchecked")
-  private MapUnstageNoKey(Operation operation) {
-    super(operation);
+  public MapUnstageNoKey(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     key = operation.output(outputIdx++);
     int valuesLength = operation.outputListLength("values");
@@ -215,6 +221,9 @@ public Options sharedName(String sharedName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = MapUnstageNoKey.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The indices input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Max.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Max.java
index 71ae0f7c0e5..bc5577a29f3 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Max.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Max.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -40,6 +42,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = Max.OP_NAME,
+    inputsClass = Max.Inputs.class
+)
 @Operator
 public final class Max extends RawOp implements Operand {
   /**
@@ -49,8 +55,8 @@ public final class Max extends RawOp implements Operand {
 
   private Output output;
 
-  private Max(Operation operation) {
-    super(operation);
+  public Max(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -129,6 +135,9 @@ public Options keepDims(Boolean keepDims) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = Max.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The tensor to reduce.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Merge.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Merge.java
index 1117e800c59..e06fe25d533 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Merge.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Merge.java
@@ -28,6 +28,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
@@ -42,6 +44,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = Merge.OP_NAME,
+    inputsClass = Merge.Inputs.class
+)
 @Operator
 public final class Merge extends RawOp {
   /**
@@ -53,8 +59,8 @@ public final class Merge extends RawOp {
 
   private Output valueIndex;
 
-  private Merge(Operation operation) {
-    super(operation);
+  public Merge(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
     valueIndex = operation.output(outputIdx++);
@@ -95,6 +101,9 @@ public Output valueIndex() {
     return valueIndex;
   }
 
+  @OpInputsMetadata(
+      outputsClass = Merge.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The input tensors, exactly one of which will become available.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Min.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Min.java
index 695c92a70e8..8e7700b2dba 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Min.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Min.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -40,6 +42,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = Min.OP_NAME,
+    inputsClass = Min.Inputs.class
+)
 @Operator
 public final class Min extends RawOp implements Operand {
   /**
@@ -49,8 +55,8 @@ public final class Min extends RawOp implements Operand {
 
   private Output output;
 
-  private Min(Operation operation) {
-    super(operation);
+  public Min(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -129,6 +135,9 @@ public Options keepDims(Boolean keepDims) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = Min.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The tensor to reduce.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPad.java
index d557fc71ea5..192e74449da 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPad.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPad.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -58,6 +60,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = MirrorPad.OP_NAME,
+    inputsClass = MirrorPad.Inputs.class
+)
 @Operator
 public final class MirrorPad extends RawOp implements Operand {
   /**
@@ -67,8 +73,8 @@ public final class MirrorPad extends RawOp implements Operand output;
 
-  private MirrorPad(Operation operation) {
-    super(operation);
+  public MirrorPad(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -114,6 +120,9 @@ public Output asOutput() {
     return output;
   }
 
+  @OpInputsMetadata(
+      outputsClass = MirrorPad.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The input tensor to be padded.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java
index 6d4efbf43c3..8f14d7d0ddb 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
 import org.tensorflow.types.family.TType;
@@ -50,6 +52,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = MirrorPadGrad.OP_NAME,
+    inputsClass = MirrorPadGrad.Inputs.class
+)
 public final class MirrorPadGrad extends RawOp implements Operand {
   /**
    * The name of this op, as known by TensorFlow core engine
@@ -58,8 +64,8 @@ public final class MirrorPadGrad extends RawOp implements Opera
 
   private Output output;
 
-  private MirrorPadGrad(Operation operation) {
-    super(operation);
+  public MirrorPadGrad(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -101,6 +107,9 @@ public Output asOutput() {
     return output;
   }
 
+  @OpInputsMetadata(
+      outputsClass = MirrorPadGrad.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The input tensor to be folded.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MlirPassthroughOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MlirPassthroughOp.java
index 67dda410e1f..8881c597d79 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MlirPassthroughOp.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MlirPassthroughOp.java
@@ -30,6 +30,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -65,6 +67,10 @@
  * graph_def = foo.get_concrete_function(tf.TensorSpec([10], tf.float32), tf.TensorSpec([10], tf.float32)).graph.as_graph_def()
  * 
*/ +@OpMetadata( + opType = MlirPassthroughOp.OP_NAME, + inputsClass = MlirPassthroughOp.Inputs.class +) @Operator public final class MlirPassthroughOp extends RawOp implements Iterable> { /** @@ -75,8 +81,8 @@ public final class MlirPassthroughOp extends RawOp implements Iterable> outputs; @SuppressWarnings("unchecked") - private MlirPassthroughOp(Operation operation) { - super(operation); + public MlirPassthroughOp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList(operation.outputList(outputIdx, outputsLength)); @@ -119,6 +125,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = MlirPassthroughOp.class + ) public static class Inputs extends RawOpInputs { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableDenseHashTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableDenseHashTable.java index ad2418679a5..6e92465db2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableDenseHashTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableDenseHashTable.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * values. Each value must be a scalar. Data can be inserted into the table using * the insert operations. It does not support the initialization operation. */ +@OpMetadata( + opType = MutableDenseHashTable.OP_NAME, + inputsClass = MutableDenseHashTable.Inputs.class +) @Operator public final class MutableDenseHashTable extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class MutableDenseHashTable extends RawOp implements Operand private Output tableHandle; @SuppressWarnings("unchecked") - private MutableDenseHashTable(Operation operation) { - super(operation); + public MutableDenseHashTable(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tableHandle = operation.output(outputIdx++); } @@ -273,6 +279,9 @@ public Options maxLoadFactor(Float maxLoadFactor) { } } + @OpInputsMetadata( + outputsClass = MutableDenseHashTable.class + ) public static class Inputs extends RawOpInputs { /** * The key used to represent empty key buckets internally. Must not diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTable.java index dec668bf3b0..80f45de4bd2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTable.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * values. Each value must be a scalar. Data can be inserted into the table using * the insert operations. It does not support the initialization operation. */ +@OpMetadata( + opType = MutableHashTable.OP_NAME, + inputsClass = MutableHashTable.Inputs.class +) @Operator public final class MutableHashTable extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class MutableHashTable extends RawOp implements Operand { private Output tableHandle; @SuppressWarnings("unchecked") - private MutableHashTable(Operation operation) { - super(operation); + public MutableHashTable(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tableHandle = operation.output(outputIdx++); } @@ -187,6 +193,9 @@ public Options useNodeNameSharing(Boolean useNodeNameSharing) { } } + @OpInputsMetadata( + outputsClass = MutableHashTable.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this table is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTableOfTensors.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTableOfTensors.java index 8c11236dc3f..8e6220eb7eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTableOfTensors.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutableHashTableOfTensors.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * values. Each value must be a vector. Data can be inserted into the table using * the insert operations. It does not support the initialization operation. */ +@OpMetadata( + opType = MutableHashTableOfTensors.OP_NAME, + inputsClass = MutableHashTableOfTensors.Inputs.class +) @Operator public final class MutableHashTableOfTensors extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class MutableHashTableOfTensors extends RawOp implements Operand tableHandle; @SuppressWarnings("unchecked") - private MutableHashTableOfTensors(Operation operation) { - super(operation); + public MutableHashTableOfTensors(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tableHandle = operation.output(outputIdx++); } @@ -212,6 +218,9 @@ public Options valueShape(Shape valueShape) { } } + @OpInputsMetadata( + outputsClass = MutableHashTableOfTensors.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this table is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Mutex.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Mutex.java index 0c50b069a1d..5bcaa8f998a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Mutex.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Mutex.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * Creates a Mutex resource that can be locked by {@code MutexLock}. */ +@OpMetadata( + opType = Mutex.OP_NAME, + inputsClass = Mutex.Inputs.class +) @Operator public final class Mutex extends RawOp implements Operand { /** @@ -43,8 +49,8 @@ public final class Mutex extends RawOp implements Operand { private Output resource; @SuppressWarnings("unchecked") - private Mutex(Operation operation) { - super(operation); + public Mutex(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resource = operation.output(outputIdx++); } @@ -147,6 +153,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = Mutex.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this variable is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutexLock.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutexLock.java index 080673e5d78..2e837ea676f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutexLock.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MutexLock.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -66,6 +68,10 @@ *

It is also useful if two separate functions must share a resource, but we * wish to ensure the usage is exclusive. */ +@OpMetadata( + opType = MutexLock.OP_NAME, + inputsClass = MutexLock.Inputs.class +) @Operator public final class MutexLock extends RawOp implements Operand { /** @@ -76,8 +82,8 @@ public final class MutexLock extends RawOp implements Operand { private Output mutexLock; @SuppressWarnings("unchecked") - private MutexLock(Operation operation) { - super(operation); + public MutexLock(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; mutexLock = operation.output(outputIdx++); } @@ -115,6 +121,9 @@ public Output asOutput() { return (Output) mutexLock; } + @OpInputsMetadata( + outputsClass = MutexLock.class + ) public static class Inputs extends RawOpInputs { /** * The mutex resource to lock. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java index a59d41302c5..9756119224b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -47,6 +49,10 @@ * * @deprecated use {@link org.tensorflow.op.distribute.NcclAllReduce} instead */ +@OpMetadata( + opType = NcclAllReduce.OP_NAME, + inputsClass = NcclAllReduce.Inputs.class +) @Deprecated public final class NcclAllReduce extends RawOp implements Operand { /** @@ -56,8 +62,8 @@ public final class NcclAllReduce extends RawOp implements Ope private Output data; - private NcclAllReduce(Operation operation) { - super(operation); + public NcclAllReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return data; } + @OpInputsMetadata( + outputsClass = NcclAllReduce.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java index c8b9e6d3d88..328ef159e3b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +46,10 @@ * * @deprecated use {@link org.tensorflow.op.distribute.NcclBroadcast} instead */ +@OpMetadata( + opType = NcclBroadcast.OP_NAME, + inputsClass = NcclBroadcast.Inputs.class +) @Deprecated public final class NcclBroadcast extends RawOp implements Operand { /** @@ -53,8 +59,8 @@ public final class NcclBroadcast extends RawOp implements Ope private Output output; - private NcclBroadcast(Operation operation) { - super(operation); + public NcclBroadcast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = NcclBroadcast.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java index 97f3b57983d..f748163d26b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +46,10 @@ * * @deprecated use {@link org.tensorflow.op.distribute.NcclReduce} instead */ +@OpMetadata( + opType = NcclReduce.OP_NAME, + inputsClass = NcclReduce.Inputs.class +) @Deprecated public final class NcclReduce extends RawOp implements Operand { /** @@ -53,8 +59,8 @@ public final class NcclReduce extends RawOp implements Operan private Output data; - private NcclReduce(Operation operation) { - super(operation); + public NcclReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return data; } + @OpInputsMetadata( + outputsClass = NcclReduce.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NextIteration.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NextIteration.java index 63f3133570e..89639dc0df2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NextIteration.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NextIteration.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = NextIteration.OP_NAME, + inputsClass = NextIteration.Inputs.class +) @Operator public final class NextIteration extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class NextIteration extends RawOp implements Opera private Output output; - private NextIteration(Operation operation) { - super(operation); + public NextIteration(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = NextIteration.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be made available to the next iteration. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NoOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NoOp.java index 1fd31154b5d..c0a17a52b07 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NoOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NoOp.java @@ -25,11 +25,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; /** * Does nothing. Only useful as a placeholder for control edges. */ +@OpMetadata( + opType = NoOp.OP_NAME, + inputsClass = NoOp.Inputs.class +) @Operator public final class NoOp extends RawOp { /** @@ -37,8 +43,8 @@ public final class NoOp extends RawOp { */ public static final String OP_NAME = "NoOp"; - private NoOp(Operation operation) { - super(operation); + public NoOp(Operation operation) { + super(operation, OP_NAME); } /** @@ -55,6 +61,9 @@ public static NoOp create(Scope scope) { return new NoOp(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = NoOp.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new NoOp(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OneHot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OneHot.java index 52e2c895139..90a6701ea87 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OneHot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OneHot.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -112,6 +114,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = OneHot.OP_NAME, + inputsClass = OneHot.Inputs.class +) @Operator public final class OneHot extends RawOp implements Operand { /** @@ -121,8 +127,8 @@ public final class OneHot extends RawOp implements Operand { private Output output; - private OneHot(Operation operation) { - super(operation); + public OneHot(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -204,6 +210,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = OneHot.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor of indices. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OnesLike.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OnesLike.java index cba88e78d14..9e5255618ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OnesLike.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OnesLike.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = OnesLike.OP_NAME, + inputsClass = OnesLike.Inputs.class +) @Operator public final class OnesLike extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class OnesLike extends RawOp implements Operand private Output y; - private OnesLike(Operation operation) { - super(operation); + public OnesLike(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = OnesLike.class + ) public static class Inputs extends RawOpInputs> { /** * a tensor of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapClear.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapClear.java index 7899b26b3a3..7590b5e7c7b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapClear.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapClear.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Op removes all elements in the underlying container. */ +@OpMetadata( + opType = OrderedMapClear.OP_NAME, + inputsClass = OrderedMapClear.Inputs.class +) @Operator public final class OrderedMapClear extends RawOp { /** @@ -41,8 +47,8 @@ public final class OrderedMapClear extends RawOp { */ public static final String OP_NAME = "OrderedMapClear"; - private OrderedMapClear(Operation operation) { - super(operation); + public OrderedMapClear(Operation operation) { + super(operation, OP_NAME); } /** @@ -179,6 +185,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapClear.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapIncompleteSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapIncompleteSize.java index f04fbc1c64b..9ea80c8e8fb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapIncompleteSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapIncompleteSize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ /** * Op returns the number of incomplete elements in the underlying container. */ +@OpMetadata( + opType = OrderedMapIncompleteSize.OP_NAME, + inputsClass = OrderedMapIncompleteSize.Inputs.class +) @Operator public final class OrderedMapIncompleteSize extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class OrderedMapIncompleteSize extends RawOp implements Operand output; - private OrderedMapIncompleteSize(Operation operation) { - super(operation); + public OrderedMapIncompleteSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -200,6 +206,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapIncompleteSize.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapPeek.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapPeek.java index 91e50b0b013..47a76d27214 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapPeek.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapPeek.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * this op will block until it does. This Op is optimized for * performance. */ +@OpMetadata( + opType = OrderedMapPeek.OP_NAME, + inputsClass = OrderedMapPeek.Inputs.class +) @Operator public final class OrderedMapPeek extends RawOp implements Iterable> { /** @@ -52,8 +58,8 @@ public final class OrderedMapPeek extends RawOp implements Iterable> values; @SuppressWarnings("unchecked") - private OrderedMapPeek(Operation operation) { - super(operation); + public OrderedMapPeek(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int valuesLength = operation.outputListLength("values"); values = Arrays.asList(operation.outputList(outputIdx, valuesLength)); @@ -213,6 +219,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapPeek.class + ) public static class Inputs extends RawOpInputs { /** * The key input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapSize.java index f8f939ba8f4..d680834b582 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapSize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ /** * Op returns the number of elements in the underlying container. */ +@OpMetadata( + opType = OrderedMapSize.OP_NAME, + inputsClass = OrderedMapSize.Inputs.class +) @Operator public final class OrderedMapSize extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class OrderedMapSize extends RawOp implements Operand { private Output output; - private OrderedMapSize(Operation operation) { - super(operation); + public OrderedMapSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -200,6 +206,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapSize.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapStage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapStage.java index 14edbf02357..320da76943d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapStage.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapStage.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * Stage (key, values) in the underlying container which behaves like a ordered * associative container. Elements are ordered by key. */ +@OpMetadata( + opType = OrderedMapStage.OP_NAME, + inputsClass = OrderedMapStage.Inputs.class +) @Operator public final class OrderedMapStage extends RawOp { /** @@ -45,8 +51,8 @@ public final class OrderedMapStage extends RawOp { */ public static final String OP_NAME = "OrderedMapStage"; - private OrderedMapStage(Operation operation) { - super(operation); + public OrderedMapStage(Operation operation) { + super(operation, OP_NAME); } /** @@ -194,6 +200,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapStage.class + ) public static class Inputs extends RawOpInputs { /** * int64 diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstage.java index e6596e04550..da638467953 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstage.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstage.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * from the underlying container. If the underlying container * does not contain this key, the op will block until it does. */ +@OpMetadata( + opType = OrderedMapUnstage.OP_NAME, + inputsClass = OrderedMapUnstage.Inputs.class +) @Operator public final class OrderedMapUnstage extends RawOp implements Iterable> { /** @@ -51,8 +57,8 @@ public final class OrderedMapUnstage extends RawOp implements Iterable> values; @SuppressWarnings("unchecked") - private OrderedMapUnstage(Operation operation) { - super(operation); + public OrderedMapUnstage(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int valuesLength = operation.outputListLength("values"); values = Arrays.asList(operation.outputList(outputIdx, valuesLength)); @@ -212,6 +218,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapUnstage.class + ) public static class Inputs extends RawOpInputs { /** * The key input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstageNoKey.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstageNoKey.java index 7bc8fb6219b..cc08b12987b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstageNoKey.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/OrderedMapUnstageNoKey.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * key from the underlying container. If the underlying container * does not contain elements, the op will block until it does. */ +@OpMetadata( + opType = OrderedMapUnstageNoKey.OP_NAME, + inputsClass = OrderedMapUnstageNoKey.Inputs.class +) @Operator public final class OrderedMapUnstageNoKey extends RawOp { /** @@ -52,8 +58,8 @@ public final class OrderedMapUnstageNoKey extends RawOp { private List> values; @SuppressWarnings("unchecked") - private OrderedMapUnstageNoKey(Operation operation) { - super(operation); + public OrderedMapUnstageNoKey(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; key = operation.output(outputIdx++); int valuesLength = operation.outputListLength("values"); @@ -215,6 +221,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OrderedMapUnstageNoKey.class + ) public static class Inputs extends RawOpInputs { /** * The indices input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Pad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Pad.java index 434a94076fa..84dc8946ab9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Pad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Pad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +59,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Pad.OP_NAME, + inputsClass = Pad.Inputs.class +) @Operator public final class Pad extends RawOp implements Operand { /** @@ -66,8 +72,8 @@ public final class Pad extends RawOp implements Operand { private Output output; - private Pad(Operation operation) { - super(operation); + public Pad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Pad.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelConcat.java index 093ac6c690d..7f8e1fb9b88 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelConcat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelConcat.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -51,6 +53,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ParallelConcat.OP_NAME, + inputsClass = ParallelConcat.Inputs.class +) @Operator public final class ParallelConcat extends RawOp implements Operand { /** @@ -60,8 +66,8 @@ public final class ParallelConcat extends RawOp implements Oper private Output output; - private ParallelConcat(Operation operation) { - super(operation); + public ParallelConcat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -102,6 +108,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ParallelConcat.class + ) public static class Inputs extends RawOpInputs> { /** * Tensors to be concatenated. All must have size 1 in the first dimension diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelDynamicStitch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelDynamicStitch.java index abba8e503f3..04a68a9162c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelDynamicStitch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ParallelDynamicStitch.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -90,6 +92,10 @@ * * @param data type for {@code merged} output */ +@OpMetadata( + opType = ParallelDynamicStitch.OP_NAME, + inputsClass = ParallelDynamicStitch.Inputs.class +) @Operator public final class ParallelDynamicStitch extends RawOp implements Operand { /** @@ -99,8 +105,8 @@ public final class ParallelDynamicStitch extends RawOp implemen private Output merged; - private ParallelDynamicStitch(Operation operation) { - super(operation); + public ParallelDynamicStitch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; merged = operation.output(outputIdx++); } @@ -139,6 +145,9 @@ public Output asOutput() { return merged; } + @OpInputsMetadata( + outputsClass = ParallelDynamicStitch.class + ) public static class Inputs extends RawOpInputs> { /** * The indices input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Placeholder.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Placeholder.java index 11a54982337..58663caf629 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Placeholder.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Placeholder.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Placeholder.OP_NAME, + inputsClass = Placeholder.Inputs.class +) @Operator public final class Placeholder extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class Placeholder extends RawOp implements Operand private Output output; - private Placeholder(Operation operation) { - super(operation); + public Placeholder(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options shape(Shape shape) { } } + @OpInputsMetadata( + outputsClass = Placeholder.class + ) public static class Inputs extends RawOpInputs> { /** * The type of elements in the tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/PlaceholderWithDefault.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/PlaceholderWithDefault.java index dec98e433f9..c12c8cbe9a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/PlaceholderWithDefault.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/PlaceholderWithDefault.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = PlaceholderWithDefault.OP_NAME, + inputsClass = PlaceholderWithDefault.Inputs.class +) @Operator public final class PlaceholderWithDefault extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class PlaceholderWithDefault extends RawOp impleme private Output output; - private PlaceholderWithDefault(Operation operation) { - super(operation); + public PlaceholderWithDefault(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = PlaceholderWithDefault.class + ) public static class Inputs extends RawOpInputs> { /** * The default value to produce when {@code output} is not fed. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Print.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Print.java index d957b42e290..5314118c811 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Print.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Print.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -33,6 +35,10 @@ * Prints a string scalar. * Prints a string scalar to the desired output_stream. */ +@OpMetadata( + opType = Print.OP_NAME, + inputsClass = Print.Inputs.class +) @Operator public final class Print extends RawOp { /** @@ -40,8 +46,8 @@ public final class Print extends RawOp { */ public static final String OP_NAME = "PrintV2"; - private Print(Operation operation) { - super(operation); + public Print(Operation operation) { + super(operation, OP_NAME); } /** @@ -125,6 +131,9 @@ public Options end(String end) { } } + @OpInputsMetadata( + outputsClass = Print.class + ) public static class Inputs extends RawOpInputs { /** * The string scalar to print. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Prod.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Prod.java index 7404306824e..fc75390d995 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Prod.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Prod.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Prod.OP_NAME, + inputsClass = Prod.Inputs.class +) @Operator public final class Prod extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class Prod extends RawOp implements Operand { private Output output; - private Prod(Operation operation) { - super(operation); + public Prod(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = Prod.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/QuantizedReshape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/QuantizedReshape.java index b301d427311..1b9bbcf38eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/QuantizedReshape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/QuantizedReshape.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedReshape.OP_NAME, + inputsClass = QuantizedReshape.Inputs.class +) @Operator public final class QuantizedReshape extends RawOp { /** @@ -51,8 +57,8 @@ public final class QuantizedReshape extends RawOp { private Output outputMax; - private QuantizedReshape(Operation operation) { - super(operation); + public QuantizedReshape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputMin = operation.output(outputIdx++); @@ -110,6 +116,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = QuantizedReshape.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Range.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Range.java index d06a61b77d7..3acfff8153b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Range.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Range.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -45,6 +47,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Range.OP_NAME, + inputsClass = Range.Inputs.class +) @Operator public final class Range extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class Range extends RawOp implements Operand private Output output; - private Range(Operation operation) { - super(operation); + public Range(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Range.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D (scalar). First entry in the sequence. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Rank.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Rank.java index 8561087dad8..ca96957eeda 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Rank.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Rank.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -45,6 +47,10 @@ * of a tensor is the number of indices required to uniquely select each element * of the tensor. Rank is also known as "order", "degree", or "ndims." */ +@OpMetadata( + opType = Rank.OP_NAME, + inputsClass = Rank.Inputs.class +) @Operator public final class Rank extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class Rank extends RawOp implements Operand { private Output output; - private Rank(Operation operation) { - super(operation); + public Rank(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Rank.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReadVariableOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReadVariableOp.java index f461b9fd855..699f3a2b6d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReadVariableOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReadVariableOp.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code value} output */ +@OpMetadata( + opType = ReadVariableOp.OP_NAME, + inputsClass = ReadVariableOp.Inputs.class +) @Operator public final class ReadVariableOp extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class ReadVariableOp extends RawOp implements Oper private Output value; - private ReadVariableOp(Operation operation) { - super(operation); + public ReadVariableOp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; value = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return value; } + @OpInputsMetadata( + outputsClass = ReadVariableOp.class + ) public static class Inputs extends RawOpInputs> { /** * handle to the resource in which to store the variable. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java index 5235cae884c..2b62d24bb82 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = Recv.OP_NAME, + inputsClass = Recv.Inputs.class +) public final class Recv extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class Recv extends RawOp implements Operand { private Output tensor; - private Recv(Operation operation) { - super(operation); + public Recv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tensor = operation.output(outputIdx++); } @@ -136,6 +142,9 @@ public Options clientTerminated(Boolean clientTerminated) { } } + @OpInputsMetadata( + outputsClass = Recv.class + ) public static class Inputs extends RawOpInputs> { /** * The tensorType attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAll.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAll.java index 8858cdeb7f4..0560ec3fe42 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAll.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAll.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ * {@code axis}. If {@code keep_dims} is true, the reduced dimensions are * retained with length 1. */ +@OpMetadata( + opType = ReduceAll.OP_NAME, + inputsClass = ReduceAll.Inputs.class +) @Operator public final class ReduceAll extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class ReduceAll extends RawOp implements Operand { private Output output; - private ReduceAll(Operation operation) { - super(operation); + public ReduceAll(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = ReduceAll.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAny.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAny.java index 5dcc0185ac8..3d37b5b7f48 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAny.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceAny.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ * {@code axis}. If {@code keep_dims} is true, the reduced dimensions are * retained with length 1. */ +@OpMetadata( + opType = ReduceAny.OP_NAME, + inputsClass = ReduceAny.Inputs.class +) @Operator public final class ReduceAny extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class ReduceAny extends RawOp implements Operand { private Output output; - private ReduceAny(Operation operation) { - super(operation); + public ReduceAny(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = ReduceAny.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMax.java index 913b043560d..1327ee68ba8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReduceMax.OP_NAME, + inputsClass = ReduceMax.Inputs.class +) @Operator public final class ReduceMax extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class ReduceMax extends RawOp implements Operand private Output output; - private ReduceMax(Operation operation) { - super(operation); + public ReduceMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = ReduceMax.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMin.java index bf69583091c..928d6affc56 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReduceMin.OP_NAME, + inputsClass = ReduceMin.Inputs.class +) @Operator public final class ReduceMin extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class ReduceMin extends RawOp implements Operand private Output output; - private ReduceMin(Operation operation) { - super(operation); + public ReduceMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = ReduceMin.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceProd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceProd.java index eb133e52ded..d4c9284eb4c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceProd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceProd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReduceProd.OP_NAME, + inputsClass = ReduceProd.Inputs.class +) @Operator public final class ReduceProd extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class ReduceProd extends RawOp implements Operand< private Output output; - private ReduceProd(Operation operation) { - super(operation); + public ReduceProd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = ReduceProd.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceSum.java index 45222cee07a..0057c41c1de 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReduceSum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReduceSum.OP_NAME, + inputsClass = ReduceSum.Inputs.class +) @Operator public final class ReduceSum extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class ReduceSum extends RawOp implements Operand output; - private ReduceSum(Operation operation) { - super(operation); + public ReduceSum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = ReduceSum.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java index aad1e84ea0d..987b8e95ef8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RefEnter.OP_NAME, + inputsClass = RefEnter.Inputs.class +) public final class RefEnter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RefEnter extends RawOp implements Operand private Output output; - private RefEnter(Operation operation) { - super(operation); + public RefEnter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -152,6 +158,9 @@ public Options parallelIterations(Long parallelIterations) { } } + @OpInputsMetadata( + outputsClass = RefEnter.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be made available to the child frame. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java index 42a11c140ed..1b7b349f507 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RefExit.OP_NAME, + inputsClass = RefExit.Inputs.class +) public final class RefExit extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RefExit extends RawOp implements Operand private Output output; - private RefExit(Operation operation) { - super(operation); + public RefExit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RefExit.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be made available to the parent frame. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java index b9623c181f1..20d8e137307 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RefIdentity.OP_NAME, + inputsClass = RefIdentity.Inputs.class +) public final class RefIdentity extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RefIdentity extends RawOp implements Operand private Output output; - private RefIdentity(Operation operation) { - super(operation); + public RefIdentity(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RefIdentity.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java index 911e7d845cf..c5e9183fb95 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RefMerge.OP_NAME, + inputsClass = RefMerge.Inputs.class +) public final class RefMerge extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -51,8 +57,8 @@ public final class RefMerge extends RawOp { private Output valueIndex; - private RefMerge(Operation operation) { - super(operation); + public RefMerge(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); valueIndex = operation.output(outputIdx++); @@ -93,6 +99,9 @@ public Output valueIndex() { return valueIndex; } + @OpInputsMetadata( + outputsClass = RefMerge.class + ) public static class Inputs extends RawOpInputs> { /** * The input tensors, exactly one of which will become available. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefNextIteration.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefNextIteration.java index 4a56e3f5394..2497ea77192 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefNextIteration.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefNextIteration.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RefNextIteration.OP_NAME, + inputsClass = RefNextIteration.Inputs.class +) @Operator public final class RefNextIteration extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class RefNextIteration extends RawOp implements Op private Output output; - private RefNextIteration(Operation operation) { - super(operation); + public RefNextIteration(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RefNextIteration.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be made available to the next iteration. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSelect.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSelect.java index 4e449068fdd..4b1a228e4c3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSelect.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSelect.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RefSelect.OP_NAME, + inputsClass = RefSelect.Inputs.class +) @Operator public final class RefSelect extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class RefSelect extends RawOp implements Operand output; - private RefSelect(Operation operation) { - super(operation); + public RefSelect(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RefSelect.class + ) public static class Inputs extends RawOpInputs> { /** * A scalar that determines the input that gets selected. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSwitch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSwitch.java index 47c7a63df80..34a38b08917 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSwitch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefSwitch.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -40,6 +42,10 @@ * * @param data type for {@code output_false} output */ +@OpMetadata( + opType = RefSwitch.OP_NAME, + inputsClass = RefSwitch.Inputs.class +) @Operator public final class RefSwitch extends RawOp { /** @@ -51,8 +57,8 @@ public final class RefSwitch extends RawOp { private Output outputTrue; - private RefSwitch(Operation operation) { - super(operation); + public RefSwitch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputFalse = operation.output(outputIdx++); outputTrue = operation.output(outputIdx++); @@ -96,6 +102,9 @@ public Output outputTrue() { return outputTrue; } + @OpInputsMetadata( + outputsClass = RefSwitch.class + ) public static class Inputs extends RawOpInputs> { /** * The ref tensor to be forwarded to the appropriate output. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RemoteCall.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RemoteCall.java index 4f53a7940a6..e2dc2c02a80 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RemoteCall.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RemoteCall.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ /** * Runs function {@code f} on a remote device indicated by {@code target}. */ +@OpMetadata( + opType = RemoteCall.OP_NAME, + inputsClass = RemoteCall.Inputs.class +) @Operator public final class RemoteCall extends RawOp implements Iterable> { /** @@ -49,8 +55,8 @@ public final class RemoteCall extends RawOp implements Iterable> private List> output; @SuppressWarnings("unchecked") - private RemoteCall(Operation operation) { - super(operation); + public RemoteCall(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -95,6 +101,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = RemoteCall.class + ) public static class Inputs extends RawOpInputs { /** * A fully specified device name where we want to run the function. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reshape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reshape.java index e412bf4ccc7..c6fa9bdcb01 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reshape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reshape.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -91,6 +93,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Reshape.OP_NAME, + inputsClass = Reshape.Inputs.class +) @Operator public final class Reshape extends RawOp implements Operand { /** @@ -100,8 +106,8 @@ public final class Reshape extends RawOp implements Operand private Output output; - private Reshape(Operation operation) { - super(operation); + public Reshape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Reshape.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceCountUpTo.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceCountUpTo.java index 40d808c3cbd..afcf81f3454 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceCountUpTo.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceCountUpTo.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ResourceCountUpTo.OP_NAME, + inputsClass = ResourceCountUpTo.Inputs.class +) @Operator public final class ResourceCountUpTo extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class ResourceCountUpTo extends RawOp implements private Output output; - private ResourceCountUpTo(Operation operation) { - super(operation); + public ResourceCountUpTo(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ResourceCountUpTo.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a scalar {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGather.java index 195677c0764..c583cafab53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGather.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -50,6 +52,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ResourceGather.OP_NAME, + inputsClass = ResourceGather.Inputs.class +) @Operator public final class ResourceGather extends RawOp implements Operand { /** @@ -59,8 +65,8 @@ public final class ResourceGather extends RawOp implements Oper private Output output; - private ResourceGather(Operation operation) { - super(operation); + public ResourceGather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -167,6 +173,9 @@ public Options validateIndices(Boolean validateIndices) { } } + @OpInputsMetadata( + outputsClass = ResourceGather.class + ) public static class Inputs extends RawOpInputs> { /** * The resource input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGatherNd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGatherNd.java index 42737e67583..3c5680dff27 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGatherNd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceGatherNd.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ResourceGatherNd.OP_NAME, + inputsClass = ResourceGatherNd.Inputs.class +) @Operator public final class ResourceGatherNd extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class ResourceGatherNd extends RawOp implements Op private Output output; - private ResourceGatherNd(Operation operation) { - super(operation); + public ResourceGatherNd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ResourceGatherNd.class + ) public static class Inputs extends RawOpInputs> { /** * The resource input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterAdd.java index 237c62609ed..d349b92d941 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterAdd.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * */ +@OpMetadata( + opType = ResourceScatterAdd.OP_NAME, + inputsClass = ResourceScatterAdd.Inputs.class +) @Operator public final class ResourceScatterAdd extends RawOp { /** @@ -58,8 +64,8 @@ public final class ResourceScatterAdd extends RawOp { */ public static final String OP_NAME = "ResourceScatterAdd"; - private ResourceScatterAdd(Operation operation) { - super(operation); + public ResourceScatterAdd(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static ResourceScatterAdd create(Scope scope, Operand re return new ResourceScatterAdd(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterAdd.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterDiv.java index 8568d28df4c..5f29312f25c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterDiv.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * */ +@OpMetadata( + opType = ResourceScatterDiv.OP_NAME, + inputsClass = ResourceScatterDiv.Inputs.class +) @Operator public final class ResourceScatterDiv extends RawOp { /** @@ -58,8 +64,8 @@ public final class ResourceScatterDiv extends RawOp { */ public static final String OP_NAME = "ResourceScatterDiv"; - private ResourceScatterDiv(Operation operation) { - super(operation); + public ResourceScatterDiv(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static ResourceScatterDiv create(Scope scope, Operand re return new ResourceScatterDiv(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterDiv.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMax.java index 68252db27f2..0773be58a8c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMax.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * */ +@OpMetadata( + opType = ResourceScatterMax.OP_NAME, + inputsClass = ResourceScatterMax.Inputs.class +) @Operator public final class ResourceScatterMax extends RawOp { /** @@ -58,8 +64,8 @@ public final class ResourceScatterMax extends RawOp { */ public static final String OP_NAME = "ResourceScatterMax"; - private ResourceScatterMax(Operation operation) { - super(operation); + public ResourceScatterMax(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static ResourceScatterMax create(Scope scope, Operand re return new ResourceScatterMax(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterMax.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMin.java index 1b8a9871b5d..408a3089249 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMin.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * */ +@OpMetadata( + opType = ResourceScatterMin.OP_NAME, + inputsClass = ResourceScatterMin.Inputs.class +) @Operator public final class ResourceScatterMin extends RawOp { /** @@ -58,8 +64,8 @@ public final class ResourceScatterMin extends RawOp { */ public static final String OP_NAME = "ResourceScatterMin"; - private ResourceScatterMin(Operation operation) { - super(operation); + public ResourceScatterMin(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static ResourceScatterMin create(Scope scope, Operand re return new ResourceScatterMin(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterMin.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMul.java index bcf7c629746..4bdb292741a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterMul.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * */ +@OpMetadata( + opType = ResourceScatterMul.OP_NAME, + inputsClass = ResourceScatterMul.Inputs.class +) @Operator public final class ResourceScatterMul extends RawOp { /** @@ -58,8 +64,8 @@ public final class ResourceScatterMul extends RawOp { */ public static final String OP_NAME = "ResourceScatterMul"; - private ResourceScatterMul(Operation operation) { - super(operation); + public ResourceScatterMul(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static ResourceScatterMul create(Scope scope, Operand re return new ResourceScatterMul(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterMul.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdAdd.java index f3a8d9746d6..f9856bdb9e9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdAdd.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -60,6 +62,10 @@ *

See {@code tf.scatter_nd} for more details about how to make updates to * slices. */ +@OpMetadata( + opType = ResourceScatterNdAdd.OP_NAME, + inputsClass = ResourceScatterNdAdd.Inputs.class +) @Operator public final class ResourceScatterNdAdd extends RawOp { /** @@ -67,8 +73,8 @@ public final class ResourceScatterNdAdd extends RawOp { */ public static final String OP_NAME = "ResourceScatterNdAdd"; - private ResourceScatterNdAdd(Operation operation) { - super(operation); + public ResourceScatterNdAdd(Operation operation) { + super(operation, OP_NAME); } /** @@ -137,6 +143,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceScatterNdAdd.class + ) public static class Inputs extends RawOpInputs { /** * A resource handle. Must be from a VarHandleOp. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMax.java index 0899d59f311..c16711b04d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMax.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * The ResourceScatterNdMax operation */ +@OpMetadata( + opType = ResourceScatterNdMax.OP_NAME, + inputsClass = ResourceScatterNdMax.Inputs.class +) @Operator public final class ResourceScatterNdMax extends RawOp { /** @@ -41,8 +47,8 @@ public final class ResourceScatterNdMax extends RawOp { */ public static final String OP_NAME = "ResourceScatterNdMax"; - private ResourceScatterNdMax(Operation operation) { - super(operation); + public ResourceScatterNdMax(Operation operation) { + super(operation, OP_NAME); } /** @@ -111,6 +117,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceScatterNdMax.class + ) public static class Inputs extends RawOpInputs { /** * A resource handle. Must be from a VarHandleOp. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMin.java index bf135c7094e..86ac1326e49 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdMin.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * The ResourceScatterNdMin operation */ +@OpMetadata( + opType = ResourceScatterNdMin.OP_NAME, + inputsClass = ResourceScatterNdMin.Inputs.class +) @Operator public final class ResourceScatterNdMin extends RawOp { /** @@ -41,8 +47,8 @@ public final class ResourceScatterNdMin extends RawOp { */ public static final String OP_NAME = "ResourceScatterNdMin"; - private ResourceScatterNdMin(Operation operation) { - super(operation); + public ResourceScatterNdMin(Operation operation) { + super(operation, OP_NAME); } /** @@ -111,6 +117,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceScatterNdMin.class + ) public static class Inputs extends RawOpInputs { /** * A resource handle. Must be from a VarHandleOp. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdSub.java index 5270a1e9bf9..054164d9f9d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdSub.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -60,6 +62,10 @@ *

See {@code tf.scatter_nd} for more details about how to make updates to * slices. */ +@OpMetadata( + opType = ResourceScatterNdSub.OP_NAME, + inputsClass = ResourceScatterNdSub.Inputs.class +) @Operator public final class ResourceScatterNdSub extends RawOp { /** @@ -67,8 +73,8 @@ public final class ResourceScatterNdSub extends RawOp { */ public static final String OP_NAME = "ResourceScatterNdSub"; - private ResourceScatterNdSub(Operation operation) { - super(operation); + public ResourceScatterNdSub(Operation operation) { + super(operation, OP_NAME); } /** @@ -137,6 +143,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceScatterNdSub.class + ) public static class Inputs extends RawOpInputs { /** * A resource handle. Must be from a VarHandleOp. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdUpdate.java index ebcfd808172..95d7f8b97be 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterNdUpdate.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -61,6 +63,10 @@ *

See {@code tf.scatter_nd} for more details about how to make updates to * slices. */ +@OpMetadata( + opType = ResourceScatterNdUpdate.OP_NAME, + inputsClass = ResourceScatterNdUpdate.Inputs.class +) @Operator public final class ResourceScatterNdUpdate extends RawOp { /** @@ -68,8 +74,8 @@ public final class ResourceScatterNdUpdate extends RawOp { */ public static final String OP_NAME = "ResourceScatterNdUpdate"; - private ResourceScatterNdUpdate(Operation operation) { - super(operation); + public ResourceScatterNdUpdate(Operation operation) { + super(operation, OP_NAME); } /** @@ -138,6 +144,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceScatterNdUpdate.class + ) public static class Inputs extends RawOpInputs { /** * A resource handle. Must be from a VarHandleOp. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterSub.java index 73050f58930..7ffb653fc5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterSub.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * */ +@OpMetadata( + opType = ResourceScatterSub.OP_NAME, + inputsClass = ResourceScatterSub.Inputs.class +) @Operator public final class ResourceScatterSub extends RawOp { /** @@ -58,8 +64,8 @@ public final class ResourceScatterSub extends RawOp { */ public static final String OP_NAME = "ResourceScatterSub"; - private ResourceScatterSub(Operation operation) { - super(operation); + public ResourceScatterSub(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static ResourceScatterSub create(Scope scope, Operand re return new ResourceScatterSub(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterSub.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterUpdate.java index ae7645b7d38..ce6757da55b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceScatterUpdate.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -45,6 +47,10 @@ * ref[indices[i, ..., j], ...] = updates[i, ..., j, ...] * */ +@OpMetadata( + opType = ResourceScatterUpdate.OP_NAME, + inputsClass = ResourceScatterUpdate.Inputs.class +) @Operator public final class ResourceScatterUpdate extends RawOp { /** @@ -52,8 +58,8 @@ public final class ResourceScatterUpdate extends RawOp { */ public static final String OP_NAME = "ResourceScatterUpdate"; - private ResourceScatterUpdate(Operation operation) { - super(operation); + public ResourceScatterUpdate(Operation operation) { + super(operation, OP_NAME); } /** @@ -77,6 +83,9 @@ public static ResourceScatterUpdate create(Scope scope, Operand return new ResourceScatterUpdate(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceScatterUpdate.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceStridedSliceAssign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceStridedSliceAssign.java index fa3670c11f9..5da15961a6a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceStridedSliceAssign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ResourceStridedSliceAssign.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ *

NOTE this op currently does not support broadcasting and so {@code value}'s * shape must be exactly the shape produced by the slice of {@code ref}. */ +@OpMetadata( + opType = ResourceStridedSliceAssign.OP_NAME, + inputsClass = ResourceStridedSliceAssign.Inputs.class +) @Operator public final class ResourceStridedSliceAssign extends RawOp { /** @@ -46,8 +52,8 @@ public final class ResourceStridedSliceAssign extends RawOp { */ public static final String OP_NAME = "ResourceStridedSliceAssign"; - private ResourceStridedSliceAssign(Operation operation) { - super(operation); + public ResourceStridedSliceAssign(Operation operation) { + super(operation, OP_NAME); } /** @@ -220,6 +226,9 @@ public Options shrinkAxisMask(Long shrinkAxisMask) { } } + @OpInputsMetadata( + outputsClass = ResourceStridedSliceAssign.class + ) public static class Inputs extends RawOpInputs { /** * The ref input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reverse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reverse.java index 6cb5d89fa7f..5bdcfc6b3dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reverse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Reverse.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -79,6 +81,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Reverse.OP_NAME, + inputsClass = Reverse.Inputs.class +) @Operator public final class Reverse extends RawOp implements Operand { /** @@ -88,8 +94,8 @@ public final class Reverse extends RawOp implements Operand private Output output; - private Reverse(Operation operation) { - super(operation); + public Reverse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Reverse.class + ) public static class Inputs extends RawOpInputs> { /** * Up to 8-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReverseSequence.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReverseSequence.java index 00bc24272c8..5fdc96725c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReverseSequence.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ReverseSequence.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -85,6 +87,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReverseSequence.OP_NAME, + inputsClass = ReverseSequence.Inputs.class +) @Operator public final class ReverseSequence extends RawOp implements Operand { /** @@ -94,8 +100,8 @@ public final class ReverseSequence extends RawOp implements Ope private Output output; - private ReverseSequence(Operation operation) { - super(operation); + public ReverseSequence(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -176,6 +182,9 @@ public Options batchDim(Long batchDim) { } } + @OpInputsMetadata( + outputsClass = ReverseSequence.class + ) public static class Inputs extends RawOpInputs> { /** * The input to reverse. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Roll.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Roll.java index 83983cf1983..70aaf92a7d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Roll.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Roll.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -55,6 +57,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Roll.OP_NAME, + inputsClass = Roll.Inputs.class +) @Operator public final class Roll extends RawOp implements Operand { /** @@ -64,8 +70,8 @@ public final class Roll extends RawOp implements Operand { private Output output; - private Roll(Operation operation) { - super(operation); + public Roll(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -114,6 +120,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Roll.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterAdd.java index 1f2bb33ba0d..60ee20a9add 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -56,6 +58,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterAdd.OP_NAME, + inputsClass = ScatterAdd.Inputs.class +) @Operator public final class ScatterAdd extends RawOp implements Operand { /** @@ -65,8 +71,8 @@ public final class ScatterAdd extends RawOp implements Operand< private Output outputRef; - private ScatterAdd(Operation operation) { - super(operation); + public ScatterAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -149,6 +155,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterAdd.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterDiv.java index da43641571e..18920cebab5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterDiv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterDiv.OP_NAME, + inputsClass = ScatterDiv.Inputs.class +) @Operator public final class ScatterDiv extends RawOp implements Operand { /** @@ -62,8 +68,8 @@ public final class ScatterDiv extends RawOp implements Operand< private Output outputRef; - private ScatterDiv(Operation operation) { - super(operation); + public ScatterDiv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterDiv.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMax.java index 136fa5c92e1..05563fcc7e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -55,6 +57,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterMax.OP_NAME, + inputsClass = ScatterMax.Inputs.class +) @Operator public final class ScatterMax extends RawOp implements Operand { /** @@ -64,8 +70,8 @@ public final class ScatterMax extends RawOp implements Operan private Output outputRef; - private ScatterMax(Operation operation) { - super(operation); + public ScatterMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterMax.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMin.java index 8b48aecb808..1e05925515a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -55,6 +57,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterMin.OP_NAME, + inputsClass = ScatterMin.Inputs.class +) @Operator public final class ScatterMin extends RawOp implements Operand { /** @@ -64,8 +70,8 @@ public final class ScatterMin extends RawOp implements Operan private Output outputRef; - private ScatterMin(Operation operation) { - super(operation); + public ScatterMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterMin.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMul.java index 8a979cf3a05..bc60bb4ecc9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterMul.OP_NAME, + inputsClass = ScatterMul.Inputs.class +) @Operator public final class ScatterMul extends RawOp implements Operand { /** @@ -62,8 +68,8 @@ public final class ScatterMul extends RawOp implements Operand< private Output outputRef; - private ScatterMul(Operation operation) { - super(operation); + public ScatterMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterMul.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNd.java index ae84b7cf6ae..24b251c016e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -105,6 +107,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ScatterNd.OP_NAME, + inputsClass = ScatterNd.Inputs.class +) @Operator public final class ScatterNd extends RawOp implements Operand { /** @@ -114,8 +120,8 @@ public final class ScatterNd extends RawOp implements Operand output; - private ScatterNd(Operation operation) { - super(operation); + public ScatterNd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -158,6 +164,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ScatterNd.class + ) public static class Inputs extends RawOpInputs> { /** * Index tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdAdd.java index 50d897a1941..9236cc60451 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -63,6 +65,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterNdAdd.OP_NAME, + inputsClass = ScatterNdAdd.Inputs.class +) @Operator public final class ScatterNdAdd extends RawOp implements Operand { /** @@ -72,8 +78,8 @@ public final class ScatterNdAdd extends RawOp implements Operan private Output outputRef; - private ScatterNdAdd(Operation operation) { - super(operation); + public ScatterNdAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -160,6 +166,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterNdAdd.class + ) public static class Inputs extends RawOpInputs> { /** * A mutable Tensor. Should be from a Variable node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java index 2783d423634..f0e0e82032d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterNdMax.OP_NAME, + inputsClass = ScatterNdMax.Inputs.class +) public final class ScatterNdMax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class ScatterNdMax extends RawOp implements Operan private Output outputRef; - private ScatterNdMax(Operation operation) { - super(operation); + public ScatterNdMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -132,6 +138,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterNdMax.class + ) public static class Inputs extends RawOpInputs> { /** * A mutable Tensor. Should be from a Variable node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java index 19415c2abd9..29c0bf8e43f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterNdMin.OP_NAME, + inputsClass = ScatterNdMin.Inputs.class +) public final class ScatterNdMin extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class ScatterNdMin extends RawOp implements Operan private Output outputRef; - private ScatterNdMin(Operation operation) { - super(operation); + public ScatterNdMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -132,6 +138,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterNdMin.class + ) public static class Inputs extends RawOpInputs> { /** * A mutable Tensor. Should be from a Variable node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdNonAliasingAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdNonAliasingAdd.java index d8e6cdb20de..95ceaa892eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdNonAliasingAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdNonAliasingAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -64,6 +66,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ScatterNdNonAliasingAdd.OP_NAME, + inputsClass = ScatterNdNonAliasingAdd.Inputs.class +) @Operator public final class ScatterNdNonAliasingAdd extends RawOp implements Operand { /** @@ -73,8 +79,8 @@ public final class ScatterNdNonAliasingAdd extends RawOp implem private Output output; - private ScatterNdNonAliasingAdd(Operation operation) { - super(operation); + public ScatterNdNonAliasingAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -118,6 +124,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ScatterNdNonAliasingAdd.class + ) public static class Inputs extends RawOpInputs> { /** * A Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdSub.java index 71100507d7d..da227ea2af8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdSub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -64,6 +66,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterNdSub.OP_NAME, + inputsClass = ScatterNdSub.Inputs.class +) @Operator public final class ScatterNdSub extends RawOp implements Operand { /** @@ -73,8 +79,8 @@ public final class ScatterNdSub extends RawOp implements Operan private Output outputRef; - private ScatterNdSub(Operation operation) { - super(operation); + public ScatterNdSub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -161,6 +167,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterNdSub.class + ) public static class Inputs extends RawOpInputs> { /** * A mutable Tensor. Should be from a Variable node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdUpdate.java index 9ee766d0c98..e4f24e90bc0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdUpdate.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -63,6 +65,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterNdUpdate.OP_NAME, + inputsClass = ScatterNdUpdate.Inputs.class +) @Operator public final class ScatterNdUpdate extends RawOp implements Operand { /** @@ -72,8 +78,8 @@ public final class ScatterNdUpdate extends RawOp implements Ope private Output outputRef; - private ScatterNdUpdate(Operation operation) { - super(operation); + public ScatterNdUpdate(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -160,6 +166,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterNdUpdate.class + ) public static class Inputs extends RawOpInputs> { /** * A mutable Tensor. Should be from a Variable node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterSub.java index 7ebbd6730ad..387b0708afe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterSub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -55,6 +57,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterSub.OP_NAME, + inputsClass = ScatterSub.Inputs.class +) @Operator public final class ScatterSub extends RawOp implements Operand { /** @@ -64,8 +70,8 @@ public final class ScatterSub extends RawOp implements Operand< private Output outputRef; - private ScatterSub(Operation operation) { - super(operation); + public ScatterSub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterSub.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterUpdate.java index cc9cebb665f..d31d14d25e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterUpdate.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -58,6 +60,10 @@ * * @param data type for {@code output_ref} output */ +@OpMetadata( + opType = ScatterUpdate.OP_NAME, + inputsClass = ScatterUpdate.Inputs.class +) @Operator public final class ScatterUpdate extends RawOp implements Operand { /** @@ -67,8 +73,8 @@ public final class ScatterUpdate extends RawOp implements Opera private Output outputRef; - private ScatterUpdate(Operation operation) { - super(operation); + public ScatterUpdate(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputRef = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ScatterUpdate.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a {@code Variable} node. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Select.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Select.java index b112acf4e8c..6a3f8bf1feb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Select.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Select.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Select.OP_NAME, + inputsClass = Select.Inputs.class +) @Operator public final class Select extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class Select extends RawOp implements Operand { private Output output; - private Select(Operation operation) { - super(operation); + public Select(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Select.class + ) public static class Inputs extends RawOpInputs> { /** * The condition input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java index 7822dbe0ab6..9cc0f7861de 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java @@ -26,20 +26,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Sends the named tensor from send_device to recv_device. */ +@OpMetadata( + opType = Send.OP_NAME, + inputsClass = Send.Inputs.class +) public final class Send extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "Send"; - private Send(Operation operation) { - super(operation); + public Send(Operation operation) { + super(operation, OP_NAME); } /** @@ -112,6 +118,9 @@ public Options clientTerminated(Boolean clientTerminated) { } } + @OpInputsMetadata( + outputsClass = Send.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to send. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetDiff1d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetDiff1d.java index b777c48ccbc..790802fa3a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetDiff1d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetDiff1d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -57,6 +59,10 @@ * * @param data type for {@code idx} output */ +@OpMetadata( + opType = SetDiff1d.OP_NAME, + inputsClass = SetDiff1d.Inputs.class +) @Operator public final class SetDiff1d extends RawOp { /** @@ -68,8 +74,8 @@ public final class SetDiff1d extends RawOp { private Output idx; - private SetDiff1d(Operation operation) { - super(operation); + public SetDiff1d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); idx = operation.output(outputIdx++); @@ -133,6 +139,9 @@ public Output idx() { return idx; } + @OpInputsMetadata( + outputsClass = SetDiff1d.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. Values to keep. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetSize.java index 8872c75a20f..8d6494d7c89 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SetSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ *

If {@code validate_indices} is {@code True}, this op validates the order and range of {@code set} * indices. */ +@OpMetadata( + opType = SetSize.OP_NAME, + inputsClass = SetSize.Inputs.class +) @Operator public final class SetSize extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class SetSize extends RawOp implements Operand { private Output output; - private SetSize(Operation operation) { - super(operation); + public SetSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -132,6 +138,9 @@ public Options validateIndices(Boolean validateIndices) { } } + @OpInputsMetadata( + outputsClass = SetSize.class + ) public static class Inputs extends RawOpInputs { /** * 2D {@code Tensor}, indices of a {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Shape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Shape.java index 836ace328b7..a4b24786234 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Shape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Shape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -45,6 +47,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Shape.OP_NAME, + inputsClass = Shape.Inputs.class +) @Operator public final class Shape extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class Shape extends RawOp implements Operand private Output output; - private Shape(Operation operation) { - super(operation); + public Shape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Shape.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ShapeN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ShapeN.java index 34a95d4de91..766e6e27880 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ShapeN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ShapeN.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ShapeN.OP_NAME, + inputsClass = ShapeN.Inputs.class +) @Operator public final class ShapeN extends RawOp implements Iterable> { /** @@ -52,8 +58,8 @@ public final class ShapeN extends RawOp implements Iterable> output; @SuppressWarnings("unchecked") - private ShapeN(Operation operation) { - super(operation); + public ShapeN(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList((Output[]) operation.outputList(outputIdx, outputLength)); @@ -109,6 +115,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = ShapeN.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Size.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Size.java index 2a0eed3e437..a15035267cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Size.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Size.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Size.OP_NAME, + inputsClass = Size.Inputs.class +) @Operator public final class Size extends RawOp implements Operand { /** @@ -55,8 +61,8 @@ public final class Size extends RawOp implements Operand { private Output output; - private Size(Operation operation) { - super(operation); + public Size(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -109,6 +115,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Size.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Skipgram.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Skipgram.java index f4ef6d93e92..950b9a6e164 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Skipgram.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Skipgram.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -34,6 +36,10 @@ /** * Parses a text file and creates a batch of examples. */ +@OpMetadata( + opType = Skipgram.OP_NAME, + inputsClass = Skipgram.Inputs.class +) @Operator public final class Skipgram extends RawOp { /** @@ -55,8 +61,8 @@ public final class Skipgram extends RawOp { private Output labels; - private Skipgram(Operation operation) { - super(operation); + public Skipgram(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; vocabWord = operation.output(outputIdx++); vocabFreq = operation.output(outputIdx++); @@ -243,6 +249,9 @@ public Options subsample(Float subsample) { } } + @OpInputsMetadata( + outputsClass = Skipgram.class + ) public static class Inputs extends RawOpInputs { /** * The corpus's text file name. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Slice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Slice.java index 733a500cca8..14e2bffac60 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Slice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Slice.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Slice.OP_NAME, + inputsClass = Slice.Inputs.class +) @Operator public final class Slice extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class Slice extends RawOp implements Operand { private Output output; - private Slice(Operation operation) { - super(operation); + public Slice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Slice.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Snapshot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Snapshot.java index 10ccbddac2f..e14bca38cb2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Snapshot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Snapshot.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Snapshot.OP_NAME, + inputsClass = Snapshot.Inputs.class +) @Operator public final class Snapshot extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class Snapshot extends RawOp implements Operand private Output output; - private Snapshot(Operation operation) { - super(operation); + public Snapshot(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Snapshot.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SpaceToBatchNd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SpaceToBatchNd.java index d7a39406724..740db3931cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SpaceToBatchNd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SpaceToBatchNd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -133,6 +135,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SpaceToBatchNd.OP_NAME, + inputsClass = SpaceToBatchNd.Inputs.class +) @Operator public final class SpaceToBatchNd extends RawOp implements Operand { /** @@ -142,8 +148,8 @@ public final class SpaceToBatchNd extends RawOp implements Oper private Output output; - private SpaceToBatchNd(Operation operation) { - super(operation); + public SpaceToBatchNd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -188,6 +194,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SpaceToBatchNd.class + ) public static class Inputs extends RawOpInputs> { /** * N-D with shape {@code input_shape = [batch] + spatial_shape + remaining_shape}, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Split.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Split.java index def301c84d3..85ff8a9bd2d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Split.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Split.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Split.OP_NAME, + inputsClass = Split.Inputs.class +) @Operator public final class Split extends RawOp implements Iterable> { /** @@ -49,8 +55,8 @@ public final class Split extends RawOp implements Iterable> output; @SuppressWarnings("unchecked") - private Split(Operation operation) { - super(operation); + public Split(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList((Output[]) operation.outputList(outputIdx, outputLength)); @@ -98,6 +104,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = Split.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D. The dimension along which to split. Must be in the range diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SplitV.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SplitV.java index 26d0cc172ce..77aad26aae6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SplitV.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SplitV.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SplitV.OP_NAME, + inputsClass = SplitV.Inputs.class +) @Operator public final class SplitV extends RawOp implements Iterable> { /** @@ -50,8 +56,8 @@ public final class SplitV extends RawOp implements Iterable> output; @SuppressWarnings("unchecked") - private SplitV(Operation operation) { - super(operation); + public SplitV(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList((Output[]) operation.outputList(outputIdx, outputLength)); @@ -102,6 +108,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = SplitV.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to split. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Squeeze.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Squeeze.java index d9c8956f5a1..d7a877da72c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Squeeze.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Squeeze.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -51,6 +53,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Squeeze.OP_NAME, + inputsClass = Squeeze.Inputs.class +) @Operator public final class Squeeze extends RawOp implements Operand { /** @@ -60,8 +66,8 @@ public final class Squeeze extends RawOp implements Operand private Output output; - private Squeeze(Operation operation) { - super(operation); + public Squeeze(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -171,6 +177,9 @@ public Options axis(Long... axis) { } } + @OpInputsMetadata( + outputsClass = Squeeze.class + ) public static class Inputs extends RawOpInputs> { /** * The {@code input} to squeeze. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stack.java index be4cdfe3921..a5378efd62e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stack.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stack.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -52,6 +54,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Stack.OP_NAME, + inputsClass = Stack.Inputs.class +) @Operator public final class Stack extends RawOp implements Operand { /** @@ -61,8 +67,8 @@ public final class Stack extends RawOp implements Operand { private Output output; - private Stack(Operation operation) { - super(operation); + public Stack(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = Stack.class + ) public static class Inputs extends RawOpInputs> { /** * Must be of same shape and type. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stage.java index adc05358314..0275d51c1d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stage.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Stage.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; @@ -35,6 +37,10 @@ * The basic functionality of this Op is similar to a queue with many * fewer capabilities and options. This Op is optimized for performance. */ +@OpMetadata( + opType = Stage.OP_NAME, + inputsClass = Stage.Inputs.class +) @Operator public final class Stage extends RawOp { /** @@ -42,8 +48,8 @@ public final class Stage extends RawOp { */ public static final String OP_NAME = "Stage"; - private Stage(Operation operation) { - super(operation); + public Stage(Operation operation) { + super(operation, OP_NAME); } /** @@ -186,6 +192,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = Stage.class + ) public static class Inputs extends RawOpInputs { /** * a list of tensors diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageClear.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageClear.java index 9431054d8a7..d549df79ab0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageClear.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageClear.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Op removes all elements in the underlying container. */ +@OpMetadata( + opType = StageClear.OP_NAME, + inputsClass = StageClear.Inputs.class +) @Operator public final class StageClear extends RawOp { /** @@ -41,8 +47,8 @@ public final class StageClear extends RawOp { */ public static final String OP_NAME = "StageClear"; - private StageClear(Operation operation) { - super(operation); + public StageClear(Operation operation) { + super(operation, OP_NAME); } /** @@ -179,6 +185,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = StageClear.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StagePeek.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StagePeek.java index dab381005af..cd8b4d3995a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StagePeek.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StagePeek.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * this op will block until it does. This Op is optimized for * performance. */ +@OpMetadata( + opType = StagePeek.OP_NAME, + inputsClass = StagePeek.Inputs.class +) @Operator public final class StagePeek extends RawOp implements Iterable> { /** @@ -51,8 +57,8 @@ public final class StagePeek extends RawOp implements Iterable> { private List> values; @SuppressWarnings("unchecked") - private StagePeek(Operation operation) { - super(operation); + public StagePeek(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int valuesLength = operation.outputListLength("values"); values = Arrays.asList(operation.outputList(outputIdx, valuesLength)); @@ -210,6 +216,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = StagePeek.class + ) public static class Inputs extends RawOpInputs { /** * The index input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageSize.java index 609a40599a3..f107766a24b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StageSize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ /** * Op returns the number of elements in the underlying container. */ +@OpMetadata( + opType = StageSize.OP_NAME, + inputsClass = StageSize.Inputs.class +) @Operator public final class StageSize extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class StageSize extends RawOp implements Operand { private Output output; - private StageSize(Operation operation) { - super(operation); + public StageSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -200,6 +206,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = StageSize.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulCase.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulCase.java index a040d5ccd26..a965157f255 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulCase.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulCase.java @@ -32,6 +32,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -58,6 +60,10 @@ * ``` * */ +@OpMetadata( + opType = StatefulCase.OP_NAME, + inputsClass = StatefulCase.Inputs.class +) @Operator public final class StatefulCase extends RawOp implements Case { /** @@ -68,8 +74,8 @@ public final class StatefulCase extends RawOp implements Case { private List> output; @SuppressWarnings("unchecked") - private StatefulCase(Operation operation) { - super(operation); + public StatefulCase(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -135,6 +141,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = StatefulCase.class + ) public static class Inputs extends RawOpInputs { /** * The branch selector, an int32 Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulIf.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulIf.java index 98dfb290406..61dbfffa2db 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulIf.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulIf.java @@ -32,6 +32,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ /** * output = cond ? then_branch(input) : else_branch(input) */ +@OpMetadata( + opType = StatefulIf.OP_NAME, + inputsClass = StatefulIf.Inputs.class +) @Operator public final class StatefulIf extends RawOp implements If { /** @@ -49,8 +55,8 @@ public final class StatefulIf extends RawOp implements If { private List> output; @SuppressWarnings("unchecked") - private StatefulIf(Operation operation) { - super(operation); + public StatefulIf(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -124,6 +130,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = StatefulIf.class + ) public static class Inputs extends RawOpInputs { /** *

diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulPartitionedCall.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulPartitionedCall.java
index 00572804fb3..15d25454f16 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulPartitionedCall.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulPartitionedCall.java
@@ -31,6 +31,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -38,6 +40,10 @@
 /**
  * returns {@code f(inputs)}, where {@code f}'s body is placed and partitioned.
  */
+@OpMetadata(
+    opType = StatefulPartitionedCall.OP_NAME,
+    inputsClass = StatefulPartitionedCall.Inputs.class
+)
 @Operator
 public final class StatefulPartitionedCall extends RawOp implements PartitionedCall {
   /**
@@ -48,8 +54,8 @@ public final class StatefulPartitionedCall extends RawOp implements PartitionedC
   private List> output;
 
   @SuppressWarnings("unchecked")
-  private StatefulPartitionedCall(Operation operation) {
-    super(operation);
+  public StatefulPartitionedCall(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     int outputLength = operation.outputListLength("output");
     output = Arrays.asList(operation.outputList(outputIdx, outputLength));
@@ -113,6 +119,9 @@ public Iterator> iterator() {
     return (Iterator) output.iterator();
   }
 
+  @OpInputsMetadata(
+      outputsClass = StatefulPartitionedCall.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * A list of input tensors.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulWhile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulWhile.java
index 0297bdd06ea..5021c2402d3 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulWhile.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatefulWhile.java
@@ -32,6 +32,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -39,6 +41,10 @@
 /**
  * output = input; While (Cond(output)) { output = Body(output) }
  */
+@OpMetadata(
+    opType = StatefulWhile.OP_NAME,
+    inputsClass = StatefulWhile.Inputs.class
+)
 @Operator
 public final class StatefulWhile extends RawOp implements While {
   /**
@@ -49,8 +55,8 @@ public final class StatefulWhile extends RawOp implements While {
   private List> output;
 
   @SuppressWarnings("unchecked")
-  private StatefulWhile(Operation operation) {
-    super(operation);
+  public StatefulWhile(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     int outputLength = operation.outputListLength("output");
     output = Arrays.asList(operation.outputList(outputIdx, outputLength));
@@ -121,6 +127,9 @@ public Iterator> iterator() {
     return (Iterator) output.iterator();
   }
 
+  @OpInputsMetadata(
+      outputsClass = StatefulWhile.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * A list of input tensors whose types are T.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java
index 3b9da5431de..6e0b000d3a3 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java
@@ -32,6 +32,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TInt32;
 import org.tensorflow.types.family.TType;
@@ -59,6 +61,10 @@
  * This should only be used when the none of branches has stateful ops.
  * 
*/ +@OpMetadata( + opType = StatelessCase.OP_NAME, + inputsClass = StatelessCase.Inputs.class +) public final class StatelessCase extends RawOp implements Case { /** * The name of this op, as known by TensorFlow core engine @@ -68,8 +74,8 @@ public final class StatelessCase extends RawOp implements Case { private List> output; @SuppressWarnings("unchecked") - private StatelessCase(Operation operation) { - super(operation); + public StatelessCase(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -135,6 +141,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = StatelessCase.class + ) public static class Inputs extends RawOpInputs { /** * The branch selector, an int32 Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessIf.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessIf.java index e4d4a60e693..2e9d3b4a12a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessIf.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessIf.java @@ -32,6 +32,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ /** * output = cond ? then_branch(input) : else_branch(input) */ +@OpMetadata( + opType = StatelessIf.OP_NAME, + inputsClass = StatelessIf.Inputs.class +) @Operator public final class StatelessIf extends RawOp implements If { /** @@ -49,8 +55,8 @@ public final class StatelessIf extends RawOp implements If { private List> output; @SuppressWarnings("unchecked") - private StatelessIf(Operation operation) { - super(operation); + public StatelessIf(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -127,6 +133,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = StatelessIf.class + ) public static class Inputs extends RawOpInputs { /** *
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessPartitionedCall.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessPartitionedCall.java
index f819677d6e7..6a0494a59d3 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessPartitionedCall.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessPartitionedCall.java
@@ -31,6 +31,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -41,6 +43,10 @@
  * within a single process. The kernel places and partitions a given function's
  * underlying graph, and executes each of the partitioned subgraphs as a function.
  */
+@OpMetadata(
+    opType = StatelessPartitionedCall.OP_NAME,
+    inputsClass = StatelessPartitionedCall.Inputs.class
+)
 @Operator
 public final class StatelessPartitionedCall extends RawOp implements PartitionedCall {
   /**
@@ -51,8 +57,8 @@ public final class StatelessPartitionedCall extends RawOp implements Partitioned
   private List> output;
 
   @SuppressWarnings("unchecked")
-  private StatelessPartitionedCall(Operation operation) {
-    super(operation);
+  public StatelessPartitionedCall(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     int outputLength = operation.outputListLength("output");
     output = Arrays.asList(operation.outputList(outputIdx, outputLength));
@@ -115,6 +121,9 @@ public Iterator> iterator() {
     return (Iterator) output.iterator();
   }
 
+  @OpInputsMetadata(
+      outputsClass = StatelessPartitionedCall.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * A list of input tensors.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessWhile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessWhile.java
index da5e3146ec7..cae67f4082e 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessWhile.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessWhile.java
@@ -32,6 +32,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -39,6 +41,10 @@
 /**
  * output = input; While (Cond(output)) { output = Body(output) }
  */
+@OpMetadata(
+    opType = StatelessWhile.OP_NAME,
+    inputsClass = StatelessWhile.Inputs.class
+)
 @Operator
 public final class StatelessWhile extends RawOp implements While {
   /**
@@ -49,8 +55,8 @@ public final class StatelessWhile extends RawOp implements While {
   private List> output;
 
   @SuppressWarnings("unchecked")
-  private StatelessWhile(Operation operation) {
-    super(operation);
+  public StatelessWhile(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     int outputLength = operation.outputListLength("output");
     output = Arrays.asList(operation.outputList(outputIdx, outputLength));
@@ -124,6 +130,9 @@ public Iterator> iterator() {
     return (Iterator) output.iterator();
   }
 
+  @OpInputsMetadata(
+      outputsClass = StatelessWhile.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * A list of input tensors whose types are T.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StopGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StopGradient.java
index 861c19e863d..5cedbd6fe57 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StopGradient.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StopGradient.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -86,6 +88,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = StopGradient.OP_NAME,
+    inputsClass = StopGradient.Inputs.class
+)
 @Operator
 public final class StopGradient extends RawOp implements Operand {
   /**
@@ -95,8 +101,8 @@ public final class StopGradient extends RawOp implements Operan
 
   private Output output;
 
-  private StopGradient(Operation operation) {
-    super(operation);
+  public StopGradient(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -132,6 +138,9 @@ public Output asOutput() {
     return output;
   }
 
+  @OpInputsMetadata(
+      outputsClass = StopGradient.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The input input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSlice.java
index 3eeabb0160b..bc77fdb4639 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSlice.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSlice.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -134,6 +136,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = StridedSlice.OP_NAME,
+    inputsClass = StridedSlice.Inputs.class
+)
 @Operator
 public final class StridedSlice extends RawOp implements Operand {
   /**
@@ -143,8 +149,8 @@ public final class StridedSlice extends RawOp implements Operan
 
   private Output output;
 
-  private StridedSlice(Operation operation) {
-    super(operation);
+  public StridedSlice(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -370,6 +376,9 @@ public Options shrinkAxisMask(Long shrinkAxisMask) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = StridedSlice.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The input input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceAssign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceAssign.java
index 644018b6e74..1ebe8c6fdc4 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceAssign.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceAssign.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -42,6 +44,10 @@
  *
  * @param  data type for {@code output_ref} output
  */
+@OpMetadata(
+    opType = StridedSliceAssign.OP_NAME,
+    inputsClass = StridedSliceAssign.Inputs.class
+)
 @Operator
 public final class StridedSliceAssign extends RawOp implements Operand {
   /**
@@ -51,8 +57,8 @@ public final class StridedSliceAssign extends RawOp implements
 
   private Output outputRef;
 
-  private StridedSliceAssign(Operation operation) {
-    super(operation);
+  public StridedSliceAssign(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     outputRef = operation.output(outputIdx++);
   }
@@ -242,6 +248,9 @@ public Options shrinkAxisMask(Long shrinkAxisMask) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = StridedSliceAssign.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The ref input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceGrad.java
index 07947acab0a..6588d96158d 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceGrad.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StridedSliceGrad.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -44,6 +46,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = StridedSliceGrad.OP_NAME,
+    inputsClass = StridedSliceGrad.Inputs.class
+)
 @Operator
 public final class StridedSliceGrad extends RawOp implements Operand {
   /**
@@ -53,8 +59,8 @@ public final class StridedSliceGrad extends RawOp implements Op
 
   private Output output;
 
-  private StridedSliceGrad(Operation operation) {
-    super(operation);
+  public StridedSliceGrad(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -244,6 +250,9 @@ public Options shrinkAxisMask(Long shrinkAxisMask) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = StridedSliceGrad.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The shape input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Sum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Sum.java
index 5df5a25d98e..579be3dc3ff 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Sum.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Sum.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TNumber;
@@ -41,6 +43,10 @@
  *
  * @param  data type for {@code output} output
  */
+@OpMetadata(
+    opType = Sum.OP_NAME,
+    inputsClass = Sum.Inputs.class
+)
 @Operator
 public final class Sum extends RawOp implements Operand {
   /**
@@ -50,8 +56,8 @@ public final class Sum extends RawOp implements Operand {
 
   private Output output;
 
-  private Sum(Operation operation) {
-    super(operation);
+  public Sum(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -130,6 +136,9 @@ public Options keepDims(Boolean keepDims) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = Sum.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The tensor to reduce.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SwitchCond.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SwitchCond.java
index 00f21eb34bc..26c17fc6187 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SwitchCond.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SwitchCond.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TBool;
@@ -40,6 +42,10 @@
  *
  * @param  data type for {@code output_false} output
  */
+@OpMetadata(
+    opType = SwitchCond.OP_NAME,
+    inputsClass = SwitchCond.Inputs.class
+)
 @Operator
 public final class SwitchCond extends RawOp {
   /**
@@ -51,8 +57,8 @@ public final class SwitchCond extends RawOp {
 
   private Output outputTrue;
 
-  private SwitchCond(Operation operation) {
-    super(operation);
+  public SwitchCond(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     outputFalse = operation.output(outputIdx++);
     outputTrue = operation.output(outputIdx++);
@@ -96,6 +102,9 @@ public Output outputTrue() {
     return outputTrue;
   }
 
+  @OpInputsMetadata(
+      outputsClass = SwitchCond.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The tensor to be forwarded to the appropriate output.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TemporaryVariable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TemporaryVariable.java
index a16282ea459..6e6c0d6b835 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TemporaryVariable.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TemporaryVariable.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.family.TType;
@@ -49,6 +51,10 @@
  *
  * @param  data type for {@code ref} output
  */
+@OpMetadata(
+    opType = TemporaryVariable.OP_NAME,
+    inputsClass = TemporaryVariable.Inputs.class
+)
 @Operator
 public final class TemporaryVariable extends RawOp implements Operand {
   /**
@@ -58,8 +64,8 @@ public final class TemporaryVariable extends RawOp implements O
 
   private Output ref;
 
-  private TemporaryVariable(Operation operation) {
-    super(operation);
+  public TemporaryVariable(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     ref = operation.output(outputIdx++);
   }
@@ -139,6 +145,9 @@ public Options varName(String varName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = TemporaryVariable.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The shape of the variable tensor.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArray.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArray.java
index 34ca0e49eab..63ed41619b2 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArray.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArray.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -39,6 +41,10 @@
  * An array of Tensors of given size.
  * Write data via Write and read via Read or Pack.
  */
+@OpMetadata(
+    opType = TensorArray.OP_NAME,
+    inputsClass = TensorArray.Inputs.class
+)
 @Operator
 public final class TensorArray extends RawOp {
   /**
@@ -51,8 +57,8 @@ public final class TensorArray extends RawOp {
   private Output flow;
 
   @SuppressWarnings("unchecked")
-  private TensorArray(Operation operation) {
-    super(operation);
+  public TensorArray(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     handle = operation.output(outputIdx++);
     flow = operation.output(outputIdx++);
@@ -263,6 +269,9 @@ public Options tensorArrayName(String tensorArrayName) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArray.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The size of the array.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayClose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayClose.java
index df1db0abdba..03b585921e4 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayClose.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayClose.java
@@ -26,6 +26,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.types.family.TType;
 
@@ -34,6 +36,10 @@
  * This enables the user to close and release the resource in the middle
  * of a step/run.
  */
+@OpMetadata(
+    opType = TensorArrayClose.OP_NAME,
+    inputsClass = TensorArrayClose.Inputs.class
+)
 @Operator
 public final class TensorArrayClose extends RawOp {
   /**
@@ -41,8 +47,8 @@ public final class TensorArrayClose extends RawOp {
    */
   public static final String OP_NAME = "TensorArrayCloseV3";
 
-  private TensorArrayClose(Operation operation) {
-    super(operation);
+  public TensorArrayClose(Operation operation) {
+    super(operation, OP_NAME);
   }
 
   /**
@@ -61,6 +67,9 @@ public static TensorArrayClose create(Scope scope, Operand hand
     return new TensorArrayClose(opBuilder.build());
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayClose.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The handle to a TensorArray (output of TensorArray or TensorArrayGrad).
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayConcat.java
index decac97c386..70212b57cf1 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayConcat.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayConcat.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -47,6 +49,10 @@
  *
  * @param  data type for {@code value} output
  */
+@OpMetadata(
+    opType = TensorArrayConcat.OP_NAME,
+    inputsClass = TensorArrayConcat.Inputs.class
+)
 @Operator
 public final class TensorArrayConcat extends RawOp {
   /**
@@ -58,8 +64,8 @@ public final class TensorArrayConcat extends RawOp {
 
   private Output lengths;
 
-  private TensorArrayConcat(Operation operation) {
-    super(operation);
+  public TensorArrayConcat(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     value = operation.output(outputIdx++);
     lengths = operation.output(outputIdx++);
@@ -154,6 +160,9 @@ public Options elementShapeExcept0(Shape elementShapeExcept0) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayConcat.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The handle to a TensorArray.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGather.java
index e21dd4eea21..2cd86b61c22 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGather.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGather.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -41,6 +43,10 @@
  *
  * @param  data type for {@code value} output
  */
+@OpMetadata(
+    opType = TensorArrayGather.OP_NAME,
+    inputsClass = TensorArrayGather.Inputs.class
+)
 @Operator
 public final class TensorArrayGather extends RawOp implements Operand {
   /**
@@ -50,8 +56,8 @@ public final class TensorArrayGather extends RawOp implements O
 
   private Output value;
 
-  private TensorArrayGather(Operation operation) {
-    super(operation);
+  public TensorArrayGather(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     value = operation.output(outputIdx++);
   }
@@ -139,6 +145,9 @@ public Options elementShape(Shape elementShape) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayGather.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The handle to a TensorArray.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGrad.java
index 2b42e5dfe78..adccb742ebf 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGrad.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGrad.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.types.TFloat32;
 import org.tensorflow.types.family.TType;
@@ -62,6 +64,10 @@
  * name when performing the creation / lookup, so that each separate gradient
  * calculation gets its own TensorArray accumulator.
  */
+@OpMetadata(
+    opType = TensorArrayGrad.OP_NAME,
+    inputsClass = TensorArrayGrad.Inputs.class
+)
 @Operator
 public final class TensorArrayGrad extends RawOp {
   /**
@@ -74,8 +80,8 @@ public final class TensorArrayGrad extends RawOp {
   private Output flowOut;
 
   @SuppressWarnings("unchecked")
-  private TensorArrayGrad(Operation operation) {
-    super(operation);
+  public TensorArrayGrad(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     gradHandle = operation.output(outputIdx++);
     flowOut = operation.output(outputIdx++);
@@ -121,6 +127,9 @@ public Output flowOut() {
     return flowOut;
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayGrad.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The handle to the forward TensorArray.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGradWithShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGradWithShape.java
index 163dec99fc4..68777024253 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGradWithShape.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayGradWithShape.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.types.TFloat32;
 import org.tensorflow.types.TInt32;
@@ -39,6 +41,10 @@
  * computed. This enables multiple gradients for the same TensorArray to be
  * calculated using the same accumulator.
  */
+@OpMetadata(
+    opType = TensorArrayGradWithShape.OP_NAME,
+    inputsClass = TensorArrayGradWithShape.Inputs.class
+)
 @Operator
 public final class TensorArrayGradWithShape extends RawOp {
   /**
@@ -51,8 +57,8 @@ public final class TensorArrayGradWithShape extends RawOp {
   private Output flowOut;
 
   @SuppressWarnings("unchecked")
-  private TensorArrayGradWithShape(Operation operation) {
-    super(operation);
+  public TensorArrayGradWithShape(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     gradHandle = operation.output(outputIdx++);
     flowOut = operation.output(outputIdx++);
@@ -102,6 +108,9 @@ public Output flowOut() {
     return flowOut;
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayGradWithShape.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The handle to the forward TensorArray.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayPack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayPack.java
index 41fe676090e..c277435d413 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayPack.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayPack.java
@@ -29,6 +29,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -40,6 +42,10 @@
  *
  * @param  data type for {@code value} output
  */
+@OpMetadata(
+    opType = TensorArrayPack.OP_NAME,
+    inputsClass = TensorArrayPack.Inputs.class
+)
 @Operator
 public final class TensorArrayPack extends RawOp implements Operand {
   /**
@@ -49,8 +55,8 @@ public final class TensorArrayPack extends RawOp implements Ope
 
   private Output value;
 
-  private TensorArrayPack(Operation operation) {
-    super(operation);
+  public TensorArrayPack(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     value = operation.output(outputIdx++);
   }
@@ -130,6 +136,9 @@ public Options elementShape(Shape elementShape) {
     }
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayPack.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The handle input
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayRead.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayRead.java
index e34c494fa0d..dc0f921e8bc 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayRead.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayRead.java
@@ -28,6 +28,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -39,6 +41,10 @@
  *
  * @param  data type for {@code value} output
  */
+@OpMetadata(
+    opType = TensorArrayRead.OP_NAME,
+    inputsClass = TensorArrayRead.Inputs.class
+)
 @Operator
 public final class TensorArrayRead extends RawOp implements Operand {
   /**
@@ -48,8 +54,8 @@ public final class TensorArrayRead extends RawOp implements Ope
 
   private Output value;
 
-  private TensorArrayRead(Operation operation) {
-    super(operation);
+  public TensorArrayRead(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     value = operation.output(outputIdx++);
   }
@@ -93,6 +99,9 @@ public Output asOutput() {
     return value;
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayRead.class
+  )
   public static class Inputs extends RawOpInputs> {
     /**
      * The handle to a TensorArray.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayScatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayScatter.java
index af15fe9187e..3bd2ec548dd 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayScatter.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayScatter.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -37,6 +39,10 @@
  * Scatter the data from the input value into specific TensorArray elements.
  * {@code indices} must be a vector, its length must match the first dim of {@code value}.
  */
+@OpMetadata(
+    opType = TensorArrayScatter.OP_NAME,
+    inputsClass = TensorArrayScatter.Inputs.class
+)
 @Operator
 public final class TensorArrayScatter extends RawOp implements Operand {
   /**
@@ -46,8 +52,8 @@ public final class TensorArrayScatter extends RawOp implements Operand
 
   private Output flowOut;
 
-  private TensorArrayScatter(Operation operation) {
-    super(operation);
+  public TensorArrayScatter(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     flowOut = operation.output(outputIdx++);
   }
@@ -89,6 +95,9 @@ public Output asOutput() {
     return flowOut;
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArrayScatter.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The handle to a TensorArray.
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySize.java
index 6bfc5e291b1..82fb8c2be4c 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySize.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySize.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.types.TFloat32;
 import org.tensorflow.types.TInt32;
@@ -35,6 +37,10 @@
 /**
  * Get the current size of the TensorArray.
  */
+@OpMetadata(
+    opType = TensorArraySize.OP_NAME,
+    inputsClass = TensorArraySize.Inputs.class
+)
 @Operator
 public final class TensorArraySize extends RawOp implements Operand {
   /**
@@ -44,8 +50,8 @@ public final class TensorArraySize extends RawOp implements Operand {
 
   private Output output;
 
-  private TensorArraySize(Operation operation) {
-    super(operation);
+  public TensorArraySize(Operation operation) {
+    super(operation, OP_NAME);
     int outputIdx = 0;
     output = operation.output(outputIdx++);
   }
@@ -83,6 +89,9 @@ public Output asOutput() {
     return output;
   }
 
+  @OpInputsMetadata(
+      outputsClass = TensorArraySize.class
+  )
   public static class Inputs extends RawOpInputs {
     /**
      * The handle to a TensorArray (output of TensorArray or TensorArrayGrad).
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySplit.java
index 3ce1424ae9c..bc0818bbc75 100644
--- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySplit.java
+++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArraySplit.java
@@ -27,6 +27,8 @@
 import org.tensorflow.op.RawOpInputs;
 import org.tensorflow.op.Scope;
 import org.tensorflow.op.annotation.Endpoint;
+import org.tensorflow.op.annotation.OpInputsMetadata;
+import org.tensorflow.op.annotation.OpMetadata;
 import org.tensorflow.op.annotation.Operator;
 import org.tensorflow.proto.framework.DataType;
 import org.tensorflow.types.TFloat32;
@@ -45,6 +47,10 @@
  * 

and having size *

{@code nt x d0 x d1 x ...} */ +@OpMetadata( + opType = TensorArraySplit.OP_NAME, + inputsClass = TensorArraySplit.Inputs.class +) @Operator public final class TensorArraySplit extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class TensorArraySplit extends RawOp implements Operand { private Output flowOut; - private TensorArraySplit(Operation operation) { - super(operation); + public TensorArraySplit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; flowOut = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return flowOut; } + @OpInputsMetadata( + outputsClass = TensorArraySplit.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a TensorArray. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayUnpack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayUnpack.java index ce65c26492a..fb7040c25b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayUnpack.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayUnpack.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -36,6 +38,10 @@ /** * The TensorArrayUnpack operation */ +@OpMetadata( + opType = TensorArrayUnpack.OP_NAME, + inputsClass = TensorArrayUnpack.Inputs.class +) @Operator public final class TensorArrayUnpack extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class TensorArrayUnpack extends RawOp implements Operand private Output flowOut; - private TensorArrayUnpack(Operation operation) { - super(operation); + public TensorArrayUnpack(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; flowOut = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return flowOut; } + @OpInputsMetadata( + outputsClass = TensorArrayUnpack.class + ) public static class Inputs extends RawOpInputs { /** * The handle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayWrite.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayWrite.java index e44e81a294d..8e18aa00dea 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayWrite.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorArrayWrite.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -36,6 +38,10 @@ /** * Push an element onto the tensor_array. */ +@OpMetadata( + opType = TensorArrayWrite.OP_NAME, + inputsClass = TensorArrayWrite.Inputs.class +) @Operator public final class TensorArrayWrite extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class TensorArrayWrite extends RawOp implements Operand { private Output flowOut; - private TensorArrayWrite(Operation operation) { - super(operation); + public TensorArrayWrite(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; flowOut = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return flowOut; } + @OpInputsMetadata( + outputsClass = TensorArrayWrite.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a TensorArray. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcat.java index 7b77b4870be..13766f6df52 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcat.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = TensorListConcat.OP_NAME, + inputsClass = TensorListConcat.Inputs.class +) @Operator public final class TensorListConcat extends RawOp { /** @@ -60,8 +66,8 @@ public final class TensorListConcat extends RawOp { private Output lengths; - private TensorListConcat(Operation operation) { - super(operation); + public TensorListConcat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tensor = operation.output(outputIdx++); lengths = operation.output(outputIdx++); @@ -110,6 +116,9 @@ public Output lengths() { return lengths; } + @OpInputsMetadata( + outputsClass = TensorListConcat.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcatLists.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcatLists.java index 1d1fc91e00d..617bec84468 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcatLists.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListConcatLists.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ /** * The TensorListConcatLists operation */ +@OpMetadata( + opType = TensorListConcatLists.OP_NAME, + inputsClass = TensorListConcatLists.Inputs.class +) @Operator public final class TensorListConcatLists extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class TensorListConcatLists extends RawOp implements Operand private Output output; @SuppressWarnings("unchecked") - private TensorListConcatLists(Operation operation) { - super(operation); + public TensorListConcatLists(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = TensorListConcatLists.class + ) public static class Inputs extends RawOpInputs { /** * The inputA input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListElementShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListElementShape.java index dbc4eb91ec5..472d8072db3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListElementShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListElementShape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code element_shape} output */ +@OpMetadata( + opType = TensorListElementShape.OP_NAME, + inputsClass = TensorListElementShape.Inputs.class +) @Operator public final class TensorListElementShape extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class TensorListElementShape extends RawOp imple private Output elementShape; - private TensorListElementShape(Operation operation) { - super(operation); + public TensorListElementShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; elementShape = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return elementShape; } + @OpInputsMetadata( + outputsClass = TensorListElementShape.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListFromTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListFromTensor.java index f193b8c7f13..cb090a2f34a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListFromTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListFromTensor.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ *

tensor: The input tensor. * output_handle: The list. */ +@OpMetadata( + opType = TensorListFromTensor.OP_NAME, + inputsClass = TensorListFromTensor.Inputs.class +) @Operator public final class TensorListFromTensor extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class TensorListFromTensor extends RawOp implements Operand private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListFromTensor(Operation operation) { - super(operation); + public TensorListFromTensor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListFromTensor.class + ) public static class Inputs extends RawOpInputs { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGather.java index d4e08702d4d..cd0732e2bf4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGather.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -43,6 +45,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = TensorListGather.OP_NAME, + inputsClass = TensorListGather.Inputs.class +) @Operator public final class TensorListGather extends RawOp implements Operand { /** @@ -52,8 +58,8 @@ public final class TensorListGather extends RawOp implements Op private Output values; - private TensorListGather(Operation operation) { - super(operation); + public TensorListGather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; values = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return values; } + @OpInputsMetadata( + outputsClass = TensorListGather.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGetItem.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGetItem.java index c1841854a1e..f39b65da026 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGetItem.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListGetItem.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code item} output */ +@OpMetadata( + opType = TensorListGetItem.OP_NAME, + inputsClass = TensorListGetItem.Inputs.class +) @Operator public final class TensorListGetItem extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class TensorListGetItem extends RawOp implements O private Output item; - private TensorListGetItem(Operation operation) { - super(operation); + public TensorListGetItem(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; item = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return item; } + @OpInputsMetadata( + outputsClass = TensorListGetItem.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListLength.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListLength.java index d3b70223fe1..390b4a4b5ff 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListLength.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListLength.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * input_handle: the input list * length: the number of tensors in the list */ +@OpMetadata( + opType = TensorListLength.OP_NAME, + inputsClass = TensorListLength.Inputs.class +) @Operator public final class TensorListLength extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class TensorListLength extends RawOp implements Operand { private Output length; - private TensorListLength(Operation operation) { - super(operation); + public TensorListLength(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; length = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return length; } + @OpInputsMetadata( + outputsClass = TensorListLength.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPopBack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPopBack.java index 7656c8f71b8..8d89bb811e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPopBack.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPopBack.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -43,6 +45,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = TensorListPopBack.OP_NAME, + inputsClass = TensorListPopBack.Inputs.class +) @Operator public final class TensorListPopBack extends RawOp { /** @@ -55,8 +61,8 @@ public final class TensorListPopBack extends RawOp { private Output tensor; @SuppressWarnings("unchecked") - private TensorListPopBack(Operation operation) { - super(operation); + public TensorListPopBack(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); tensor = operation.output(outputIdx++); @@ -102,6 +108,9 @@ public Output tensor() { return tensor; } + @OpInputsMetadata( + outputsClass = TensorListPopBack.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBack.java index d5d4b471c56..4349d023f08 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBack.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBack.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * element_dtype: the type of elements in the list. * element_shape: a shape compatible with that of elements in the list. */ +@OpMetadata( + opType = TensorListPushBack.OP_NAME, + inputsClass = TensorListPushBack.Inputs.class +) @Operator public final class TensorListPushBack extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class TensorListPushBack extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListPushBack(Operation operation) { - super(operation); + public TensorListPushBack(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListPushBack.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBackBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBackBatch.java index 694cc698ca3..1e011f256ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBackBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListPushBackBatch.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * The TensorListPushBackBatch operation */ +@OpMetadata( + opType = TensorListPushBackBatch.OP_NAME, + inputsClass = TensorListPushBackBatch.Inputs.class +) @Operator public final class TensorListPushBackBatch extends RawOp implements Operand { /** @@ -44,8 +50,8 @@ public final class TensorListPushBackBatch extends RawOp implements Operand outputHandles; @SuppressWarnings("unchecked") - private TensorListPushBackBatch(Operation operation) { - super(operation); + public TensorListPushBackBatch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandles = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return (Output) outputHandles; } + @OpInputsMetadata( + outputsClass = TensorListPushBackBatch.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandles input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListReserve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListReserve.java index b539e5114a7..ff029d62d9d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListReserve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListReserve.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * handle: the output list * element_dtype: the desired type of elements in the list. */ +@OpMetadata( + opType = TensorListReserve.OP_NAME, + inputsClass = TensorListReserve.Inputs.class +) @Operator public final class TensorListReserve extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class TensorListReserve extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TensorListReserve(Operation operation) { - super(operation); + public TensorListReserve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TensorListReserve.class + ) public static class Inputs extends RawOpInputs { /** * The elementShape input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListResize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListResize.java index 445f0177118..489e619b61a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListResize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListResize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * input_handle: the input list * size: size of the output list */ +@OpMetadata( + opType = TensorListResize.OP_NAME, + inputsClass = TensorListResize.Inputs.class +) @Operator public final class TensorListResize extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class TensorListResize extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListResize(Operation operation) { - super(operation); + public TensorListResize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListResize.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatter.java index c321d7f227a..4d4098ad8b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatter.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * the largest index in indices. * output_handle: The TensorList. */ +@OpMetadata( + opType = TensorListScatter.OP_NAME, + inputsClass = TensorListScatter.Inputs.class +) @Operator public final class TensorListScatter extends RawOp implements Operand { /** @@ -56,8 +62,8 @@ public final class TensorListScatter extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListScatter(Operation operation) { - super(operation); + public TensorListScatter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListScatter.class + ) public static class Inputs extends RawOpInputs { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatterIntoExistingList.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatterIntoExistingList.java index 15c38200e5d..5dc8e59b2e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatterIntoExistingList.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListScatterIntoExistingList.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * indices: The indices used to index into the list. * output_handle: The TensorList. */ +@OpMetadata( + opType = TensorListScatterIntoExistingList.OP_NAME, + inputsClass = TensorListScatterIntoExistingList.Inputs.class +) @Operator public final class TensorListScatterIntoExistingList extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class TensorListScatterIntoExistingList extends RawOp implements Op private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListScatterIntoExistingList(Operation operation) { - super(operation); + public TensorListScatterIntoExistingList(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListScatterIntoExistingList.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSetItem.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSetItem.java index d8010760cdb..8615c3073e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSetItem.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSetItem.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -35,6 +37,10 @@ /** * The TensorListSetItem operation */ +@OpMetadata( + opType = TensorListSetItem.OP_NAME, + inputsClass = TensorListSetItem.Inputs.class +) @Operator public final class TensorListSetItem extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class TensorListSetItem extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListSetItem(Operation operation) { - super(operation); + public TensorListSetItem(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListSetItem.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSplit.java index 27121f34aff..aa2903b13f3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSplit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListSplit.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -42,6 +44,10 @@ * lengths: Vector of sizes of the 0th dimension of tensors in the list. * output_handle: The list. */ +@OpMetadata( + opType = TensorListSplit.OP_NAME, + inputsClass = TensorListSplit.Inputs.class +) @Operator public final class TensorListSplit extends RawOp implements Operand { /** @@ -52,8 +58,8 @@ public final class TensorListSplit extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorListSplit(Operation operation) { - super(operation); + public TensorListSplit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorListSplit.class + ) public static class Inputs extends RawOpInputs { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListStack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListStack.java index 1f0a2364c20..eb2a1e197e9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListStack.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorListStack.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = TensorListStack.OP_NAME, + inputsClass = TensorListStack.Inputs.class +) @Operator public final class TensorListStack extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class TensorListStack extends RawOp implements Ope private Output tensor; - private TensorListStack(Operation operation) { - super(operation); + public TensorListStack(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tensor = operation.output(outputIdx++); } @@ -133,6 +139,9 @@ public Options numElements(Long numElements) { } } + @OpInputsMetadata( + outputsClass = TensorListStack.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapErase.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapErase.java index 93d15ad6fcf..d4271c4918e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapErase.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapErase.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * output_handle: the map with value from given key removed * key: the key of the value to be erased */ +@OpMetadata( + opType = TensorMapErase.OP_NAME, + inputsClass = TensorMapErase.Inputs.class +) @Operator public final class TensorMapErase extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class TensorMapErase extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorMapErase(Operation operation) { - super(operation); + public TensorMapErase(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorMapErase.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapHasKey.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapHasKey.java index 1d4659c3e24..d8807e475e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapHasKey.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapHasKey.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -38,6 +40,10 @@ * key: the key to check * has_key: whether the key is already in the map or not */ +@OpMetadata( + opType = TensorMapHasKey.OP_NAME, + inputsClass = TensorMapHasKey.Inputs.class +) @Operator public final class TensorMapHasKey extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class TensorMapHasKey extends RawOp implements Operand { private Output hasKey; - private TensorMapHasKey(Operation operation) { - super(operation); + public TensorMapHasKey(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; hasKey = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return hasKey; } + @OpInputsMetadata( + outputsClass = TensorMapHasKey.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapInsert.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapInsert.java index 7c4c494fe22..cc08f9340c8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapInsert.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapInsert.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * key: the key to be inserted * value: the value to be inserted */ +@OpMetadata( + opType = TensorMapInsert.OP_NAME, + inputsClass = TensorMapInsert.Inputs.class +) @Operator public final class TensorMapInsert extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class TensorMapInsert extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private TensorMapInsert(Operation operation) { - super(operation); + public TensorMapInsert(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = TensorMapInsert.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapLookup.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapLookup.java index 203fd5ea752..49765bdb9d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapLookup.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapLookup.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code value} output */ +@OpMetadata( + opType = TensorMapLookup.OP_NAME, + inputsClass = TensorMapLookup.Inputs.class +) @Operator public final class TensorMapLookup extends RawOp implements Operand { /** @@ -49,8 +55,8 @@ public final class TensorMapLookup extends RawOp implements Ope private Output value; - private TensorMapLookup(Operation operation) { - super(operation); + public TensorMapLookup(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; value = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return value; } + @OpInputsMetadata( + outputsClass = TensorMapLookup.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapSize.java index 8efc6ef987b..34af3944003 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * input_handle: the input map * size: the number of tensors in the map */ +@OpMetadata( + opType = TensorMapSize.OP_NAME, + inputsClass = TensorMapSize.Inputs.class +) @Operator public final class TensorMapSize extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class TensorMapSize extends RawOp implements Operand { private Output output; - private TensorMapSize(Operation operation) { - super(operation); + public TensorMapSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorMapSize.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapStackKeys.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapStackKeys.java index 3badf5126d8..0e7c23f2ce5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapStackKeys.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorMapStackKeys.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code keys} output */ +@OpMetadata( + opType = TensorMapStackKeys.OP_NAME, + inputsClass = TensorMapStackKeys.Inputs.class +) @Operator public final class TensorMapStackKeys extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class TensorMapStackKeys extends RawOp implements private Output keys; - private TensorMapStackKeys(Operation operation) { - super(operation); + public TensorMapStackKeys(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; keys = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return keys; } + @OpInputsMetadata( + outputsClass = TensorMapStackKeys.class + ) public static class Inputs extends RawOpInputs> { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdAdd.java index 2e676d0c7f2..29db1a5a709 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -93,6 +95,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorScatterNdAdd.OP_NAME, + inputsClass = TensorScatterNdAdd.Inputs.class +) @Operator public final class TensorScatterNdAdd extends RawOp implements Operand { /** @@ -102,8 +108,8 @@ public final class TensorScatterNdAdd extends RawOp implements private Output output; - private TensorScatterNdAdd(Operation operation) { - super(operation); + public TensorScatterNdAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorScatterNdAdd.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor to copy/update. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMax.java index 8693ee3efd1..07de1b38c81 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorScatterNdMax.OP_NAME, + inputsClass = TensorScatterNdMax.Inputs.class +) @Operator public final class TensorScatterNdMax extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class TensorScatterNdMax extends RawOp implements private Output output; - private TensorScatterNdMax(Operation operation) { - super(operation); + public TensorScatterNdMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorScatterNdMax.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor to update. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMin.java index 6826e3c035d..35287015c16 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorScatterNdMin.OP_NAME, + inputsClass = TensorScatterNdMin.Inputs.class +) @Operator public final class TensorScatterNdMin extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class TensorScatterNdMin extends RawOp implements private Output output; - private TensorScatterNdMin(Operation operation) { - super(operation); + public TensorScatterNdMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorScatterNdMin.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor to update. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdSub.java index a03f78e1f59..3188033d09c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdSub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -92,6 +94,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorScatterNdSub.OP_NAME, + inputsClass = TensorScatterNdSub.Inputs.class +) @Operator public final class TensorScatterNdSub extends RawOp implements Operand { /** @@ -101,8 +107,8 @@ public final class TensorScatterNdSub extends RawOp implements private Output output; - private TensorScatterNdSub(Operation operation) { - super(operation); + public TensorScatterNdSub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -143,6 +149,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorScatterNdSub.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor to copy/update. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdUpdate.java index 3cfec508b4b..d71f06861eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorScatterNdUpdate.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -66,6 +68,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorScatterNdUpdate.OP_NAME, + inputsClass = TensorScatterNdUpdate.Inputs.class +) @Operator public final class TensorScatterNdUpdate extends RawOp implements Operand { /** @@ -75,8 +81,8 @@ public final class TensorScatterNdUpdate extends RawOp implemen private Output output; - private TensorScatterNdUpdate(Operation operation) { - super(operation); + public TensorScatterNdUpdate(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -118,6 +124,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorScatterNdUpdate.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor to copy/update. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorStridedSliceUpdate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorStridedSliceUpdate.java index f2a68cb155a..8e95464edb1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorStridedSliceUpdate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TensorStridedSliceUpdate.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorStridedSliceUpdate.OP_NAME, + inputsClass = TensorStridedSliceUpdate.Inputs.class +) @Operator public final class TensorStridedSliceUpdate extends RawOp implements Operand { /** @@ -51,8 +57,8 @@ public final class TensorStridedSliceUpdate extends RawOp imple private Output output; - private TensorStridedSliceUpdate(Operation operation) { - super(operation); + public TensorStridedSliceUpdate(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -242,6 +248,9 @@ public Options shrinkAxisMask(Long shrinkAxisMask) { } } + @OpInputsMetadata( + outputsClass = TensorStridedSliceUpdate.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Tile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Tile.java index 4dbd1ef3a64..a13733f9c51 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Tile.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Tile.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -68,6 +70,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Tile.OP_NAME, + inputsClass = Tile.Inputs.class +) @Operator public final class Tile extends RawOp implements Operand { /** @@ -77,8 +83,8 @@ public final class Tile extends RawOp implements Operand { private Output output; - private Tile(Operation operation) { - super(operation); + public Tile(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -117,6 +123,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Tile.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D or higher. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Timestamp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Timestamp.java index e77de2d52ee..a19809b89a8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Timestamp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Timestamp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat64; @@ -36,6 +38,10 @@ *

Note: the timestamp is computed when the op is executed, not when it is added * to the graph. */ +@OpMetadata( + opType = Timestamp.OP_NAME, + inputsClass = Timestamp.Inputs.class +) @Operator public final class Timestamp extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class Timestamp extends RawOp implements Operand { private Output ts; - private Timestamp(Operation operation) { - super(operation); + public Timestamp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; ts = operation.output(outputIdx++); } @@ -79,6 +85,9 @@ public Output asOutput() { return ts; } + @OpInputsMetadata( + outputsClass = Timestamp.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new Timestamp(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKUnique.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKUnique.java index 081529475b0..0d9a2444692 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKUnique.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKUnique.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * padding value will be returned. The semantics are not the same as * kth_order_statistic. */ +@OpMetadata( + opType = TopKUnique.OP_NAME, + inputsClass = TopKUnique.Inputs.class +) @Operator public final class TopKUnique extends RawOp { /** @@ -57,8 +63,8 @@ public final class TopKUnique extends RawOp { private Output topkIndices; - private TopKUnique(Operation operation) { - super(operation); + public TopKUnique(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; topk = operation.output(outputIdx++); topkIndices = operation.output(outputIdx++); @@ -100,6 +106,9 @@ public Output topkIndices() { return topkIndices; } + @OpInputsMetadata( + outputsClass = TopKUnique.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKWithUnique.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKWithUnique.java index 856b03835ec..997bff2858d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKWithUnique.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/TopKWithUnique.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * of K and the input size. NaNs are never returned. Subnormal numbers are flushed * to zero. */ +@OpMetadata( + opType = TopKWithUnique.OP_NAME, + inputsClass = TopKWithUnique.Inputs.class +) @Operator public final class TopKWithUnique extends RawOp { /** @@ -50,8 +56,8 @@ public final class TopKWithUnique extends RawOp { private Output topkIndices; - private TopKWithUnique(Operation operation) { - super(operation); + public TopKWithUnique(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; topk = operation.output(outputIdx++); topkIndices = operation.output(outputIdx++); @@ -93,6 +99,9 @@ public Output topkIndices() { return topkIndices; } + @OpInputsMetadata( + outputsClass = TopKWithUnique.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unbatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unbatch.java index f39e219d59a..2ee6f6d8edd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unbatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unbatch.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -54,6 +56,10 @@ * * @param data type for {@code unbatched_tensor} output */ +@OpMetadata( + opType = Unbatch.OP_NAME, + inputsClass = Unbatch.Inputs.class +) @Operator public final class Unbatch extends RawOp implements Operand { /** @@ -63,8 +69,8 @@ public final class Unbatch extends RawOp implements Operand private Output unbatchedTensor; - private Unbatch(Operation operation) { - super(operation); + public Unbatch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; unbatchedTensor = operation.output(outputIdx++); } @@ -172,6 +178,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = Unbatch.class + ) public static class Inputs extends RawOpInputs> { /** * The batchedTensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnbatchGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnbatchGrad.java index 71687f10ad6..c81c07ee827 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnbatchGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnbatchGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -50,6 +52,10 @@ * * @param data type for {@code batched_grad} output */ +@OpMetadata( + opType = UnbatchGrad.OP_NAME, + inputsClass = UnbatchGrad.Inputs.class +) @Operator public final class UnbatchGrad extends RawOp implements Operand { /** @@ -59,8 +65,8 @@ public final class UnbatchGrad extends RawOp implements Operand private Output batchedGrad; - private UnbatchGrad(Operation operation) { - super(operation); + public UnbatchGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; batchedGrad = operation.output(outputIdx++); } @@ -168,6 +174,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = UnbatchGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The originalInput input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unique.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unique.java index e08cc3c6309..3b6be95c9a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unique.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unique.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -77,6 +79,10 @@ * * @param data type for {@code idx} output */ +@OpMetadata( + opType = Unique.OP_NAME, + inputsClass = Unique.Inputs.class +) @Operator public final class Unique extends RawOp { /** @@ -88,8 +94,8 @@ public final class Unique extends RawOp { private Output idx; - private Unique(Operation operation) { - super(operation); + public Unique(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); idx = operation.output(outputIdx++); @@ -156,6 +162,9 @@ public Output idx() { return idx; } + @OpInputsMetadata( + outputsClass = Unique.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniqueWithCounts.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniqueWithCounts.java index 0fa3499fab6..3b3e441e50d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniqueWithCounts.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniqueWithCounts.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -81,6 +83,10 @@ * * @param data type for {@code idx} output */ +@OpMetadata( + opType = UniqueWithCounts.OP_NAME, + inputsClass = UniqueWithCounts.Inputs.class +) @Operator public final class UniqueWithCounts extends RawOp { /** @@ -94,8 +100,8 @@ public final class UniqueWithCounts extends private Output count; - private UniqueWithCounts(Operation operation) { - super(operation); + public UniqueWithCounts(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); idx = operation.output(outputIdx++); @@ -172,6 +178,9 @@ public Output count() { return count; } + @OpInputsMetadata( + outputsClass = UniqueWithCounts.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnravelIndex.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnravelIndex.java index 17ca0f50647..b472efa7892 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnravelIndex.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UnravelIndex.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = UnravelIndex.OP_NAME, + inputsClass = UnravelIndex.Inputs.class +) @Operator public final class UnravelIndex extends RawOp implements Operand { /** @@ -62,8 +68,8 @@ public final class UnravelIndex extends RawOp implements Oper private Output output; - private UnravelIndex(Operation operation) { - super(operation); + public UnravelIndex(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UnravelIndex.class + ) public static class Inputs extends RawOpInputs> { /** * An 0-D or 1-D {@code int} Tensor whose elements are indices into the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstack.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstack.java index 0038c66838c..f82a894db5d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstack.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstack.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -47,6 +49,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Unstack.OP_NAME, + inputsClass = Unstack.Inputs.class +) @Operator public final class Unstack extends RawOp implements Iterable> { /** @@ -57,8 +63,8 @@ public final class Unstack extends RawOp implements Iterable> output; @SuppressWarnings("unchecked") - private Unstack(Operation operation) { - super(operation); + public Unstack(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList((Output[]) operation.outputList(outputIdx, outputLength)); @@ -141,6 +147,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = Unstack.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D or higher, with {@code axis} dimension size equal to {@code num}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstage.java index def9253c00c..e551c044389 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstage.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Unstage.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * The basic functionality is similar to dequeue with many fewer * capabilities and options. This Op is optimized for performance. */ +@OpMetadata( + opType = Unstage.OP_NAME, + inputsClass = Unstage.Inputs.class +) @Operator public final class Unstage extends RawOp implements Iterable> { /** @@ -49,8 +55,8 @@ public final class Unstage extends RawOp implements Iterable> { private List> values; @SuppressWarnings("unchecked") - private Unstage(Operation operation) { - super(operation); + public Unstage(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int valuesLength = operation.outputListLength("values"); values = Arrays.asList(operation.outputList(outputIdx, valuesLength)); @@ -206,6 +212,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = Unstage.class + ) public static class Inputs extends RawOpInputs { /** * The capacity attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java index f7a13f62d04..e9b79bbeaaf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = UpperBound.OP_NAME, + inputsClass = UpperBound.Inputs.class +) public final class UpperBound extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -59,8 +65,8 @@ public final class UpperBound extends RawOp implements Operan private Output output; - private UpperBound(Operation operation) { - super(operation); + public UpperBound(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -123,6 +129,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UpperBound.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D Tensor where each row is ordered. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarHandleOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarHandleOp.java index 67e1454d37c..3bec2d4b16a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarHandleOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarHandleOp.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a handle to a Variable resource. */ +@OpMetadata( + opType = VarHandleOp.OP_NAME, + inputsClass = VarHandleOp.Inputs.class +) @Operator public final class VarHandleOp extends RawOp implements Operand { /** @@ -47,8 +53,8 @@ public final class VarHandleOp extends RawOp implements Operand { private Output resource; @SuppressWarnings("unchecked") - private VarHandleOp(Operation operation) { - super(operation); + public VarHandleOp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resource = operation.output(outputIdx++); } @@ -209,6 +215,9 @@ public Options allowedDevices(String... allowedDevices) { } } + @OpInputsMetadata( + outputsClass = VarHandleOp.class + ) public static class Inputs extends RawOpInputs { /** * the container this variable is placed in. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarIsInitializedOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarIsInitializedOp.java index 25156efd4ec..aad05c3037f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarIsInitializedOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VarIsInitializedOp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Checks whether a resource handle-based variable has been initialized. */ +@OpMetadata( + opType = VarIsInitializedOp.OP_NAME, + inputsClass = VarIsInitializedOp.Inputs.class +) @Operator public final class VarIsInitializedOp extends RawOp implements Operand { /** @@ -43,8 +49,8 @@ public final class VarIsInitializedOp extends RawOp implements Operand { private Output isInitialized; - private VarIsInitializedOp(Operation operation) { - super(operation); + public VarIsInitializedOp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; isInitialized = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return isInitialized; } + @OpInputsMetadata( + outputsClass = VarIsInitializedOp.class + ) public static class Inputs extends RawOpInputs { /** * the input resource handle. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Variable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Variable.java index 12eebf8189c..0ed775841cb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Variable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Variable.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * * @param data type for {@code ref} output */ +@OpMetadata( + opType = Variable.OP_NAME, + inputsClass = Variable.Inputs.class +) @Operator public final class Variable extends RawOp implements Operand { /** @@ -50,8 +56,8 @@ public final class Variable extends RawOp implements Operand private Output ref; - private Variable(Operation operation) { - super(operation); + public Variable(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; ref = operation.output(outputIdx++); } @@ -159,6 +165,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = Variable.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the variable tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VariableShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VariableShape.java index f7d5a59a1b0..f5a90fb5896 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VariableShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/VariableShape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -45,6 +47,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = VariableShape.OP_NAME, + inputsClass = VariableShape.Inputs.class +) @Operator public final class VariableShape extends RawOp implements Operand { /** @@ -54,8 +60,8 @@ public final class VariableShape extends RawOp implements Ope private Output output; - private VariableShape(Operation operation) { - super(operation); + public VariableShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = VariableShape.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Where.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Where.java index 4fc22c56aed..d10d3c5561c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Where.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Where.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -92,6 +94,10 @@ * [2, 1, 1]] *

*/ +@OpMetadata( + opType = Where.OP_NAME, + inputsClass = Where.Inputs.class +) @Operator public final class Where extends RawOp implements Operand { /** @@ -101,8 +107,8 @@ public final class Where extends RawOp implements Operand { private Output index; - private Where(Operation operation) { - super(operation); + public Where(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; index = operation.output(outputIdx++); } @@ -137,6 +143,9 @@ public Output asOutput() { return index; } + @OpInputsMetadata( + outputsClass = Where.class + ) public static class Inputs extends RawOpInputs { /** * The condition input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Window.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Window.java index a363a2eab9e..ab241b515ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Window.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Window.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * The Window operation */ +@OpMetadata( + opType = Window.OP_NAME, + inputsClass = Window.Inputs.class +) public final class Window extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class Window extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private Window(Operation operation) { - super(operation); + public Window(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = Window.class + ) public static class Inputs extends RawOpInputs { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ZerosLike.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ZerosLike.java index ece2adac5ce..ebac3a1d700 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ZerosLike.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ZerosLike.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = ZerosLike.OP_NAME, + inputsClass = ZerosLike.Inputs.class +) @Operator public final class ZerosLike extends RawOp implements Operand { /** @@ -45,8 +51,8 @@ public final class ZerosLike extends RawOp implements Operand y; - private ZerosLike(Operation operation) { - super(operation); + public ZerosLike(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = ZerosLike.class + ) public static class Inputs extends RawOpInputs> { /** * a tensor of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousIterator.java index 010fdceba22..5ecb7359a44 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousIterator.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ /** * A container for an iterator resource. */ +@OpMetadata( + opType = AnonymousIterator.OP_NAME, + inputsClass = AnonymousIterator.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class AnonymousIterator extends RawOp { private Output deleter; @SuppressWarnings("unchecked") - private AnonymousIterator(Operation operation) { - super(operation); + public AnonymousIterator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); deleter = operation.output(outputIdx++); @@ -101,6 +107,9 @@ public Output deleter() { return deleter; } + @OpInputsMetadata( + outputsClass = AnonymousIterator.class + ) public static class Inputs extends RawOpInputs { /** * The outputTypes attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java index 1ab01851c85..9194702ad24 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java @@ -26,11 +26,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The AnonymousMemoryCache operation */ +@OpMetadata( + opType = AnonymousMemoryCache.OP_NAME, + inputsClass = AnonymousMemoryCache.Inputs.class +) public final class AnonymousMemoryCache extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class AnonymousMemoryCache extends RawOp { private Output deleter; @SuppressWarnings("unchecked") - private AnonymousMemoryCache(Operation operation) { - super(operation); + public AnonymousMemoryCache(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); deleter = operation.output(outputIdx++); @@ -81,6 +87,9 @@ public Output deleter() { return deleter; } + @OpInputsMetadata( + outputsClass = AnonymousMemoryCache.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new AnonymousMemoryCache(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMultiDeviceIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMultiDeviceIterator.java index 19f0ae37083..cce27d9053a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMultiDeviceIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMultiDeviceIterator.java @@ -29,12 +29,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * A container for a multi device iterator resource. */ +@OpMetadata( + opType = AnonymousMultiDeviceIterator.OP_NAME, + inputsClass = AnonymousMultiDeviceIterator.Inputs.class +) public final class AnonymousMultiDeviceIterator extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class AnonymousMultiDeviceIterator extends RawOp { private Output deleter; @SuppressWarnings("unchecked") - private AnonymousMultiDeviceIterator(Operation operation) { - super(operation); + public AnonymousMultiDeviceIterator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); deleter = operation.output(outputIdx++); @@ -103,6 +109,9 @@ public Output deleter() { return deleter; } + @OpInputsMetadata( + outputsClass = AnonymousMultiDeviceIterator.class + ) public static class Inputs extends RawOpInputs { /** * The devices attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertCardinalityDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertCardinalityDataset.java index 004b64d085e..344fbead34c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertCardinalityDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertCardinalityDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * The AssertCardinalityDataset operation */ +@OpMetadata( + opType = AssertCardinalityDataset.OP_NAME, + inputsClass = AssertCardinalityDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class AssertCardinalityDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private AssertCardinalityDataset(Operation operation) { - super(operation); + public AssertCardinalityDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = AssertCardinalityDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertNextDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertNextDataset.java index 957589aa4c7..183b8ac9ddb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertNextDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertNextDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -45,6 +47,10 @@ * means that the check happens after any static optimizations are applied * to the dataset graph. */ +@OpMetadata( + opType = AssertNextDataset.OP_NAME, + inputsClass = AssertNextDataset.Inputs.class +) @Operator( group = "data" ) @@ -57,8 +63,8 @@ public final class AssertNextDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private AssertNextDataset(Operation operation) { - super(operation); + public AssertNextDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = AssertNextDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AutoShardDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AutoShardDataset.java index 46150ca633f..b54f548540c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AutoShardDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AutoShardDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -44,6 +46,10 @@ *

This dataset will throw a NotFound error if we cannot shard the dataset * automatically. */ +@OpMetadata( + opType = AutoShardDataset.OP_NAME, + inputsClass = AutoShardDataset.Inputs.class +) @Operator( group = "data" ) @@ -56,8 +62,8 @@ public final class AutoShardDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private AutoShardDataset(Operation operation) { - super(operation); + public AutoShardDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -172,6 +178,9 @@ public Options numReplicas(Long numReplicas) { } } + @OpInputsMetadata( + outputsClass = AutoShardDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BatchDataset.java index aaf79b9104d..fbd817a8f20 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ /** * Creates a dataset that batches {@code batch_size} elements from {@code input_dataset}. */ +@OpMetadata( + opType = BatchDataset.OP_NAME, + inputsClass = BatchDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class BatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private BatchDataset(Operation operation) { - super(operation); + public BatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -142,6 +148,9 @@ public Options parallelCopy(Boolean parallelCopy) { } } + @OpInputsMetadata( + outputsClass = BatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BytesProducedStatsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BytesProducedStatsDataset.java index ead0ae179b3..19eb682960b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BytesProducedStatsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/BytesProducedStatsDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ /** * Records the bytes size of each element of {@code input_dataset} in a StatsAggregator. */ +@OpMetadata( + opType = BytesProducedStatsDataset.OP_NAME, + inputsClass = BytesProducedStatsDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class BytesProducedStatsDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private BytesProducedStatsDataset(Operation operation) { - super(operation); + public BytesProducedStatsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = BytesProducedStatsDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CSVDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CSVDataset.java index 5865a5bbfff..07b39ecd081 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CSVDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CSVDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -40,6 +42,10 @@ /** * The CSVDatasetV2 operation */ +@OpMetadata( + opType = CSVDataset.OP_NAME, + inputsClass = CSVDataset.Inputs.class +) @Operator( group = "data" ) @@ -52,8 +58,8 @@ public final class CSVDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private CSVDataset(Operation operation) { - super(operation); + public CSVDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -117,6 +123,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = CSVDataset.class + ) public static class Inputs extends RawOpInputs { /** * The filenames input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CacheDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CacheDataset.java index 8634e504fa2..842112ec012 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CacheDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CacheDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ /** * The CacheDatasetV2 operation */ +@OpMetadata( + opType = CacheDataset.OP_NAME, + inputsClass = CacheDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class CacheDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private CacheDataset(Operation operation) { - super(operation); + public CacheDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = CacheDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestBranchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestBranchDataset.java index ce7bc509f4e..75295649425 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestBranchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestBranchDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * The ChooseFastestBranchDataset operation */ +@OpMetadata( + opType = ChooseFastestBranchDataset.OP_NAME, + inputsClass = ChooseFastestBranchDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class ChooseFastestBranchDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private ChooseFastestBranchDataset(Operation operation) { - super(operation); + public ChooseFastestBranchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -120,6 +126,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ChooseFastestBranchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestDataset.java index a617ab45fa3..8550528ae05 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ChooseFastestDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The ChooseFastestDataset operation */ +@OpMetadata( + opType = ChooseFastestDataset.OP_NAME, + inputsClass = ChooseFastestDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class ChooseFastestDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private ChooseFastestDataset(Operation operation) { - super(operation); + public ChooseFastestDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ChooseFastestDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDatasets input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java index 6fc99b577b4..6edd206187f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java @@ -28,12 +28,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Compresses a dataset element. */ +@OpMetadata( + opType = CompressElement.OP_NAME, + inputsClass = CompressElement.Inputs.class +) public final class CompressElement extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class CompressElement extends RawOp implements Operand { private Output compressed; @SuppressWarnings("unchecked") - private CompressElement(Operation operation) { - super(operation); + public CompressElement(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; compressed = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return (Output) compressed; } + @OpInputsMetadata( + outputsClass = CompressElement.class + ) public static class Inputs extends RawOpInputs { /** * The components input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ConcatenateDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ConcatenateDataset.java index b5f7cb8c11a..1a58173cecd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ConcatenateDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ConcatenateDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that concatenates {@code input_dataset} with {@code another_dataset}. */ +@OpMetadata( + opType = ConcatenateDataset.OP_NAME, + inputsClass = ConcatenateDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class ConcatenateDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ConcatenateDataset(Operation operation) { - super(operation); + public ConcatenateDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ConcatenateDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DataServiceDatasetV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DataServiceDatasetV2.java index 20cf98e1806..0574b406440 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DataServiceDatasetV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DataServiceDatasetV2.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * Creates a dataset that reads data from the tf.data service. */ +@OpMetadata( + opType = DataServiceDatasetV2.OP_NAME, + inputsClass = DataServiceDatasetV2.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class DataServiceDatasetV2 extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private DataServiceDatasetV2(Operation operation) { - super(operation); + public DataServiceDatasetV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -207,6 +213,9 @@ public Options targetWorkers(String targetWorkers) { } } + @OpInputsMetadata( + outputsClass = DataServiceDatasetV2.class + ) public static class Inputs extends RawOpInputs { /** * The datasetId input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetCardinality.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetCardinality.java index 4d16a0c6556..1465873848f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetCardinality.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetCardinality.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Returns the cardinality of {@code input_dataset}. * Returns the cardinality of {@code input_dataset}. */ +@OpMetadata( + opType = DatasetCardinality.OP_NAME, + inputsClass = DatasetCardinality.Inputs.class +) @Operator( group = "data" ) @@ -46,8 +52,8 @@ public final class DatasetCardinality extends RawOp implements Operand { private Output cardinality; - private DatasetCardinality(Operation operation) { - super(operation); + public DatasetCardinality(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; cardinality = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return cardinality; } + @OpInputsMetadata( + outputsClass = DatasetCardinality.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the dataset to return cardinality for. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetFromGraph.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetFromGraph.java index b1e0f690462..52927c5281a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetFromGraph.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetFromGraph.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Creates a dataset from the given {@code graph_def}. * Creates a dataset from the provided {@code graph_def}. */ +@OpMetadata( + opType = DatasetFromGraph.OP_NAME, + inputsClass = DatasetFromGraph.Inputs.class +) @Operator( group = "data" ) @@ -47,8 +53,8 @@ public final class DatasetFromGraph extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private DatasetFromGraph(Operation operation) { - super(operation); + public DatasetFromGraph(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DatasetFromGraph.class + ) public static class Inputs extends RawOpInputs { /** * The graph representation of the dataset (as serialized GraphDef). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToGraph.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToGraph.java index b8c92cd4c61..94b2149b76f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToGraph.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToGraph.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Returns a serialized GraphDef representing {@code input_dataset}. * Returns a graph representation for {@code input_dataset}. */ +@OpMetadata( + opType = DatasetToGraph.OP_NAME, + inputsClass = DatasetToGraph.Inputs.class +) @Operator( group = "data" ) @@ -46,8 +52,8 @@ public final class DatasetToGraph extends RawOp implements Operand { private Output graph; - private DatasetToGraph(Operation operation) { - super(operation); + public DatasetToGraph(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; graph = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options stripDeviceAssignment(Boolean stripDeviceAssignment) { } } + @OpInputsMetadata( + outputsClass = DatasetToGraph.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the dataset to return the graph representation for. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToSingleElement.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToSingleElement.java index 1ebef27faa2..bf038dd2493 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToSingleElement.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToSingleElement.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Outputs the single element from the given dataset. */ +@OpMetadata( + opType = DatasetToSingleElement.OP_NAME, + inputsClass = DatasetToSingleElement.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class DatasetToSingleElement extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private DatasetToSingleElement(Operation operation) { - super(operation); + public DatasetToSingleElement(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -98,6 +104,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = DatasetToSingleElement.class + ) public static class Inputs extends RawOpInputs { /** * A handle to a dataset that contains a single element. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToTfRecord.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToTfRecord.java index 02a9153d092..f52a044474b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToTfRecord.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DatasetToTfRecord.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -33,6 +35,10 @@ /** * Writes the given dataset to the given file using the TFRecord format. */ +@OpMetadata( + opType = DatasetToTfRecord.OP_NAME, + inputsClass = DatasetToTfRecord.Inputs.class +) @Operator( group = "data" ) @@ -42,8 +48,8 @@ public final class DatasetToTfRecord extends RawOp { */ public static final String OP_NAME = "DatasetToTFRecord"; - private DatasetToTfRecord(Operation operation) { - super(operation); + public DatasetToTfRecord(Operation operation) { + super(operation, OP_NAME); } /** @@ -68,6 +74,9 @@ public static DatasetToTfRecord create(Scope scope, Operand inp return new DatasetToTfRecord(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DatasetToTfRecord.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the dataset to write. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteIterator.java index 6888dfa07ca..b7577b7b537 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteIterator.java @@ -26,12 +26,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * A container for an iterator resource. */ +@OpMetadata( + opType = DeleteIterator.OP_NAME, + inputsClass = DeleteIterator.Inputs.class +) @Operator( group = "data" ) @@ -41,8 +47,8 @@ public final class DeleteIterator extends RawOp { */ public static final String OP_NAME = "DeleteIterator"; - private DeleteIterator(Operation operation) { - super(operation); + public DeleteIterator(Operation operation) { + super(operation, OP_NAME); } /** @@ -64,6 +70,9 @@ public static DeleteIterator create(Scope scope, Operand handle return new DeleteIterator(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DeleteIterator.class + ) public static class Inputs extends RawOpInputs { /** * A handle to the iterator to delete. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java index d342a701ddf..a7925564c1a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The DeleteMemoryCache operation */ +@OpMetadata( + opType = DeleteMemoryCache.OP_NAME, + inputsClass = DeleteMemoryCache.Inputs.class +) public final class DeleteMemoryCache extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "DeleteMemoryCache"; - private DeleteMemoryCache(Operation operation) { - super(operation); + public DeleteMemoryCache(Operation operation) { + super(operation, OP_NAME); } /** @@ -60,6 +66,9 @@ public static DeleteMemoryCache create(Scope scope, Operand han return new DeleteMemoryCache(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DeleteMemoryCache.class + ) public static class Inputs extends RawOpInputs { /** * The handle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java index cc1239be199..1f3e1f6f3d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java @@ -27,19 +27,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * A container for an iterator resource. */ +@OpMetadata( + opType = DeleteMultiDeviceIterator.OP_NAME, + inputsClass = DeleteMultiDeviceIterator.Inputs.class +) public final class DeleteMultiDeviceIterator extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "DeleteMultiDeviceIterator"; - private DeleteMultiDeviceIterator(Operation operation) { - super(operation); + public DeleteMultiDeviceIterator(Operation operation) { + super(operation, OP_NAME); } /** @@ -64,6 +70,9 @@ public static DeleteMultiDeviceIterator create(Scope scope, return new DeleteMultiDeviceIterator(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DeleteMultiDeviceIterator.class + ) public static class Inputs extends RawOpInputs { /** * A handle to the multi device iterator to delete. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DenseToSparseBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DenseToSparseBatchDataset.java index 69815a21d5c..d28aea1ce70 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DenseToSparseBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DenseToSparseBatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that batches input elements into a SparseTensor. */ +@OpMetadata( + opType = DenseToSparseBatchDataset.OP_NAME, + inputsClass = DenseToSparseBatchDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class DenseToSparseBatchDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private DenseToSparseBatchDataset(Operation operation) { - super(operation); + public DenseToSparseBatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -104,6 +110,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DenseToSparseBatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * A handle to an input dataset. Must have a single component. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeserializeIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeserializeIterator.java index 0492d8e3a5b..9b994548e1e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeserializeIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeserializeIterator.java @@ -26,12 +26,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * Converts the given variant tensor to an iterator and stores it in the given resource. */ +@OpMetadata( + opType = DeserializeIterator.OP_NAME, + inputsClass = DeserializeIterator.Inputs.class +) @Operator( group = "data" ) @@ -41,8 +47,8 @@ public final class DeserializeIterator extends RawOp { */ public static final String OP_NAME = "DeserializeIterator"; - private DeserializeIterator(Operation operation) { - super(operation); + public DeserializeIterator(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static DeserializeIterator create(Scope scope, Operand r return new DeserializeIterator(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DeserializeIterator.class + ) public static class Inputs extends RawOpInputs { /** * A handle to an iterator resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DirectedInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DirectedInterleaveDataset.java index 9f2b7602cc6..5b8bf250501 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DirectedInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DirectedInterleaveDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * A substitute for {@code InterleaveDataset} on a fixed list of {@code N} datasets. */ +@OpMetadata( + opType = DirectedInterleaveDataset.OP_NAME, + inputsClass = DirectedInterleaveDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class DirectedInterleaveDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private DirectedInterleaveDataset(Operation operation) { - super(operation); + public DirectedInterleaveDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Options stopOnEmptyDataset(Boolean stopOnEmptyDataset) { } } + @OpInputsMetadata( + outputsClass = DirectedInterleaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * A dataset of scalar {@code DT_INT64} elements that determines which of the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java index 10c0ad3c957..2223010b220 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The DummyIterationCounter operation */ +@OpMetadata( + opType = DummyIterationCounter.OP_NAME, + inputsClass = DummyIterationCounter.Inputs.class +) public final class DummyIterationCounter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class DummyIterationCounter extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private DummyIterationCounter(Operation operation) { - super(operation); + public DummyIterationCounter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -76,6 +82,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DummyIterationCounter.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new DummyIterationCounter(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterByLastComponentDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterByLastComponentDataset.java index 20474ebe360..3f8d67c1891 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterByLastComponentDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterByLastComponentDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset containing elements of first component of {@code input_dataset} having true in the last component. */ +@OpMetadata( + opType = FilterByLastComponentDataset.OP_NAME, + inputsClass = FilterByLastComponentDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class FilterByLastComponentDataset extends RawOp implements Operand private Output output; @SuppressWarnings("unchecked") - private FilterByLastComponentDataset(Operation operation) { - super(operation); + public FilterByLastComponentDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = FilterByLastComponentDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterDataset.java index 46b5c47e762..473ffa03ea3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FilterDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ *

  • One tensor for each value in {@code other_arguments}.
  • * */ +@OpMetadata( + opType = FilterDataset.OP_NAME, + inputsClass = FilterDataset.Inputs.class +) @Operator( group = "data" ) @@ -56,8 +62,8 @@ public final class FilterDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private FilterDataset(Operation operation) { - super(operation); + public FilterDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = FilterDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FinalizeDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FinalizeDataset.java index a020f21fddf..4f36eee9dab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FinalizeDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FinalizeDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset by applying {@code tf.data.Options} to {@code input_dataset}. */ +@OpMetadata( + opType = FinalizeDataset.OP_NAME, + inputsClass = FinalizeDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class FinalizeDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private FinalizeDataset(Operation operation) { - super(operation); + public FinalizeDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -134,6 +140,9 @@ public Options hasCapturedRef(Boolean hasCapturedRef) { } } + @OpInputsMetadata( + outputsClass = FinalizeDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FixedLengthRecordDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FixedLengthRecordDataset.java index 7f0b41e25b9..5613dfd373c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FixedLengthRecordDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FixedLengthRecordDataset.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ /** * The FixedLengthRecordDatasetV2 operation */ +@OpMetadata( + opType = FixedLengthRecordDataset.OP_NAME, + inputsClass = FixedLengthRecordDataset.Inputs.class +) @Operator( group = "data" ) @@ -47,8 +53,8 @@ public final class FixedLengthRecordDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private FixedLengthRecordDataset(Operation operation) { - super(operation); + public FixedLengthRecordDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = FixedLengthRecordDataset.class + ) public static class Inputs extends RawOpInputs { /** * The filenames input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FlatMapDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FlatMapDataset.java index 2a084772e7a..02fda22a75d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FlatMapDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/FlatMapDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * Dataset variant, and FlatMapDataset will flatten successive results * into a single Dataset. */ +@OpMetadata( + opType = FlatMapDataset.OP_NAME, + inputsClass = FlatMapDataset.Inputs.class +) @Operator( group = "data" ) @@ -53,8 +59,8 @@ public final class FlatMapDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private FlatMapDataset(Operation operation) { - super(operation); + public FlatMapDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = FlatMapDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GeneratorDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GeneratorDataset.java index fa2b2c66bc6..3b5d55c07b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GeneratorDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GeneratorDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Creates a dataset that invokes a function to generate elements. */ +@OpMetadata( + opType = GeneratorDataset.OP_NAME, + inputsClass = GeneratorDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class GeneratorDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private GeneratorDataset(Operation operation) { - super(operation); + public GeneratorDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = GeneratorDataset.class + ) public static class Inputs extends RawOpInputs { /** * The initFuncOtherArgs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByReducerDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByReducerDataset.java index 92c3b2023f8..71b10569545 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByReducerDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByReducerDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * Creates a dataset that computes a group-by on {@code input_dataset}. * Creates a dataset that computes a group-by on {@code input_dataset}. */ +@OpMetadata( + opType = GroupByReducerDataset.OP_NAME, + inputsClass = GroupByReducerDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class GroupByReducerDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private GroupByReducerDataset(Operation operation) { - super(operation); + public GroupByReducerDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -124,6 +130,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = GroupByReducerDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByWindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByWindowDataset.java index 539d3cecf14..13065f77464 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByWindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/GroupByWindowDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * Creates a dataset that computes a windowed group-by on {@code input_dataset}. * // TODO(mrry): Support non-int64 keys. */ +@OpMetadata( + opType = GroupByWindowDataset.OP_NAME, + inputsClass = GroupByWindowDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class GroupByWindowDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private GroupByWindowDataset(Operation operation) { - super(operation); + public GroupByWindowDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = GroupByWindowDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IgnoreErrorsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IgnoreErrorsDataset.java index 60216480497..e53d9b15825 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IgnoreErrorsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IgnoreErrorsDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that contains the elements of {@code input_dataset} ignoring errors. */ +@OpMetadata( + opType = IgnoreErrorsDataset.OP_NAME, + inputsClass = IgnoreErrorsDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class IgnoreErrorsDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private IgnoreErrorsDataset(Operation operation) { - super(operation); + public IgnoreErrorsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -134,6 +140,9 @@ public Options logWarning(Boolean logWarning) { } } + @OpInputsMetadata( + outputsClass = IgnoreErrorsDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InitializeTableFromDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InitializeTableFromDataset.java index dba5858d2ef..16fcaa2e8c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InitializeTableFromDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InitializeTableFromDataset.java @@ -26,12 +26,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The InitializeTableFromDataset operation */ +@OpMetadata( + opType = InitializeTableFromDataset.OP_NAME, + inputsClass = InitializeTableFromDataset.Inputs.class +) @Operator( group = "data" ) @@ -41,8 +47,8 @@ public final class InitializeTableFromDataset extends RawOp { */ public static final String OP_NAME = "InitializeTableFromDataset"; - private InitializeTableFromDataset(Operation operation) { - super(operation); + public InitializeTableFromDataset(Operation operation) { + super(operation, OP_NAME); } /** @@ -64,6 +70,9 @@ public static InitializeTableFromDataset create(Scope scope, Operand { /** * The tableHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InterleaveDataset.java index 25685e80ba5..4cc5b8d7fdf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/InterleaveDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -44,6 +46,10 @@ * InterleaveDataset will interleave sequences of up to {@code block_length} * consecutive elements from {@code cycle_length} input elements. */ +@OpMetadata( + opType = InterleaveDataset.OP_NAME, + inputsClass = InterleaveDataset.Inputs.class +) @Operator( group = "data" ) @@ -56,8 +62,8 @@ public final class InterleaveDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private InterleaveDataset(Operation operation) { - super(operation); + public InterleaveDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = InterleaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/Iterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/Iterator.java index 35b4da85be7..beb7563bd2f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/Iterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/Iterator.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The IteratorV2 operation */ +@OpMetadata( + opType = Iterator.OP_NAME, + inputsClass = Iterator.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class Iterator extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private Iterator(Operation operation) { - super(operation); + public Iterator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = Iterator.class + ) public static class Inputs extends RawOpInputs { /** * The sharedName attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java index 26695a9c0bf..9988643201b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The IteratorFromStringHandleV2 operation */ +@OpMetadata( + opType = IteratorFromStringHandle.OP_NAME, + inputsClass = IteratorFromStringHandle.Inputs.class +) public final class IteratorFromStringHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class IteratorFromStringHandle extends RawOp implements Operand resourceHandle; @SuppressWarnings("unchecked") - private IteratorFromStringHandle(Operation operation) { - super(operation); + public IteratorFromStringHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resourceHandle = operation.output(outputIdx++); } @@ -150,6 +156,9 @@ public Options outputShapes(Shape... outputShapes) { } } + @OpInputsMetadata( + outputsClass = IteratorFromStringHandle.class + ) public static class Inputs extends RawOpInputs { /** * The stringHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java index 922c13d28f2..48793131cab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Returns the name of the device on which {@code resource} has been placed. */ +@OpMetadata( + opType = IteratorGetDevice.OP_NAME, + inputsClass = IteratorGetDevice.Inputs.class +) public final class IteratorGetDevice extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class IteratorGetDevice extends RawOp implements Operand { private Output device; - private IteratorGetDevice(Operation operation) { - super(operation); + public IteratorGetDevice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; device = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return device; } + @OpInputsMetadata( + outputsClass = IteratorGetDevice.class + ) public static class Inputs extends RawOpInputs { /** * The resource input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNext.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNext.java index 9d48aeb39a5..f8055319926 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNext.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNext.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Gets the next output from the given iterator . */ +@OpMetadata( + opType = IteratorGetNext.OP_NAME, + inputsClass = IteratorGetNext.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class IteratorGetNext extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private IteratorGetNext(Operation operation) { - super(operation); + public IteratorGetNext(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -98,6 +104,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = IteratorGetNext.class + ) public static class Inputs extends RawOpInputs { /** * The iterator input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextAsOptional.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextAsOptional.java index c679cdbae31..6693787d360 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextAsOptional.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextAsOptional.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Gets the next output from the given iterator as an Optional variant. */ +@OpMetadata( + opType = IteratorGetNextAsOptional.OP_NAME, + inputsClass = IteratorGetNextAsOptional.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class IteratorGetNextAsOptional extends RawOp implements Operand optional; @SuppressWarnings("unchecked") - private IteratorGetNextAsOptional(Operation operation) { - super(operation); + public IteratorGetNextAsOptional(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; optional = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) optional; } + @OpInputsMetadata( + outputsClass = IteratorGetNextAsOptional.class + ) public static class Inputs extends RawOpInputs { /** * The iterator input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextSync.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextSync.java index 1de3a7c0131..c0212523eb7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextSync.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetNextSync.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * the calling thread is not a member of the thread pool used to execute parallel * operations (e.g. in eager mode). */ +@OpMetadata( + opType = IteratorGetNextSync.OP_NAME, + inputsClass = IteratorGetNextSync.Inputs.class +) @Operator( group = "data" ) @@ -54,8 +60,8 @@ public final class IteratorGetNextSync extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private IteratorGetNextSync(Operation operation) { - super(operation); + public IteratorGetNextSync(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -102,6 +108,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = IteratorGetNextSync.class + ) public static class Inputs extends RawOpInputs { /** * The iterator input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorToStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorToStringHandle.java index 6854c2574c7..c89371833b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorToStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorToStringHandle.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Converts the given {@code resource_handle} representing an iterator to a string. */ +@OpMetadata( + opType = IteratorToStringHandle.OP_NAME, + inputsClass = IteratorToStringHandle.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class IteratorToStringHandle extends RawOp implements Operand stringHandle; - private IteratorToStringHandle(Operation operation) { - super(operation); + public IteratorToStringHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; stringHandle = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return stringHandle; } + @OpInputsMetadata( + outputsClass = IteratorToStringHandle.class + ) public static class Inputs extends RawOpInputs { /** * A handle to an iterator resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LMDBDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LMDBDataset.java index f65b0c6a2ab..cd2f25f50f7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LMDBDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LMDBDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -46,6 +48,10 @@ *

    LMDB uses different file formats on big- and little-endian machines. * {@code data.LMDBDataset} can only read files in the format of the host machine. */ +@OpMetadata( + opType = LMDBDataset.OP_NAME, + inputsClass = LMDBDataset.Inputs.class +) @Operator( group = "data" ) @@ -58,8 +64,8 @@ public final class LMDBDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private LMDBDataset(Operation operation) { - super(operation); + public LMDBDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = LMDBDataset.class + ) public static class Inputs extends RawOpInputs { /** * A scalar or a vector containing the name(s) of the binary file(s) to be diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LatencyStatsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LatencyStatsDataset.java index bc8c24e59ff..23caca44faf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LatencyStatsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LatencyStatsDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ /** * Records the latency of producing {@code input_dataset} elements in a StatsAggregator. */ +@OpMetadata( + opType = LatencyStatsDataset.OP_NAME, + inputsClass = LatencyStatsDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class LatencyStatsDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private LatencyStatsDataset(Operation operation) { - super(operation); + public LatencyStatsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = LatencyStatsDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java index b30117b7c13..f7aab374ff6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = LeakyReluGrad.OP_NAME, + inputsClass = LeakyReluGrad.Inputs.class +) public final class LeakyReluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class LeakyReluGrad extends RawOp implements Ope private Output backprops; - private LeakyReluGrad(Operation operation) { - super(operation); + public LeakyReluGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -123,6 +129,9 @@ public Options alpha(Float alpha) { } } + @OpInputsMetadata( + outputsClass = LeakyReluGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding LeakyRelu operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LegacyParallelInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LegacyParallelInterleaveDataset.java index 94c8f08e22c..e80cc1d2319 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LegacyParallelInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LegacyParallelInterleaveDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -45,6 +47,10 @@ * allows the training step to proceed so long as some data is available. *

    !! WARNING !! This dataset is not deterministic! */ +@OpMetadata( + opType = LegacyParallelInterleaveDataset.OP_NAME, + inputsClass = LegacyParallelInterleaveDataset.Inputs.class +) @Operator( group = "data" ) @@ -57,8 +63,8 @@ public final class LegacyParallelInterleaveDataset extends RawOp implements Oper private Output handle; @SuppressWarnings("unchecked") - private LegacyParallelInterleaveDataset(Operation operation) { - super(operation); + public LegacyParallelInterleaveDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -160,6 +166,9 @@ public Options deterministic(String deterministic) { } } + @OpInputsMetadata( + outputsClass = LegacyParallelInterleaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LoadDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LoadDataset.java index be67a1a67a3..08bc5254788 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LoadDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LoadDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ /** * The LoadDataset operation */ +@OpMetadata( + opType = LoadDataset.OP_NAME, + inputsClass = LoadDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class LoadDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private LoadDataset(Operation operation) { - super(operation); + public LoadDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options compression(String compression) { } } + @OpInputsMetadata( + outputsClass = LoadDataset.class + ) public static class Inputs extends RawOpInputs { /** * The path input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MakeIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MakeIterator.java index 93cca862b96..fb2eb9a4ebf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MakeIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MakeIterator.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ * This operation may be executed multiple times. Each execution will reset the * iterator in {@code iterator} to the first element of {@code dataset}. */ +@OpMetadata( + opType = MakeIterator.OP_NAME, + inputsClass = MakeIterator.Inputs.class +) @Operator( group = "data" ) @@ -43,8 +49,8 @@ public final class MakeIterator extends RawOp { */ public static final String OP_NAME = "MakeIterator"; - private MakeIterator(Operation operation) { - super(operation); + public MakeIterator(Operation operation) { + super(operation, OP_NAME); } /** @@ -66,6 +72,9 @@ public static MakeIterator create(Scope scope, Operand dataset, return new MakeIterator(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = MakeIterator.class + ) public static class Inputs extends RawOpInputs { /** * The dataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapAndBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapAndBatchDataset.java index b1bbacd6622..6d3f2f0cb84 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapAndBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapAndBatchDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -44,6 +46,10 @@ *

    Unlike a "MapDataset", which applies {@code f} sequentially, this dataset invokes up * to {@code batch_size * num_parallel_batches} copies of {@code f} in parallel. */ +@OpMetadata( + opType = MapAndBatchDataset.OP_NAME, + inputsClass = MapAndBatchDataset.Inputs.class +) @Operator( group = "data" ) @@ -56,8 +62,8 @@ public final class MapAndBatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private MapAndBatchDataset(Operation operation) { - super(operation); + public MapAndBatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -159,6 +165,9 @@ public Options preserveCardinality(Boolean preserveCardinality) { } } + @OpInputsMetadata( + outputsClass = MapAndBatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapDataset.java index 81c569034d3..41ed07c9da3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MapDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset}. */ +@OpMetadata( + opType = MapDataset.OP_NAME, + inputsClass = MapDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class MapDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private MapDataset(Operation operation) { - super(operation); + public MapDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -166,6 +172,9 @@ public Options preserveCardinality(Boolean preserveCardinality) { } } + @OpInputsMetadata( + outputsClass = MapDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MatchingFilesDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MatchingFilesDataset.java index c8d0af517a3..015f28e5bdb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MatchingFilesDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MatchingFilesDataset.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * The MatchingFilesDataset operation */ +@OpMetadata( + opType = MatchingFilesDataset.OP_NAME, + inputsClass = MatchingFilesDataset.Inputs.class +) @Operator( group = "data" ) @@ -46,8 +52,8 @@ public final class MatchingFilesDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private MatchingFilesDataset(Operation operation) { - super(operation); + public MatchingFilesDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = MatchingFilesDataset.class + ) public static class Inputs extends RawOpInputs { /** * The patterns input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MaxIntraOpParallelismDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MaxIntraOpParallelismDataset.java index 883390d70de..1106758abe9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MaxIntraOpParallelismDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MaxIntraOpParallelismDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that overrides the maximum intra-op parallelism. */ +@OpMetadata( + opType = MaxIntraOpParallelismDataset.OP_NAME, + inputsClass = MaxIntraOpParallelismDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class MaxIntraOpParallelismDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private MaxIntraOpParallelismDataset(Operation operation) { - super(operation); + public MaxIntraOpParallelismDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = MaxIntraOpParallelismDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ModelDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ModelDataset.java index b600addbc87..4545f2aee1b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ModelDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ModelDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * Identity transformation that models performance. * Identity transformation that models performance. */ +@OpMetadata( + opType = ModelDataset.OP_NAME, + inputsClass = ModelDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class ModelDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ModelDataset(Operation operation) { - super(operation); + public ModelDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -187,6 +193,9 @@ public Options ramBudget(Long ramBudget) { } } + @OpInputsMetadata( + outputsClass = ModelDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java index 60589923320..0fa7f6b0aea 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Creates a MultiDeviceIterator resource. */ +@OpMetadata( + opType = MultiDeviceIterator.OP_NAME, + inputsClass = MultiDeviceIterator.Inputs.class +) public final class MultiDeviceIterator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class MultiDeviceIterator extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private MultiDeviceIterator(Operation operation) { - super(operation); + public MultiDeviceIterator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = MultiDeviceIterator.class + ) public static class Inputs extends RawOpInputs { /** * A list of devices the iterator works across. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java index ce3b30d17f7..5f0bea11def 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Generates a MultiDeviceIterator resource from its provided string handle. */ +@OpMetadata( + opType = MultiDeviceIteratorFromStringHandle.OP_NAME, + inputsClass = MultiDeviceIteratorFromStringHandle.Inputs.class +) public final class MultiDeviceIteratorFromStringHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class MultiDeviceIteratorFromStringHandle extends RawOp implements private Output multiDeviceIterator; @SuppressWarnings("unchecked") - private MultiDeviceIteratorFromStringHandle(Operation operation) { - super(operation); + public MultiDeviceIteratorFromStringHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; multiDeviceIterator = operation.output(outputIdx++); } @@ -150,6 +156,9 @@ public Options outputShapes(Shape... outputShapes) { } } + @OpInputsMetadata( + outputsClass = MultiDeviceIteratorFromStringHandle.class + ) public static class Inputs extends RawOpInputs { /** * String representing the resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java index 778bf06b10b..6c1edc53a1c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * Gets next element for the provided shard number. */ +@OpMetadata( + opType = MultiDeviceIteratorGetNextFromShard.OP_NAME, + inputsClass = MultiDeviceIteratorGetNextFromShard.Inputs.class +) public final class MultiDeviceIteratorGetNextFromShard extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class MultiDeviceIteratorGetNextFromShard extends RawOp implements private List> components; @SuppressWarnings("unchecked") - private MultiDeviceIteratorGetNextFromShard(Operation operation) { - super(operation); + public MultiDeviceIteratorGetNextFromShard(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -102,6 +108,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = MultiDeviceIteratorGetNextFromShard.class + ) public static class Inputs extends RawOpInputs { /** * A MultiDeviceIterator resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java index 107b2c1cb5f..a6879702e46 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; /** * Initializes the multi device iterator with the given dataset. */ +@OpMetadata( + opType = MultiDeviceIteratorInit.OP_NAME, + inputsClass = MultiDeviceIteratorInit.Inputs.class +) public final class MultiDeviceIteratorInit extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class MultiDeviceIteratorInit extends RawOp implements Operand incarnationId; - private MultiDeviceIteratorInit(Operation operation) { - super(operation); + public MultiDeviceIteratorInit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; incarnationId = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return incarnationId; } + @OpInputsMetadata( + outputsClass = MultiDeviceIteratorInit.class + ) public static class Inputs extends RawOpInputs { /** * Dataset to be iterated upon. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java index d65385e4f31..bf4763765c7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Produces a string handle for the given MultiDeviceIterator. */ +@OpMetadata( + opType = MultiDeviceIteratorToStringHandle.OP_NAME, + inputsClass = MultiDeviceIteratorToStringHandle.Inputs.class +) public final class MultiDeviceIteratorToStringHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class MultiDeviceIteratorToStringHandle extends RawOp implements Op private Output stringHandle; - private MultiDeviceIteratorToStringHandle(Operation operation) { - super(operation); + public MultiDeviceIteratorToStringHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; stringHandle = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return stringHandle; } + @OpInputsMetadata( + outputsClass = MultiDeviceIteratorToStringHandle.class + ) public static class Inputs extends RawOpInputs { /** * A MultiDeviceIterator resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/NonSerializableDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/NonSerializableDataset.java index a57c2fc3513..b7ce803ce39 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/NonSerializableDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/NonSerializableDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The NonSerializableDataset operation */ +@OpMetadata( + opType = NonSerializableDataset.OP_NAME, + inputsClass = NonSerializableDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class NonSerializableDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private NonSerializableDataset(Operation operation) { - super(operation); + public NonSerializableDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = NonSerializableDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OneShotIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OneShotIterator.java index a2744d8bacb..f022e05c941 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OneShotIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OneShotIterator.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -54,6 +56,10 @@ * (including fed values) as parameters, and which may be reset multiple * times by rerunning "MakeIterator". */ +@OpMetadata( + opType = OneShotIterator.OP_NAME, + inputsClass = OneShotIterator.Inputs.class +) @Operator( group = "data" ) @@ -66,8 +72,8 @@ public final class OneShotIterator extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private OneShotIterator(Operation operation) { - super(operation); + public OneShotIterator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -179,6 +185,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = OneShotIterator.class + ) public static class Inputs extends RawOpInputs { /** * The outputTypes attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptimizeDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptimizeDataset.java index 17b48b79f36..a30c7587cdc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptimizeDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptimizeDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ * Creates a dataset by applying related optimizations to {@code input_dataset}. * Creates a dataset by applying related optimizations to {@code input_dataset}. */ +@OpMetadata( + opType = OptimizeDataset.OP_NAME, + inputsClass = OptimizeDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class OptimizeDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private OptimizeDataset(Operation operation) { - super(operation); + public OptimizeDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -169,6 +175,9 @@ public Options optimizationConfigs(String... optimizationConfigs) { } } + @OpInputsMetadata( + outputsClass = OptimizeDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalFromValue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalFromValue.java index 87d6a090953..1d8c245fb7f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalFromValue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalFromValue.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ /** * Constructs an Optional variant from a tuple of tensors. */ +@OpMetadata( + opType = OptionalFromValue.OP_NAME, + inputsClass = OptionalFromValue.Inputs.class +) @Operator( group = "data" ) @@ -47,8 +53,8 @@ public final class OptionalFromValue extends RawOp implements Operand { private Output optional; @SuppressWarnings("unchecked") - private OptionalFromValue(Operation operation) { - super(operation); + public OptionalFromValue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; optional = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return (Output) optional; } + @OpInputsMetadata( + outputsClass = OptionalFromValue.class + ) public static class Inputs extends RawOpInputs { /** * The components input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalGetValue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalGetValue.java index aa3320ef50e..4293d28a907 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalGetValue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalGetValue.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Returns the value stored in an Optional variant or raises an error if none exists. */ +@OpMetadata( + opType = OptionalGetValue.OP_NAME, + inputsClass = OptionalGetValue.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class OptionalGetValue extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private OptionalGetValue(Operation operation) { - super(operation); + public OptionalGetValue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -98,6 +104,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = OptionalGetValue.class + ) public static class Inputs extends RawOpInputs { /** * The optional input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalHasValue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalHasValue.java index 53dac930b5a..930f9e7725e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalHasValue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalHasValue.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Returns true if and only if the given Optional variant has a value. */ +@OpMetadata( + opType = OptionalHasValue.OP_NAME, + inputsClass = OptionalHasValue.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class OptionalHasValue extends RawOp implements Operand { private Output hasValue; - private OptionalHasValue(Operation operation) { - super(operation); + public OptionalHasValue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; hasValue = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return hasValue; } + @OpInputsMetadata( + outputsClass = OptionalHasValue.class + ) public static class Inputs extends RawOpInputs { /** * The optional input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalNone.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalNone.java index f051ca0c79b..5b2ca755908 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalNone.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionalNone.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * Creates an Optional variant with no value. */ +@OpMetadata( + opType = OptionalNone.OP_NAME, + inputsClass = OptionalNone.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class OptionalNone extends RawOp implements Operand { private Output optional; @SuppressWarnings("unchecked") - private OptionalNone(Operation operation) { - super(operation); + public OptionalNone(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; optional = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return (Output) optional; } + @OpInputsMetadata( + outputsClass = OptionalNone.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new OptionalNone(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionsDataset.java index c62b67b68c4..63b202eb184 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/OptionsDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset by attaching tf.data.Options to {@code input_dataset}. */ +@OpMetadata( + opType = OptionsDataset.OP_NAME, + inputsClass = OptionsDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class OptionsDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private OptionsDataset(Operation operation) { - super(operation); + public OptionsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = OptionsDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PaddedBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PaddedBatchDataset.java index fe1e629e8cd..5941ac19a16 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PaddedBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PaddedBatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ /** * Creates a dataset that batches and pads {@code batch_size} elements from the input. */ +@OpMetadata( + opType = PaddedBatchDataset.OP_NAME, + inputsClass = PaddedBatchDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class PaddedBatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private PaddedBatchDataset(Operation operation) { - super(operation); + public PaddedBatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -150,6 +156,9 @@ public Options parallelCopy(Boolean parallelCopy) { } } + @OpInputsMetadata( + outputsClass = PaddedBatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelBatchDataset.java index 15805d15774..08e3928b4b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelBatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -39,6 +41,10 @@ /** * The ParallelBatchDataset operation */ +@OpMetadata( + opType = ParallelBatchDataset.OP_NAME, + inputsClass = ParallelBatchDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class ParallelBatchDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private ParallelBatchDataset(Operation operation) { - super(operation); + public ParallelBatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -169,6 +175,9 @@ public Options deterministic(String deterministic) { } } + @OpInputsMetadata( + outputsClass = ParallelBatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelInterleaveDataset.java index c5e17ed7388..edbe27c6f73 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelInterleaveDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -50,6 +52,10 @@ * {@code experimental_deterministic} parameter of {@code tf.data.Options} to {@code False}. * This can improve performance at the expense of non-determinism. */ +@OpMetadata( + opType = ParallelInterleaveDataset.OP_NAME, + inputsClass = ParallelInterleaveDataset.Inputs.class +) @Operator( group = "data" ) @@ -62,8 +68,8 @@ public final class ParallelInterleaveDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private ParallelInterleaveDataset(Operation operation) { - super(operation); + public ParallelInterleaveDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -182,6 +188,9 @@ public Options deterministic(String deterministic) { } } + @OpInputsMetadata( + outputsClass = ParallelInterleaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * Dataset that produces a stream of arguments for the function {@code f}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelMapDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelMapDataset.java index 455c6b600e9..1768a7c488d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelMapDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelMapDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * Unlike a "MapDataset", which applies {@code f} sequentially, this dataset invokes up * to {@code num_parallel_calls} copies of {@code f} in parallel. */ +@OpMetadata( + opType = ParallelMapDataset.OP_NAME, + inputsClass = ParallelMapDataset.Inputs.class +) @Operator( group = "data" ) @@ -53,8 +59,8 @@ public final class ParallelMapDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ParallelMapDataset(Operation operation) { - super(operation); + public ParallelMapDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -198,6 +204,9 @@ public Options preserveCardinality(Boolean preserveCardinality) { } } + @OpInputsMetadata( + outputsClass = ParallelMapDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParseExampleDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParseExampleDataset.java index 501716524f8..1fae1337fc4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParseExampleDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParseExampleDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * Transforms {@code input_dataset} containing {@code Example} protos as vectors of DT_STRING into a dataset of {@code Tensor} or {@code SparseTensor} objects representing the parsed features. */ +@OpMetadata( + opType = ParseExampleDataset.OP_NAME, + inputsClass = ParseExampleDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class ParseExampleDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ParseExampleDataset(Operation operation) { - super(operation); + public ParseExampleDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -240,6 +246,9 @@ public Options raggedKeys(String... raggedKeys) { } } + @OpInputsMetadata( + outputsClass = ParseExampleDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrefetchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrefetchDataset.java index b27ce7b56ef..8334e1aa890 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrefetchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrefetchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that asynchronously prefetches elements from {@code input_dataset}. */ +@OpMetadata( + opType = PrefetchDataset.OP_NAME, + inputsClass = PrefetchDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class PrefetchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private PrefetchDataset(Operation operation) { - super(operation); + public PrefetchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -191,6 +197,9 @@ public Options bufferSizeMin(Long bufferSizeMin) { } } + @OpInputsMetadata( + outputsClass = PrefetchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrivateThreadPoolDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrivateThreadPoolDataset.java index 637436c7385..307eef240ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrivateThreadPoolDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/PrivateThreadPoolDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. */ +@OpMetadata( + opType = PrivateThreadPoolDataset.OP_NAME, + inputsClass = PrivateThreadPoolDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class PrivateThreadPoolDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private PrivateThreadPoolDataset(Operation operation) { - super(operation); + public PrivateThreadPoolDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = PrivateThreadPoolDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RandomDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RandomDataset.java index c02dc5045d1..9bab5a5f5b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RandomDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RandomDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -46,6 +48,10 @@ * performed is determined by the {@code experimental_optimization.hoist_random_uniform} * option of {@code tf.data.Options}. */ +@OpMetadata( + opType = RandomDataset.OP_NAME, + inputsClass = RandomDataset.Inputs.class +) @Operator( group = "data" ) @@ -58,8 +64,8 @@ public final class RandomDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RandomDataset(Operation operation) { - super(operation); + public RandomDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = RandomDataset.class + ) public static class Inputs extends RawOpInputs { /** * A scalar seed for the random number generator. If either seed or diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RangeDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RangeDataset.java index a77dbc0faa1..fe3b9afedfe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RangeDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RangeDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset with a range of values. Corresponds to python's xrange. */ +@OpMetadata( + opType = RangeDataset.OP_NAME, + inputsClass = RangeDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class RangeDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RangeDataset(Operation operation) { - super(operation); + public RangeDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = RangeDataset.class + ) public static class Inputs extends RawOpInputs { /** * corresponds to start in python's xrange(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RebatchDatasetV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RebatchDatasetV2.java index f2f29dfa889..1d83f3b9a49 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RebatchDatasetV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RebatchDatasetV2.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -41,6 +43,10 @@ * Creates a dataset that rebatches elements from {@code input_dataset} into new batch * sizes. */ +@OpMetadata( + opType = RebatchDatasetV2.OP_NAME, + inputsClass = RebatchDatasetV2.Inputs.class +) @Operator( group = "data" ) @@ -53,8 +59,8 @@ public final class RebatchDatasetV2 extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RebatchDatasetV2(Operation operation) { - super(operation); + public RebatchDatasetV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = RebatchDatasetV2.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ReduceDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ReduceDataset.java index 5c5933a4ecc..26d4cdb1210 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ReduceDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ReduceDataset.java @@ -32,6 +32,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ /** * Reduces the input dataset to a singleton using a reduce function. */ +@OpMetadata( + opType = ReduceDataset.OP_NAME, + inputsClass = ReduceDataset.Inputs.class +) @Operator( group = "data" ) @@ -51,8 +57,8 @@ public final class ReduceDataset extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private ReduceDataset(Operation operation) { - super(operation); + public ReduceDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -148,6 +154,9 @@ public Options useInterOpParallelism(Boolean useInterOpParallelism) { } } + @OpInputsMetadata( + outputsClass = ReduceDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RegisterDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RegisterDataset.java index faad87061c8..73adca2a8a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RegisterDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RegisterDataset.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ /** * Registers a dataset with the tf.data service. */ +@OpMetadata( + opType = RegisterDataset.OP_NAME, + inputsClass = RegisterDataset.Inputs.class +) @Operator( group = "data" ) @@ -46,8 +52,8 @@ public final class RegisterDataset extends RawOp implements Operand { private Output datasetId; - private RegisterDataset(Operation operation) { - super(operation); + public RegisterDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; datasetId = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return datasetId; } + @OpInputsMetadata( + outputsClass = RegisterDataset.class + ) public static class Inputs extends RawOpInputs { /** * The dataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RepeatDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RepeatDataset.java index be3bf8d47fe..5139087b24f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RepeatDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RepeatDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that emits the outputs of {@code input_dataset} {@code count} times. */ +@OpMetadata( + opType = RepeatDataset.OP_NAME, + inputsClass = RepeatDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class RepeatDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RepeatDataset(Operation operation) { - super(operation); + public RepeatDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = RepeatDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SamplingDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SamplingDataset.java index f114ab2c440..8f1658d5034 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SamplingDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SamplingDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -44,6 +46,10 @@ * {@code experimental_optimization.filter_with_random_uniform_fusion} option of * {@code tf.data.Options}. */ +@OpMetadata( + opType = SamplingDataset.OP_NAME, + inputsClass = SamplingDataset.Inputs.class +) @Operator( group = "data" ) @@ -56,8 +62,8 @@ public final class SamplingDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SamplingDataset(Operation operation) { - super(operation); + public SamplingDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -110,6 +116,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SamplingDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SaveDataset.java index b74ce0f5c33..709cf5a7037 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SaveDataset.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -36,6 +38,10 @@ /** * The SaveDataset operation */ +@OpMetadata( + opType = SaveDataset.OP_NAME, + inputsClass = SaveDataset.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class SaveDataset extends RawOp { */ public static final String OP_NAME = "SaveDataset"; - private SaveDataset(Operation operation) { - super(operation); + public SaveDataset(Operation operation) { + super(operation, OP_NAME); } /** @@ -138,6 +144,9 @@ public Options useShardFunc(Boolean useShardFunc) { } } + @OpInputsMetadata( + outputsClass = SaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ScanDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ScanDataset.java index 3507f938ad7..dcb6a1d8309 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ScanDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ScanDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Creates a dataset successively reduces {@code f} over the elements of {@code input_dataset}. */ +@OpMetadata( + opType = ScanDataset.OP_NAME, + inputsClass = ScanDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class ScanDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ScanDataset(Operation operation) { - super(operation); + public ScanDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -168,6 +174,9 @@ public Options useDefaultDevice(Boolean useDefaultDevice) { } } + @OpInputsMetadata( + outputsClass = ScanDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SerializeIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SerializeIterator.java index 9afe58663a0..34466b422f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SerializeIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SerializeIterator.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * Converts the given {@code resource_handle} representing an iterator to a variant tensor. */ +@OpMetadata( + opType = SerializeIterator.OP_NAME, + inputsClass = SerializeIterator.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class SerializeIterator extends RawOp implements Operand { private Output serialized; @SuppressWarnings("unchecked") - private SerializeIterator(Operation operation) { - super(operation); + public SerializeIterator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; serialized = operation.output(outputIdx++); } @@ -123,6 +129,9 @@ public Options externalStatePolicy(Long externalStatePolicy) { } } + @OpInputsMetadata( + outputsClass = SerializeIterator.class + ) public static class Inputs extends RawOpInputs { /** * A handle to an iterator resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SetStatsAggregatorDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SetStatsAggregatorDataset.java index 7a3fbe5988e..59208610c19 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SetStatsAggregatorDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SetStatsAggregatorDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ /** * The SetStatsAggregatorDataset operation */ +@OpMetadata( + opType = SetStatsAggregatorDataset.OP_NAME, + inputsClass = SetStatsAggregatorDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class SetStatsAggregatorDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private SetStatsAggregatorDataset(Operation operation) { - super(operation); + public SetStatsAggregatorDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -104,6 +110,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SetStatsAggregatorDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShardDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShardDataset.java index e380b54d962..951878e72d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShardDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShardDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a {@code Dataset} that includes only 1/{@code num_shards} of this dataset. */ +@OpMetadata( + opType = ShardDataset.OP_NAME, + inputsClass = ShardDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class ShardDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ShardDataset(Operation operation) { - super(operation); + public ShardDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Options requireNonEmpty(Boolean requireNonEmpty) { } } + @OpInputsMetadata( + outputsClass = ShardDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleAndRepeatDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleAndRepeatDataset.java index 47d1782ba34..318f305bd8b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleAndRepeatDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleAndRepeatDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * The ShuffleAndRepeatDatasetV2 operation */ +@OpMetadata( + opType = ShuffleAndRepeatDataset.OP_NAME, + inputsClass = ShuffleAndRepeatDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class ShuffleAndRepeatDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private ShuffleAndRepeatDataset(Operation operation) { - super(operation); + public ShuffleAndRepeatDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -147,6 +153,9 @@ public Options reshuffleEachIteration(Boolean reshuffleEachIteration) { } } + @OpInputsMetadata( + outputsClass = ShuffleAndRepeatDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleDataset.java index dc222f3c559..92cf83d4c4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ShuffleDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * The ShuffleDatasetV3 operation */ +@OpMetadata( + opType = ShuffleDataset.OP_NAME, + inputsClass = ShuffleDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class ShuffleDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ShuffleDataset(Operation operation) { - super(operation); + public ShuffleDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -145,6 +151,9 @@ public Options reshuffleEachIteration(Boolean reshuffleEachIteration) { } } + @OpInputsMetadata( + outputsClass = ShuffleDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SkipDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SkipDataset.java index 366b29b17d2..2734d6ff08d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SkipDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SkipDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that skips {@code count} elements from the {@code input_dataset}. */ +@OpMetadata( + opType = SkipDataset.OP_NAME, + inputsClass = SkipDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class SkipDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SkipDataset(Operation operation) { - super(operation); + public SkipDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SkipDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SleepDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SleepDataset.java index 9d687853a45..f337c5d3fbb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SleepDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SleepDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * The SleepDataset operation */ +@OpMetadata( + opType = SleepDataset.OP_NAME, + inputsClass = SleepDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class SleepDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SleepDataset(Operation operation) { - super(operation); + public SleepDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SleepDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SlidingWindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SlidingWindowDataset.java index 4d46ce5a558..673e100adc6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SlidingWindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SlidingWindowDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that passes a sliding window over {@code input_dataset}. */ +@OpMetadata( + opType = SlidingWindowDataset.OP_NAME, + inputsClass = SlidingWindowDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class SlidingWindowDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private SlidingWindowDataset(Operation operation) { - super(operation); + public SlidingWindowDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SlidingWindowDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDataset.java index 45001f8e541..323fa9514b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -43,6 +45,10 @@ * If not, it will run the preprocessing pipeline as usual, and write out a * snapshot of the data processed for future use. */ +@OpMetadata( + opType = SnapshotDataset.OP_NAME, + inputsClass = SnapshotDataset.Inputs.class +) @Operator( group = "data" ) @@ -55,8 +61,8 @@ public final class SnapshotDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SnapshotDataset(Operation operation) { - super(operation); + public SnapshotDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -257,6 +263,9 @@ public Options hash(Long hash) { } } + @OpInputsMetadata( + outputsClass = SnapshotDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java index 0d07f5021ea..b41f6dbf768 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ /** * The SnapshotDatasetReader operation */ +@OpMetadata( + opType = SnapshotDatasetReader.OP_NAME, + inputsClass = SnapshotDatasetReader.Inputs.class +) public final class SnapshotDatasetReader extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class SnapshotDatasetReader extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private SnapshotDatasetReader(Operation operation) { - super(operation); + public SnapshotDatasetReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -137,6 +143,9 @@ public Options compression(String compression) { } } + @OpInputsMetadata( + outputsClass = SnapshotDatasetReader.class + ) public static class Inputs extends RawOpInputs { /** * The shardDir input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java index 8fa6cd57f37..8167304faba 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * The SnapshotNestedDatasetReader operation */ +@OpMetadata( + opType = SnapshotNestedDatasetReader.OP_NAME, + inputsClass = SnapshotNestedDatasetReader.Inputs.class +) public final class SnapshotNestedDatasetReader extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class SnapshotNestedDatasetReader extends RawOp implements Operand< private Output handle; @SuppressWarnings("unchecked") - private SnapshotNestedDatasetReader(Operation operation) { - super(operation); + public SnapshotNestedDatasetReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SnapshotNestedDatasetReader.class + ) public static class Inputs extends RawOpInputs { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SparseTensorSliceDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SparseTensorSliceDataset.java index ba357f78d13..6bea76cd06e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SparseTensorSliceDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SparseTensorSliceDataset.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -35,6 +37,10 @@ /** * Creates a dataset that splits a SparseTensor into elements row-wise. */ +@OpMetadata( + opType = SparseTensorSliceDataset.OP_NAME, + inputsClass = SparseTensorSliceDataset.Inputs.class +) @Operator( group = "data" ) @@ -47,8 +53,8 @@ public final class SparseTensorSliceDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private SparseTensorSliceDataset(Operation operation) { - super(operation); + public SparseTensorSliceDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SparseTensorSliceDataset.class + ) public static class Inputs extends RawOpInputs { /** * The indices input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SqlDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SqlDataset.java index 7e63d4a1117..81ecba9af64 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SqlDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SqlDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ /** * Creates a dataset that executes a SQL query and emits rows of the result set. */ +@OpMetadata( + opType = SqlDataset.OP_NAME, + inputsClass = SqlDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class SqlDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SqlDataset(Operation operation) { - super(operation); + public SqlDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SqlDataset.class + ) public static class Inputs extends RawOpInputs { /** * The database type. Currently, the only supported type is 'sqlite'. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java index 5e3f2df0d94..d27939fd91b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The StatsAggregatorHandleV2 operation */ +@OpMetadata( + opType = StatsAggregatorHandle.OP_NAME, + inputsClass = StatsAggregatorHandle.Inputs.class +) public final class StatsAggregatorHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class StatsAggregatorHandle extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private StatsAggregatorHandle(Operation operation) { - super(operation); + public StatsAggregatorHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = StatsAggregatorHandle.class + ) public static class Inputs extends RawOpInputs { /** * The container attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java index 5df60be9489..ce220e627d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * Set a summary_writer_interface to record statistics using given stats_aggregator. */ +@OpMetadata( + opType = StatsAggregatorSetSummaryWriter.OP_NAME, + inputsClass = StatsAggregatorSetSummaryWriter.Inputs.class +) public final class StatsAggregatorSetSummaryWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "StatsAggregatorSetSummaryWriter"; - private StatsAggregatorSetSummaryWriter(Operation operation) { - super(operation); + public StatsAggregatorSetSummaryWriter(Operation operation) { + super(operation, OP_NAME); } /** @@ -60,6 +66,9 @@ public static StatsAggregatorSetSummaryWriter create(Scope scope, return new StatsAggregatorSetSummaryWriter(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = StatsAggregatorSetSummaryWriter.class + ) public static class Inputs extends RawOpInputs { /** * The statsAggregator input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeDataset.java index dcd201e8938..83cdbb86ee1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ /** * Creates a dataset that contains {@code count} elements from the {@code input_dataset}. */ +@OpMetadata( + opType = TakeDataset.OP_NAME, + inputsClass = TakeDataset.Inputs.class +) @Operator( group = "data" ) @@ -50,8 +56,8 @@ public final class TakeDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TakeDataset(Operation operation) { - super(operation); + public TakeDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TakeDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeWhileDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeWhileDataset.java index e1fdb783b61..372da7e0af7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeWhileDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TakeWhileDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ *

  • One tensor for each value in {@code other_arguments}.
  • * */ +@OpMetadata( + opType = TakeWhileDataset.OP_NAME, + inputsClass = TakeWhileDataset.Inputs.class +) @Operator( group = "data" ) @@ -56,8 +62,8 @@ public final class TakeWhileDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TakeWhileDataset(Operation operation) { - super(operation); + public TakeWhileDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TakeWhileDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorDataset.java index 96406ec3243..59f597498a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that emits {@code components} as a tuple of tensors once. */ +@OpMetadata( + opType = TensorDataset.OP_NAME, + inputsClass = TensorDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class TensorDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TensorDataset(Operation operation) { - super(operation); + public TensorDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TensorDataset.class + ) public static class Inputs extends RawOpInputs { /** * The components input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorSliceDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorSliceDataset.java index cffdcc7140b..1e74a18e2b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorSliceDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TensorSliceDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that emits each dim-0 slice of {@code components} once. */ +@OpMetadata( + opType = TensorSliceDataset.OP_NAME, + inputsClass = TensorSliceDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class TensorSliceDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TensorSliceDataset(Operation operation) { - super(operation); + public TensorSliceDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TensorSliceDataset.class + ) public static class Inputs extends RawOpInputs { /** * The components input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TextLineDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TextLineDataset.java index 1c64b7a5913..e58a0eaad0a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TextLineDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TextLineDataset.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ /** * Creates a dataset that emits the lines of one or more text files. */ +@OpMetadata( + opType = TextLineDataset.OP_NAME, + inputsClass = TextLineDataset.Inputs.class +) @Operator( group = "data" ) @@ -47,8 +53,8 @@ public final class TextLineDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TextLineDataset(Operation operation) { - super(operation); + public TextLineDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TextLineDataset.class + ) public static class Inputs extends RawOpInputs { /** * A scalar or a vector containing the name(s) of the file(s) to be diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TfRecordDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TfRecordDataset.java index 87345662d72..d46b73555f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TfRecordDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/TfRecordDataset.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ /** * Creates a dataset that emits the records from one or more TFRecord files. */ +@OpMetadata( + opType = TfRecordDataset.OP_NAME, + inputsClass = TfRecordDataset.Inputs.class +) @Operator( group = "data" ) @@ -47,8 +53,8 @@ public final class TfRecordDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TfRecordDataset(Operation operation) { - super(operation); + public TfRecordDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TfRecordDataset.class + ) public static class Inputs extends RawOpInputs { /** * A scalar or vector containing the name(s) of the file(s) to be diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolDataset.java index c9ff2b00969..08e8e93d318 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. */ +@OpMetadata( + opType = ThreadPoolDataset.OP_NAME, + inputsClass = ThreadPoolDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class ThreadPoolDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ThreadPoolDataset(Operation operation) { - super(operation); + public ThreadPoolDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ThreadPoolDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java index 17d867e8b8a..ebb41671d3f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. */ +@OpMetadata( + opType = ThreadPoolHandle.OP_NAME, + inputsClass = ThreadPoolHandle.Inputs.class +) public final class ThreadPoolHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class ThreadPoolHandle extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ThreadPoolHandle(Operation operation) { - super(operation); + public ThreadPoolHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -177,6 +183,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = ThreadPoolHandle.class + ) public static class Inputs extends RawOpInputs { /** * The number of threads in the thread pool. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnbatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnbatchDataset.java index 0abf0d19313..5256a74b953 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnbatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnbatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * A dataset that splits the elements of its input into multiple elements. */ +@OpMetadata( + opType = UnbatchDataset.OP_NAME, + inputsClass = UnbatchDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class UnbatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private UnbatchDataset(Operation operation) { - super(operation); + public UnbatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = UnbatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java index a48c055083a..1036da5df67 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java @@ -31,12 +31,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Uncompresses a compressed dataset element. */ +@OpMetadata( + opType = UncompressElement.OP_NAME, + inputsClass = UncompressElement.Inputs.class +) public final class UncompressElement extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class UncompressElement extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private UncompressElement(Operation operation) { - super(operation); + public UncompressElement(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -94,6 +100,9 @@ public Iterator> iterator() { return (Iterator) components.iterator(); } + @OpInputsMetadata( + outputsClass = UncompressElement.class + ) public static class Inputs extends RawOpInputs { /** * The compressed input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UniqueDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UniqueDataset.java index 86feaca406d..61cb6ac4538 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UniqueDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UniqueDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that contains the unique elements of {@code input_dataset}. */ +@OpMetadata( + opType = UniqueDataset.OP_NAME, + inputsClass = UniqueDataset.Inputs.class +) @Operator( group = "data" ) @@ -49,8 +55,8 @@ public final class UniqueDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private UniqueDataset(Operation operation) { - super(operation); + public UniqueDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = UniqueDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnwrapDatasetVariant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnwrapDatasetVariant.java index eb4ed977823..f2e837a4f8e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnwrapDatasetVariant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UnwrapDatasetVariant.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The UnwrapDatasetVariant operation */ +@OpMetadata( + opType = UnwrapDatasetVariant.OP_NAME, + inputsClass = UnwrapDatasetVariant.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class UnwrapDatasetVariant extends RawOp implements Operand private Output outputHandle; @SuppressWarnings("unchecked") - private UnwrapDatasetVariant(Operation operation) { - super(operation); + public UnwrapDatasetVariant(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = UnwrapDatasetVariant.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowDataset.java index adce99206cc..384bcf89c8e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -74,6 +76,10 @@ * produces {@code {{"a": {0, 1}}, {"a": {2, 3}}}} * */ +@OpMetadata( + opType = WindowDataset.OP_NAME, + inputsClass = WindowDataset.Inputs.class +) @Operator( group = "data" ) @@ -86,8 +92,8 @@ public final class WindowDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private WindowDataset(Operation operation) { - super(operation); + public WindowDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = WindowDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WrapDatasetVariant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WrapDatasetVariant.java index 5a52404b85e..36ade21b269 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WrapDatasetVariant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WrapDatasetVariant.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The WrapDatasetVariant operation */ +@OpMetadata( + opType = WrapDatasetVariant.OP_NAME, + inputsClass = WrapDatasetVariant.Inputs.class +) @Operator( group = "data" ) @@ -45,8 +51,8 @@ public final class WrapDatasetVariant extends RawOp implements Operand { private Output outputHandle; @SuppressWarnings("unchecked") - private WrapDatasetVariant(Operation operation) { - super(operation); + public WrapDatasetVariant(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputHandle = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) outputHandle; } + @OpInputsMetadata( + outputsClass = WrapDatasetVariant.class + ) public static class Inputs extends RawOpInputs { /** * The inputHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ZipDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ZipDataset.java index 874b24717a5..89192eb8559 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ZipDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ZipDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ *

    The size of the resulting dataset will match the size of the smallest input * dataset, and no error will be raised if input datasets have different sizes. */ +@OpMetadata( + opType = ZipDataset.OP_NAME, + inputsClass = ZipDataset.Inputs.class +) @Operator( group = "data" ) @@ -53,8 +59,8 @@ public final class ZipDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ZipDataset(Operation operation) { - super(operation); + public ZipDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ZipDataset.class + ) public static class Inputs extends RawOpInputs { /** * List of {@code N} variant Tensors representing datasets to be zipped together. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java index f2c7acf738f..facd34343cb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The ExperimentalAssertNextDataset operation */ +@OpMetadata( + opType = AssertNextDataset.OP_NAME, + inputsClass = AssertNextDataset.Inputs.class +) public final class AssertNextDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class AssertNextDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private AssertNextDataset(Operation operation) { - super(operation); + public AssertNextDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = AssertNextDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java index 69e5e5a2ba2..419a5c57ee3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ *

    This dataset will throw a NotFound error if we cannot shard the dataset * automatically. */ +@OpMetadata( + opType = AutoShardDataset.OP_NAME, + inputsClass = AutoShardDataset.Inputs.class +) public final class AutoShardDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -52,8 +58,8 @@ public final class AutoShardDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private AutoShardDataset(Operation operation) { - super(operation); + public AutoShardDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -142,6 +148,9 @@ public Options autoShardPolicy(Long autoShardPolicy) { } } + @OpInputsMetadata( + outputsClass = AutoShardDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java index 13fb1165700..5c2ec41eaad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Records the bytes size of each element of {@code input_dataset} in a StatsAggregator. */ +@OpMetadata( + opType = BytesProducedStatsDataset.OP_NAME, + inputsClass = BytesProducedStatsDataset.Inputs.class +) public final class BytesProducedStatsDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class BytesProducedStatsDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private BytesProducedStatsDataset(Operation operation) { - super(operation); + public BytesProducedStatsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = BytesProducedStatsDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java index 996e8d5744e..373ffb23b73 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * The ExperimentalCSVDataset operation */ +@OpMetadata( + opType = CSVDataset.OP_NAME, + inputsClass = CSVDataset.Inputs.class +) public final class CSVDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class CSVDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private CSVDataset(Operation operation) { - super(operation); + public CSVDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -110,6 +116,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = CSVDataset.class + ) public static class Inputs extends RawOpInputs { /** * The filenames input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java index a6d593bda51..cc1d2bb52ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * The ExperimentalChooseFastestDataset operation */ +@OpMetadata( + opType = ChooseFastestDataset.OP_NAME, + inputsClass = ChooseFastestDataset.Inputs.class +) public final class ChooseFastestDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class ChooseFastestDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private ChooseFastestDataset(Operation operation) { - super(operation); + public ChooseFastestDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ChooseFastestDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDatasets input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java index d1c6050a755..2c2c35fd6e3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ * Returns the cardinality of {@code input_dataset}. * Returns the cardinality of {@code input_dataset}. */ +@OpMetadata( + opType = DatasetCardinality.OP_NAME, + inputsClass = DatasetCardinality.Inputs.class +) public final class DatasetCardinality extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class DatasetCardinality extends RawOp implements Operand { private Output cardinality; - private DatasetCardinality(Operation operation) { - super(operation); + public DatasetCardinality(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; cardinality = operation.output(outputIdx++); } @@ -79,6 +85,9 @@ public Output asOutput() { return cardinality; } + @OpInputsMetadata( + outputsClass = DatasetCardinality.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the dataset to return cardinality for. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java index 6ad2bd5ec12..e110a9c7722 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java @@ -26,20 +26,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Writes the given dataset to the given file using the TFRecord format. */ +@OpMetadata( + opType = DatasetToTFRecord.OP_NAME, + inputsClass = DatasetToTFRecord.Inputs.class +) public final class DatasetToTFRecord extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ExperimentalDatasetToTFRecord"; - private DatasetToTFRecord(Operation operation) { - super(operation); + public DatasetToTFRecord(Operation operation) { + super(operation, OP_NAME); } /** @@ -64,6 +70,9 @@ public static DatasetToTFRecord create(Scope scope, Operand inp return new DatasetToTFRecord(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DatasetToTFRecord.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the dataset to write. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java index ae25c651a41..92b8f88135e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that batches input elements into a SparseTensor. */ +@OpMetadata( + opType = DenseToSparseBatchDataset.OP_NAME, + inputsClass = DenseToSparseBatchDataset.Inputs.class +) public final class DenseToSparseBatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class DenseToSparseBatchDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private DenseToSparseBatchDataset(Operation operation) { - super(operation); + public DenseToSparseBatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DenseToSparseBatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * A handle to an input dataset. Must have a single component. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java index 04d9e0bad91..74c7d32e932 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * A substitute for {@code InterleaveDataset} on a fixed list of {@code N} datasets. */ +@OpMetadata( + opType = DirectedInterleaveDataset.OP_NAME, + inputsClass = DirectedInterleaveDataset.Inputs.class +) public final class DirectedInterleaveDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class DirectedInterleaveDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private DirectedInterleaveDataset(Operation operation) { - super(operation); + public DirectedInterleaveDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DirectedInterleaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * A dataset of scalar {@code DT_INT64} elements that determines which of the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java index 735a080bd78..b933563c4b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * Creates a dataset that computes a group-by on {@code input_dataset}. * Creates a dataset that computes a group-by on {@code input_dataset}. */ +@OpMetadata( + opType = GroupByReducerDataset.OP_NAME, + inputsClass = GroupByReducerDataset.Inputs.class +) public final class GroupByReducerDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class GroupByReducerDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private GroupByReducerDataset(Operation operation) { - super(operation); + public GroupByReducerDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -120,6 +126,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = GroupByReducerDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java index 7b4bd4e1333..4acd2c56896 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * Creates a dataset that computes a windowed group-by on {@code input_dataset}. * // TODO(mrry): Support non-int64 keys. */ +@OpMetadata( + opType = GroupByWindowDataset.OP_NAME, + inputsClass = GroupByWindowDataset.Inputs.class +) public final class GroupByWindowDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class GroupByWindowDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private GroupByWindowDataset(Operation operation) { - super(operation); + public GroupByWindowDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -109,6 +115,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = GroupByWindowDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java index 995d5514c4a..fa94ae55315 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Creates a dataset that contains the elements of {@code input_dataset} ignoring errors. */ +@OpMetadata( + opType = IgnoreErrorsDataset.OP_NAME, + inputsClass = IgnoreErrorsDataset.Inputs.class +) public final class IgnoreErrorsDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class IgnoreErrorsDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private IgnoreErrorsDataset(Operation operation) { - super(operation); + public IgnoreErrorsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options logWarning(Boolean logWarning) { } } + @OpInputsMetadata( + outputsClass = IgnoreErrorsDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java index d8232ba0743..fd4d5181956 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Returns the name of the device on which {@code resource} has been placed. */ +@OpMetadata( + opType = IteratorGetDevice.OP_NAME, + inputsClass = IteratorGetDevice.Inputs.class +) public final class IteratorGetDevice extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class IteratorGetDevice extends RawOp implements Operand { private Output device; - private IteratorGetDevice(Operation operation) { - super(operation); + public IteratorGetDevice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; device = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return device; } + @OpInputsMetadata( + outputsClass = IteratorGetDevice.class + ) public static class Inputs extends RawOpInputs { /** * The resource input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java index 5a772ac68c6..d8658c4843f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Records the latency of producing {@code input_dataset} elements in a StatsAggregator. */ +@OpMetadata( + opType = LatencyStatsDataset.OP_NAME, + inputsClass = LatencyStatsDataset.Inputs.class +) public final class LatencyStatsDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class LatencyStatsDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private LatencyStatsDataset(Operation operation) { - super(operation); + public LatencyStatsDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = LatencyStatsDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java index a19e2f1dc41..07e1bec2735 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The ExperimentalLMDBDataset operation */ +@OpMetadata( + opType = LmdbDataset.OP_NAME, + inputsClass = LmdbDataset.Inputs.class +) public final class LmdbDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class LmdbDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private LmdbDataset(Operation operation) { - super(operation); + public LmdbDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = LmdbDataset.class + ) public static class Inputs extends RawOpInputs { /** * The filenames input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java index 81165535555..6b22e495b2d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; @@ -43,6 +45,10 @@ *

    Unlike a "MapDataset", which applies {@code f} sequentially, this dataset invokes up * to {@code batch_size * num_parallel_batches} copies of {@code f} in parallel. */ +@OpMetadata( + opType = MapAndBatchDataset.OP_NAME, + inputsClass = MapAndBatchDataset.Inputs.class +) public final class MapAndBatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -52,8 +58,8 @@ public final class MapAndBatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private MapAndBatchDataset(Operation operation) { - super(operation); + public MapAndBatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -155,6 +161,9 @@ public Options preserveCardinality(Boolean preserveCardinality) { } } + @OpInputsMetadata( + outputsClass = MapAndBatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java index dac1b4bbc4b..64d41ecb3b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java @@ -31,12 +31,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset}. */ +@OpMetadata( + opType = MapDataset.OP_NAME, + inputsClass = MapDataset.Inputs.class +) public final class MapDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class MapDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private MapDataset(Operation operation) { - super(operation); + public MapDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -162,6 +168,9 @@ public Options preserveCardinality(Boolean preserveCardinality) { } } + @OpInputsMetadata( + outputsClass = MapDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java index d56e2d11450..cf3fb557d72 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * The ExperimentalMatchingFilesDataset operation */ +@OpMetadata( + opType = MatchingFilesDataset.OP_NAME, + inputsClass = MatchingFilesDataset.Inputs.class +) public final class MatchingFilesDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class MatchingFilesDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private MatchingFilesDataset(Operation operation) { - super(operation); + public MatchingFilesDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -79,6 +85,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = MatchingFilesDataset.class + ) public static class Inputs extends RawOpInputs { /** * The patterns input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java index a934b3641f2..bd1e4db285d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that overrides the maximum intra-op parallelism. */ +@OpMetadata( + opType = MaxIntraOpParallelismDataset.OP_NAME, + inputsClass = MaxIntraOpParallelismDataset.Inputs.class +) public final class MaxIntraOpParallelismDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class MaxIntraOpParallelismDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private MaxIntraOpParallelismDataset(Operation operation) { - super(operation); + public MaxIntraOpParallelismDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = MaxIntraOpParallelismDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java index a8098c45132..91d544a288d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * The ExperimentalNonSerializableDataset operation */ +@OpMetadata( + opType = NonSerializableDataset.OP_NAME, + inputsClass = NonSerializableDataset.Inputs.class +) public final class NonSerializableDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class NonSerializableDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private NonSerializableDataset(Operation operation) { - super(operation); + public NonSerializableDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = NonSerializableDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java index 67dfc418202..17f05939fa5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; @@ -45,6 +47,10 @@ * allows the training step to proceed so long as some data is available. *

    !! WARNING !! This dataset is not deterministic! */ +@OpMetadata( + opType = ParallelInterleaveDataset.OP_NAME, + inputsClass = ParallelInterleaveDataset.Inputs.class +) public final class ParallelInterleaveDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -54,8 +60,8 @@ public final class ParallelInterleaveDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private ParallelInterleaveDataset(Operation operation) { - super(operation); + public ParallelInterleaveDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -119,6 +125,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ParallelInterleaveDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java index 8805d0ebf5e..cc1bcb7934e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Transforms {@code input_dataset} containing {@code Example} protos as vectors of DT_STRING into a dataset of {@code Tensor} or {@code SparseTensor} objects representing the parsed features. */ +@OpMetadata( + opType = ParseExampleDataset.OP_NAME, + inputsClass = ParseExampleDataset.Inputs.class +) public final class ParseExampleDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class ParseExampleDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ParseExampleDataset(Operation operation) { - super(operation); + public ParseExampleDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -170,6 +176,9 @@ public Options sloppy(Boolean sloppy) { } } + @OpInputsMetadata( + outputsClass = ParseExampleDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java index 36aafe3272e..e5ace0daa57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. */ +@OpMetadata( + opType = PrivateThreadPoolDataset.OP_NAME, + inputsClass = PrivateThreadPoolDataset.Inputs.class +) public final class PrivateThreadPoolDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class PrivateThreadPoolDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private PrivateThreadPoolDataset(Operation operation) { - super(operation); + public PrivateThreadPoolDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = PrivateThreadPoolDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java index 416f4ce7d62..d1423c9ba75 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a Dataset that returns pseudorandom numbers. */ +@OpMetadata( + opType = RandomDataset.OP_NAME, + inputsClass = RandomDataset.Inputs.class +) public final class RandomDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class RandomDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RandomDataset(Operation operation) { - super(operation); + public RandomDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = RandomDataset.class + ) public static class Inputs extends RawOpInputs { /** * A scalar seed for the random number generator. If either seed or diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java index 1ae694d9914..04f8c393f71 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * Creates a dataset that changes the batch size of the dataset to current batch * size // num_replicas. */ +@OpMetadata( + opType = RebatchDataset.OP_NAME, + inputsClass = RebatchDataset.Inputs.class +) public final class RebatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class RebatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RebatchDataset(Operation operation) { - super(operation); + public RebatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -138,6 +144,9 @@ public Options useFallback(Boolean useFallback) { } } + @OpInputsMetadata( + outputsClass = RebatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java index b02550f116a..c3e468cf121 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java @@ -31,12 +31,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Creates a dataset successively reduces {@code f} over the elements of {@code input_dataset}. */ +@OpMetadata( + opType = ScanDataset.OP_NAME, + inputsClass = ScanDataset.Inputs.class +) public final class ScanDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class ScanDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ScanDataset(Operation operation) { - super(operation); + public ScanDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -138,6 +144,9 @@ public Options preserveCardinality(Boolean preserveCardinality) { } } + @OpInputsMetadata( + outputsClass = ScanDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java index 6bcf9d87e29..7ea20feef54 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The ExperimentalSetStatsAggregatorDataset operation */ +@OpMetadata( + opType = SetStatsAggregatorDataset.OP_NAME, + inputsClass = SetStatsAggregatorDataset.Inputs.class +) public final class SetStatsAggregatorDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class SetStatsAggregatorDataset extends RawOp implements Operand handle; @SuppressWarnings("unchecked") - private SetStatsAggregatorDataset(Operation operation) { - super(operation); + public SetStatsAggregatorDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SetStatsAggregatorDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java index ae879e0fbf4..2327119a60e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * The ExperimentalSleepDataset operation */ +@OpMetadata( + opType = SleepDataset.OP_NAME, + inputsClass = SleepDataset.Inputs.class +) public final class SleepDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class SleepDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SleepDataset(Operation operation) { - super(operation); + public SleepDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SleepDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java index 7939710ff4a..922a9b8ad0b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that passes a sliding window over {@code input_dataset}. */ +@OpMetadata( + opType = SlidingWindowDataset.OP_NAME, + inputsClass = SlidingWindowDataset.Inputs.class +) public final class SlidingWindowDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class SlidingWindowDataset extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private SlidingWindowDataset(Operation operation) { - super(operation); + public SlidingWindowDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -102,6 +108,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SlidingWindowDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java index 3a62b4c17bd..23f7b31ef02 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * Creates a dataset that executes a SQL query and emits rows of the result set. */ +@OpMetadata( + opType = SqlDataset.OP_NAME, + inputsClass = SqlDataset.Inputs.class +) public final class SqlDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class SqlDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private SqlDataset(Operation operation) { - super(operation); + public SqlDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = SqlDataset.class + ) public static class Inputs extends RawOpInputs { /** * The database type. Currently, the only supported type is 'sqlite'. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java index ce6bc4bd54b..be406a52156 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * Creates a statistics manager resource. */ +@OpMetadata( + opType = StatsAggregatorHandle.OP_NAME, + inputsClass = StatsAggregatorHandle.Inputs.class +) public final class StatsAggregatorHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class StatsAggregatorHandle extends RawOp implements Operand private Output handle; @SuppressWarnings("unchecked") - private StatsAggregatorHandle(Operation operation) { - super(operation); + public StatsAggregatorHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = StatsAggregatorHandle.class + ) public static class Inputs extends RawOpInputs { /** * The container attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java index 61aeec7600b..02899c3457e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Produces a summary of any statistics recorded by the given statistics manager. */ +@OpMetadata( + opType = StatsAggregatorSummary.OP_NAME, + inputsClass = StatsAggregatorSummary.Inputs.class +) public final class StatsAggregatorSummary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class StatsAggregatorSummary extends RawOp implements Operand summary; - private StatsAggregatorSummary(Operation operation) { - super(operation); + public StatsAggregatorSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return summary; } + @OpInputsMetadata( + outputsClass = StatsAggregatorSummary.class + ) public static class Inputs extends RawOpInputs { /** * The iterator input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java index f11aa6444dd..f14151ab235 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ *

  • One tensor for each value in {@code other_arguments}.
  • * */ +@OpMetadata( + opType = TakeWhileDataset.OP_NAME, + inputsClass = TakeWhileDataset.Inputs.class +) public final class TakeWhileDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -52,8 +58,8 @@ public final class TakeWhileDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private TakeWhileDataset(Operation operation) { - super(operation); + public TakeWhileDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -104,6 +110,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = TakeWhileDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java index 379c0a245a3..6a4eedd6649 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. */ +@OpMetadata( + opType = ThreadPoolDataset.OP_NAME, + inputsClass = ThreadPoolDataset.Inputs.class +) public final class ThreadPoolDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class ThreadPoolDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ThreadPoolDataset(Operation operation) { - super(operation); + public ThreadPoolDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = ThreadPoolDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java index 27a66b8df1a..98d8efa2836 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. */ +@OpMetadata( + opType = ThreadPoolHandle.OP_NAME, + inputsClass = ThreadPoolHandle.Inputs.class +) public final class ThreadPoolHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class ThreadPoolHandle extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private ThreadPoolHandle(Operation operation) { - super(operation); + public ThreadPoolHandle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -177,6 +183,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = ThreadPoolHandle.class + ) public static class Inputs extends RawOpInputs { /** * The number of threads in the thread pool. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java index b427d4e576b..e4aba42d75b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * A dataset that splits the elements of its input into multiple elements. */ +@OpMetadata( + opType = UnbatchDataset.OP_NAME, + inputsClass = UnbatchDataset.Inputs.class +) public final class UnbatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class UnbatchDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private UnbatchDataset(Operation operation) { - super(operation); + public UnbatchDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = UnbatchDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java index 6d81f2ab1a3..c2b0e001679 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Creates a dataset that contains the unique elements of {@code input_dataset}. */ +@OpMetadata( + opType = UniqueDataset.OP_NAME, + inputsClass = UniqueDataset.Inputs.class +) public final class UniqueDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class UniqueDataset extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private UniqueDataset(Operation operation) { - super(operation); + public UniqueDataset(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = UniqueDataset.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java index 644bc1091de..b1db34d171e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CheckNumerics.OP_NAME, + inputsClass = CheckNumerics.Inputs.class +) public final class CheckNumerics extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class CheckNumerics extends RawOp implements Ope private Output output; - private CheckNumerics(Operation operation) { - super(operation); + public CheckNumerics(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = CheckNumerics.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientIdentity.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientIdentity.java index e72c3442a7d..568577a0dcc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientIdentity.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientIdentity.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DebugGradientIdentity.OP_NAME, + inputsClass = DebugGradientIdentity.Inputs.class +) public final class DebugGradientIdentity extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class DebugGradientIdentity extends RawOp implemen private Output output; - private DebugGradientIdentity(Operation operation) { - super(operation); + public DebugGradientIdentity(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = DebugGradientIdentity.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientRefIdentity.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientRefIdentity.java index 1546bc11128..e8ce9ccbd17 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientRefIdentity.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugGradientRefIdentity.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DebugGradientRefIdentity.OP_NAME, + inputsClass = DebugGradientRefIdentity.Inputs.class +) public final class DebugGradientRefIdentity extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class DebugGradientRefIdentity extends RawOp imple private Output output; - private DebugGradientRefIdentity(Operation operation) { - super(operation); + public DebugGradientRefIdentity(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = DebugGradientRefIdentity.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugIdentity.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugIdentity.java index 01167c395e6..5036e9d28af 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugIdentity.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugIdentity.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DebugIdentity.OP_NAME, + inputsClass = DebugIdentity.Inputs.class +) public final class DebugIdentity extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -52,8 +58,8 @@ public final class DebugIdentity extends RawOp implements Opera private Output output; - private DebugIdentity(Operation operation) { - super(operation); + public DebugIdentity(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -316,6 +322,9 @@ public Options tfdbgRunId(String tfdbgRunId) { } } + @OpInputsMetadata( + outputsClass = DebugIdentity.class + ) public static class Inputs extends RawOpInputs> { /** * Input tensor, non-Reference type diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNanCount.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNanCount.java index 9925d851593..59969c5f9f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNanCount.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNanCount.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * Debug NaN Value Counter Op. * Counts number of NaNs in the input tensor, for debugging. */ +@OpMetadata( + opType = DebugNanCount.OP_NAME, + inputsClass = DebugNanCount.Inputs.class +) public final class DebugNanCount extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class DebugNanCount extends RawOp implements Operand { private Output output; - private DebugNanCount(Operation operation) { - super(operation); + public DebugNanCount(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -237,6 +243,9 @@ public Options gatedGrpc(Boolean gatedGrpc) { } } + @OpInputsMetadata( + outputsClass = DebugNanCount.class + ) public static class Inputs extends RawOpInputs { /** * Input tensor, non-Reference type. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNumericsSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNumericsSummary.java index 6dd55b043fc..181e399efbf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNumericsSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/DebugNumericsSummary.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DebugNumericsSummary.OP_NAME, + inputsClass = DebugNumericsSummary.Inputs.class +) public final class DebugNumericsSummary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class DebugNumericsSummary extends RawOp impleme private Output output; - private DebugNumericsSummary(Operation operation) { - super(operation); + public DebugNumericsSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -264,6 +270,9 @@ public Options tensorId(Long tensorId) { } } + @OpInputsMetadata( + outputsClass = DebugNumericsSummary.class + ) public static class Inputs extends RawOpInputs> { /** * Input tensor, to be summarized by the op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java index fe38870830d..2cbb04ff1a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -45,6 +47,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = NcclAllReduce.OP_NAME, + inputsClass = NcclAllReduce.Inputs.class +) public final class NcclAllReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -53,8 +59,8 @@ public final class NcclAllReduce extends RawOp implements Ope private Output data; - private NcclAllReduce(Operation operation) { - super(operation); + public NcclAllReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return data; } + @OpInputsMetadata( + outputsClass = NcclAllReduce.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java index 79c330beaa1..bd70773b6f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = NcclBroadcast.OP_NAME, + inputsClass = NcclBroadcast.Inputs.class +) public final class NcclBroadcast extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class NcclBroadcast extends RawOp implements Ope private Output output; - private NcclBroadcast(Operation operation) { - super(operation); + public NcclBroadcast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = NcclBroadcast.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java index 759c6ea5a92..94d0e2abdb2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = NcclReduce.OP_NAME, + inputsClass = NcclReduce.Inputs.class +) public final class NcclReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class NcclReduce extends RawOp implements Operan private Output data; - private NcclReduce(Operation operation) { - super(operation); + public NcclReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return data; } + @OpInputsMetadata( + outputsClass = NcclReduce.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/AsString.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/AsString.java index a7de0ccc4e0..8de902fee91 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/AsString.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/AsString.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -50,6 +52,10 @@ * * */ +@OpMetadata( + opType = AsString.OP_NAME, + inputsClass = AsString.Inputs.class +) @Operator( group = "dtypes" ) @@ -61,8 +67,8 @@ public final class AsString extends RawOp implements Operand { private Output output; - private AsString(Operation operation) { - super(operation); + public AsString(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -250,6 +256,9 @@ public Options fill(String fill) { } } + @OpInputsMetadata( + outputsClass = AsString.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Cast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Cast.java index cc9cbb10ec2..556ba9cb99c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Cast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Cast.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Cast.OP_NAME, + inputsClass = Cast.Inputs.class +) @Operator( group = "dtypes" ) @@ -48,8 +54,8 @@ public final class Cast extends RawOp implements Operand { private Output y; - private Cast(Operation operation) { - super(operation); + public Cast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options Truncate(Boolean Truncate) { } } + @OpInputsMetadata( + outputsClass = Cast.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Complex.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Complex.java index 86d216c50c0..bcd586c8046 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Complex.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/Complex.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -49,6 +51,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = Complex.OP_NAME, + inputsClass = Complex.Inputs.class +) @Operator( group = "dtypes" ) @@ -60,8 +66,8 @@ public final class Complex extends RawOp implements Operand private Output out; - private Complex(Operation operation) { - super(operation); + public Complex(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -103,6 +109,9 @@ public Output asOutput() { return out; } + @OpInputsMetadata( + outputsClass = Complex.class + ) public static class Inputs extends RawOpInputs> { /** * The real input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java index 44ac418d3d2..fb8c3dc3a3e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; @@ -49,6 +51,10 @@ *

    This matches the behavior of If and While for determining if a tensor counts * as true/false for a branch condition. */ +@OpMetadata( + opType = ToBool.OP_NAME, + inputsClass = ToBool.Inputs.class +) public final class ToBool extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -57,8 +63,8 @@ public final class ToBool extends RawOp implements Operand { private Output output; - private ToBool(Operation operation) { - super(operation); + public ToBool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ToBool.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java index 7b0529beffe..011d1c7793e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -34,6 +36,10 @@ * Aggregates the summary of accumulated stats for the batch. * The summary stats contains gradients and hessians accumulated for each node, feature dimension id and bucket. */ +@OpMetadata( + opType = BoostedTreesAggregateStats.OP_NAME, + inputsClass = BoostedTreesAggregateStats.Inputs.class +) public final class BoostedTreesAggregateStats extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class BoostedTreesAggregateStats extends RawOp implements Operand statsSummary; - private BoostedTreesAggregateStats(Operation operation) { - super(operation); + public BoostedTreesAggregateStats(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; statsSummary = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return statsSummary; } + @OpInputsMetadata( + outputsClass = BoostedTreesAggregateStats.class + ) public static class Inputs extends RawOpInputs { /** * int32; Rank 1 Tensor containing node ids for each example, shape [batch_size]. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java index c9e18b2a292..3d2cf540ff5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * An op that returns a list of float tensors, where each tensor represents the * bucketized values for a single feature. */ +@OpMetadata( + opType = BoostedTreesBucketize.OP_NAME, + inputsClass = BoostedTreesBucketize.Inputs.class +) public final class BoostedTreesBucketize extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class BoostedTreesBucketize extends RawOp implements Iterable> buckets; @SuppressWarnings("unchecked") - private BoostedTreesBucketize(Operation operation) { - super(operation); + public BoostedTreesBucketize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int bucketsLength = operation.outputListLength("buckets"); buckets = Arrays.asList((Output[]) operation.outputList(outputIdx, bucketsLength)); @@ -90,6 +96,9 @@ public Iterator> iterator() { return (Iterator) buckets.iterator(); } + @OpInputsMetadata( + outputsClass = BoostedTreesBucketize.class + ) public static class Inputs extends RawOpInputs { /** * float; List of Rank 1 Tensor each containing float values for a single feature. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java index 8979b37a37c..6157ee489e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ *

    In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). *

    The output shapes are compatible in a way that the first dimension of all tensors are the same and equal to the number of possible split nodes for each feature. */ +@OpMetadata( + opType = BoostedTreesCalculateBestFeatureSplit.OP_NAME, + inputsClass = BoostedTreesCalculateBestFeatureSplit.Inputs.class +) public final class BoostedTreesCalculateBestFeatureSplit extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -58,8 +64,8 @@ public final class BoostedTreesCalculateBestFeatureSplit extends RawOp { private Output splitWithDefaultDirections; - private BoostedTreesCalculateBestFeatureSplit(Operation operation) { - super(operation); + public BoostedTreesCalculateBestFeatureSplit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; nodeIds = operation.output(outputIdx++); gains = operation.output(outputIdx++); @@ -205,6 +211,9 @@ public Options splitType(String splitType) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesCalculateBestFeatureSplit.class + ) public static class Inputs extends RawOpInputs { /** * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java index d733c5f82bc..32e5e7b99ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ *

    In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). *

    The output shapes are compatible in a way that the first dimension of all tensors are the same and equal to the number of possible split nodes for each feature. */ +@OpMetadata( + opType = BoostedTreesCalculateBestFeatureSplitV2.OP_NAME, + inputsClass = BoostedTreesCalculateBestFeatureSplitV2.Inputs.class +) public final class BoostedTreesCalculateBestFeatureSplitV2 extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -61,8 +67,8 @@ public final class BoostedTreesCalculateBestFeatureSplitV2 extends RawOp { private Output splitWithDefaultDirections; - private BoostedTreesCalculateBestFeatureSplitV2(Operation operation) { - super(operation); + public BoostedTreesCalculateBestFeatureSplitV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; nodeIds = operation.output(outputIdx++); gains = operation.output(outputIdx++); @@ -184,6 +190,9 @@ public Output splitWithDefaultDirections() { return splitWithDefaultDirections; } + @OpInputsMetadata( + outputsClass = BoostedTreesCalculateBestFeatureSplitV2.class + ) public static class Inputs extends RawOpInputs { /** * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java index 4d77413901b..524fce9e204 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ *

    The length of output lists are all of the same length, {@code num_features}. * The output shapes are compatible in a way that the first dimension of all tensors of all lists are the same and equal to the number of possible split nodes for each feature. */ +@OpMetadata( + opType = BoostedTreesCalculateBestGainsPerFeature.OP_NAME, + inputsClass = BoostedTreesCalculateBestGainsPerFeature.Inputs.class +) public final class BoostedTreesCalculateBestGainsPerFeature extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -57,8 +63,8 @@ public final class BoostedTreesCalculateBestGainsPerFeature extends RawOp { private List> rightNodeContribsList; @SuppressWarnings("unchecked") - private BoostedTreesCalculateBestGainsPerFeature(Operation operation) { - super(operation); + public BoostedTreesCalculateBestGainsPerFeature(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int nodeIdsListLength = operation.outputListLength("node_ids_list"); nodeIdsList = Arrays.asList((Output[]) operation.outputList(outputIdx, nodeIdsListLength)); @@ -153,6 +159,9 @@ public List> rightNodeContribsList() { return rightNodeContribsList; } + @OpInputsMetadata( + outputsClass = BoostedTreesCalculateBestGainsPerFeature.class + ) public static class Inputs extends RawOpInputs { /** * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java index 320bebfbc4c..d1f0c32bc1c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Calculates the prior from the training data (the bias) and fills in the first node with the logits' prior. Returns a boolean indicating whether to continue centering. */ +@OpMetadata( + opType = BoostedTreesCenterBias.OP_NAME, + inputsClass = BoostedTreesCenterBias.Inputs.class +) public final class BoostedTreesCenterBias extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class BoostedTreesCenterBias extends RawOp implements Operand continueCentering; - private BoostedTreesCenterBias(Operation operation) { - super(operation); + public BoostedTreesCenterBias(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; continueCentering = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return continueCentering; } + @OpInputsMetadata( + outputsClass = BoostedTreesCenterBias.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the tree ensemble. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java index 31df4690605..154efbd1298 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -33,14 +35,18 @@ /** * Creates a tree ensemble model and returns a handle to it. */ +@OpMetadata( + opType = BoostedTreesCreateEnsemble.OP_NAME, + inputsClass = BoostedTreesCreateEnsemble.Inputs.class +) public final class BoostedTreesCreateEnsemble extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesCreateEnsemble"; - private BoostedTreesCreateEnsemble(Operation operation) { - super(operation); + public BoostedTreesCreateEnsemble(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static BoostedTreesCreateEnsemble create(Scope scope, return new BoostedTreesCreateEnsemble(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = BoostedTreesCreateEnsemble.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the tree ensemble resource to be created. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java index 9d5a68ea32d..fae66e5afe4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -33,14 +35,18 @@ /** * Create the Resource for Quantile Streams. */ +@OpMetadata( + opType = BoostedTreesCreateQuantileStreamResource.OP_NAME, + inputsClass = BoostedTreesCreateQuantileStreamResource.Inputs.class +) public final class BoostedTreesCreateQuantileStreamResource extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesCreateQuantileStreamResource"; - private BoostedTreesCreateQuantileStreamResource(Operation operation) { - super(operation); + public BoostedTreesCreateQuantileStreamResource(Operation operation) { + super(operation, OP_NAME); } /** @@ -104,6 +110,9 @@ public Options maxElements(Long maxElements) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesCreateQuantileStreamResource.class + ) public static class Inputs extends RawOpInputs { /** * resource; Handle to quantile stream resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java index fbeb4604ca7..6b1feb1f49d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * Deserializes a serialized tree ensemble config and replaces current tree * ensemble. */ +@OpMetadata( + opType = BoostedTreesDeserializeEnsemble.OP_NAME, + inputsClass = BoostedTreesDeserializeEnsemble.Inputs.class +) public final class BoostedTreesDeserializeEnsemble extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesDeserializeEnsemble"; - private BoostedTreesDeserializeEnsemble(Operation operation) { - super(operation); + public BoostedTreesDeserializeEnsemble(Operation operation) { + super(operation, OP_NAME); } /** @@ -66,6 +72,9 @@ public static BoostedTreesDeserializeEnsemble create(Scope scope, return new BoostedTreesDeserializeEnsemble(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = BoostedTreesDeserializeEnsemble.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the tree ensemble. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java index 8fffb3d912a..2108d30c45f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * Creates a handle to a BoostedTreesEnsembleResource */ +@OpMetadata( + opType = BoostedTreesEnsembleResourceHandleOp.OP_NAME, + inputsClass = BoostedTreesEnsembleResourceHandleOp.Inputs.class +) public final class BoostedTreesEnsembleResourceHandleOp extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class BoostedTreesEnsembleResourceHandleOp extends RawOp implements private Output resource; @SuppressWarnings("unchecked") - private BoostedTreesEnsembleResourceHandleOp(Operation operation) { - super(operation); + public BoostedTreesEnsembleResourceHandleOp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resource = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesEnsembleResourceHandleOp.class + ) public static class Inputs extends RawOpInputs { /** * The container attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java index 0a8a5b87ca1..4fcba1a9e4a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * such as getting split feature ids and logits after each split along the decision * path used to compute directional feature contributions. */ +@OpMetadata( + opType = BoostedTreesExampleDebugOutputs.OP_NAME, + inputsClass = BoostedTreesExampleDebugOutputs.Inputs.class +) public final class BoostedTreesExampleDebugOutputs extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class BoostedTreesExampleDebugOutputs extends RawOp implements Oper private Output examplesDebugOutputsSerialized; - private BoostedTreesExampleDebugOutputs(Operation operation) { - super(operation); + public BoostedTreesExampleDebugOutputs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; examplesDebugOutputsSerialized = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return examplesDebugOutputsSerialized; } + @OpInputsMetadata( + outputsClass = BoostedTreesExampleDebugOutputs.class + ) public static class Inputs extends RawOpInputs { /** * The treeEnsembleHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java index 04ff2d43bd4..c5af68bc7c8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * Each summary Tensor is rank 2, containing summaries (value, weight, min_rank, * max_rank) for a single feature. */ +@OpMetadata( + opType = BoostedTreesFlushQuantileSummaries.OP_NAME, + inputsClass = BoostedTreesFlushQuantileSummaries.Inputs.class +) public final class BoostedTreesFlushQuantileSummaries extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class BoostedTreesFlushQuantileSummaries extends RawOp implements I private List> summaries; @SuppressWarnings("unchecked") - private BoostedTreesFlushQuantileSummaries(Operation operation) { - super(operation); + public BoostedTreesFlushQuantileSummaries(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int summariesLength = operation.outputListLength("summaries"); summaries = Arrays.asList((Output[]) operation.outputList(outputIdx, summariesLength)); @@ -89,6 +95,9 @@ public Iterator> iterator() { return (Iterator) summaries.iterator(); } + @OpInputsMetadata( + outputsClass = BoostedTreesFlushQuantileSummaries.class + ) public static class Inputs extends RawOpInputs { /** * resource handle referring to a QuantileStreamResource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java index 47da16b400c..5359f45b999 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Retrieves the tree ensemble resource stamp token, number of trees and growing statistics. */ +@OpMetadata( + opType = BoostedTreesGetEnsembleStates.OP_NAME, + inputsClass = BoostedTreesGetEnsembleStates.Inputs.class +) public final class BoostedTreesGetEnsembleStates extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class BoostedTreesGetEnsembleStates extends RawOp { private Output lastLayerNodesRange; - private BoostedTreesGetEnsembleStates(Operation operation) { - super(operation); + public BoostedTreesGetEnsembleStates(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; stampToken = operation.output(outputIdx++); numTrees = operation.output(outputIdx++); @@ -123,6 +129,9 @@ public Output lastLayerNodesRange() { return lastLayerNodesRange; } + @OpInputsMetadata( + outputsClass = BoostedTreesGetEnsembleStates.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the tree ensemble. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java index b94831dbbe8..51875c0a714 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -37,6 +39,10 @@ * An op that takes a list of tensors (one tensor per feature) and outputs the * quantile summaries for each tensor. */ +@OpMetadata( + opType = BoostedTreesMakeQuantileSummaries.OP_NAME, + inputsClass = BoostedTreesMakeQuantileSummaries.Inputs.class +) public final class BoostedTreesMakeQuantileSummaries extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class BoostedTreesMakeQuantileSummaries extends RawOp implements It private List> summaries; @SuppressWarnings("unchecked") - private BoostedTreesMakeQuantileSummaries(Operation operation) { - super(operation); + public BoostedTreesMakeQuantileSummaries(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int summariesLength = operation.outputListLength("summaries"); summaries = Arrays.asList((Output[]) operation.outputList(outputIdx, summariesLength)); @@ -92,6 +98,9 @@ public Iterator> iterator() { return (Iterator) summaries.iterator(); } + @OpInputsMetadata( + outputsClass = BoostedTreesMakeQuantileSummaries.class + ) public static class Inputs extends RawOpInputs { /** * float; List of Rank 1 Tensors each containing values for a single feature. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java index 62268ff047f..9bf16045532 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -35,6 +37,10 @@ * Makes the summary of accumulated stats for the batch. * The summary stats contains gradients and hessians accumulated into the corresponding node and bucket for each example. */ +@OpMetadata( + opType = BoostedTreesMakeStatsSummary.OP_NAME, + inputsClass = BoostedTreesMakeStatsSummary.Inputs.class +) public final class BoostedTreesMakeStatsSummary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BoostedTreesMakeStatsSummary extends RawOp implements Operand private Output statsSummary; - private BoostedTreesMakeStatsSummary(Operation operation) { - super(operation); + public BoostedTreesMakeStatsSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; statsSummary = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return statsSummary; } + @OpInputsMetadata( + outputsClass = BoostedTreesMakeStatsSummary.class + ) public static class Inputs extends RawOpInputs { /** * int32 Rank 1 Tensor containing node ids, which each example falls into for the requested layer. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java index 292fbb00e52..52bf9539c72 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * computes the logits. It is designed to be used during prediction. * It traverses all the trees and calculates the final score for each instance. */ +@OpMetadata( + opType = BoostedTreesPredict.OP_NAME, + inputsClass = BoostedTreesPredict.Inputs.class +) public final class BoostedTreesPredict extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class BoostedTreesPredict extends RawOp implements Operand logits; - private BoostedTreesPredict(Operation operation) { - super(operation); + public BoostedTreesPredict(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; logits = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return logits; } + @OpInputsMetadata( + outputsClass = BoostedTreesPredict.class + ) public static class Inputs extends RawOpInputs { /** * The treeEnsembleHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java index ccc5869c5ea..567bbc4e077 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -36,14 +38,18 @@ * summary Tensor is rank 2, containing summaries (value, weight, min_rank, max_rank) * for a single feature. */ +@OpMetadata( + opType = BoostedTreesQuantileStreamResourceAddSummaries.OP_NAME, + inputsClass = BoostedTreesQuantileStreamResourceAddSummaries.Inputs.class +) public final class BoostedTreesQuantileStreamResourceAddSummaries extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesQuantileStreamResourceAddSummaries"; - private BoostedTreesQuantileStreamResourceAddSummaries(Operation operation) { - super(operation); + public BoostedTreesQuantileStreamResourceAddSummaries(Operation operation) { + super(operation, OP_NAME); } /** @@ -66,6 +72,9 @@ public static BoostedTreesQuantileStreamResourceAddSummaries create(Scope scope, return new BoostedTreesQuantileStreamResourceAddSummaries(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = BoostedTreesQuantileStreamResourceAddSummaries.class + ) public static class Inputs extends RawOpInputs { /** * resource handle referring to a QuantileStreamResource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java index 264d083247c..0cc54ad8f71 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * Deserialize bucket boundaries and ready flag into current QuantileAccumulator. * An op that deserializes bucket boundaries and are boundaries ready flag into current QuantileAccumulator. */ +@OpMetadata( + opType = BoostedTreesQuantileStreamResourceDeserialize.OP_NAME, + inputsClass = BoostedTreesQuantileStreamResourceDeserialize.Inputs.class +) public final class BoostedTreesQuantileStreamResourceDeserialize extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesQuantileStreamResourceDeserialize"; - private BoostedTreesQuantileStreamResourceDeserialize(Operation operation) { - super(operation); + public BoostedTreesQuantileStreamResourceDeserialize(Operation operation) { + super(operation, OP_NAME); } /** @@ -64,6 +70,9 @@ public static BoostedTreesQuantileStreamResourceDeserialize create(Scope scope, return new BoostedTreesQuantileStreamResourceDeserialize(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = BoostedTreesQuantileStreamResourceDeserialize.class + ) public static class Inputs extends RawOpInputs { /** * resource handle referring to a QuantileStreamResource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java index cdad72ec124..a3cef6232f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -33,14 +35,18 @@ * Flush the summaries for a quantile stream resource. * An op that flushes the summaries for a quantile stream resource. */ +@OpMetadata( + opType = BoostedTreesQuantileStreamResourceFlush.OP_NAME, + inputsClass = BoostedTreesQuantileStreamResourceFlush.Inputs.class +) public final class BoostedTreesQuantileStreamResourceFlush extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesQuantileStreamResourceFlush"; - private BoostedTreesQuantileStreamResourceFlush(Operation operation) { - super(operation); + public BoostedTreesQuantileStreamResourceFlush(Operation operation) { + super(operation, OP_NAME); } /** @@ -112,6 +118,9 @@ public Options generateQuantiles(Boolean generateQuantiles) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesQuantileStreamResourceFlush.class + ) public static class Inputs extends RawOpInputs { /** * resource handle referring to a QuantileStreamResource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java index 90146726c58..8f8e096c69e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * An op that returns a list of float tensors for a quantile stream resource. Each * tensor is Rank 1 containing bucket boundaries for a single feature. */ +@OpMetadata( + opType = BoostedTreesQuantileStreamResourceGetBucketBoundaries.OP_NAME, + inputsClass = BoostedTreesQuantileStreamResourceGetBucketBoundaries.Inputs.class +) public final class BoostedTreesQuantileStreamResourceGetBucketBoundaries extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class BoostedTreesQuantileStreamResourceGetBucketBoundaries extends private List> bucketBoundaries; @SuppressWarnings("unchecked") - private BoostedTreesQuantileStreamResourceGetBucketBoundaries(Operation operation) { - super(operation); + public BoostedTreesQuantileStreamResourceGetBucketBoundaries(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int bucketBoundariesLength = operation.outputListLength("bucket_boundaries"); bucketBoundaries = Arrays.asList((Output[]) operation.outputList(outputIdx, bucketBoundariesLength)); @@ -88,6 +94,9 @@ public Iterator> iterator() { return (Iterator) bucketBoundaries.iterator(); } + @OpInputsMetadata( + outputsClass = BoostedTreesQuantileStreamResourceGetBucketBoundaries.class + ) public static class Inputs extends RawOpInputs { /** * resource handle referring to a QuantileStreamResource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java index 2d9cc1649f7..4d6c0fdd948 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * Creates a handle to a BoostedTreesQuantileStreamResource. */ +@OpMetadata( + opType = BoostedTreesQuantileStreamResourceHandleOp.OP_NAME, + inputsClass = BoostedTreesQuantileStreamResourceHandleOp.Inputs.class +) public final class BoostedTreesQuantileStreamResourceHandleOp extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class BoostedTreesQuantileStreamResourceHandleOp extends RawOp impl private Output resource; @SuppressWarnings("unchecked") - private BoostedTreesQuantileStreamResourceHandleOp(Operation operation) { - super(operation); + public BoostedTreesQuantileStreamResourceHandleOp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resource = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesQuantileStreamResourceHandleOp.class + ) public static class Inputs extends RawOpInputs { /** * The container attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java index bcec982dda1..5d2c38e2c24 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Serializes the tree ensemble to a proto. */ +@OpMetadata( + opType = BoostedTreesSerializeEnsemble.OP_NAME, + inputsClass = BoostedTreesSerializeEnsemble.Inputs.class +) public final class BoostedTreesSerializeEnsemble extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class BoostedTreesSerializeEnsemble extends RawOp { private Output treeEnsembleSerialized; - private BoostedTreesSerializeEnsemble(Operation operation) { - super(operation); + public BoostedTreesSerializeEnsemble(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; stampToken = operation.output(outputIdx++); treeEnsembleSerialized = operation.output(outputIdx++); @@ -86,6 +92,9 @@ public Output treeEnsembleSerialized() { return treeEnsembleSerialized; } + @OpInputsMetadata( + outputsClass = BoostedTreesSerializeEnsemble.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the tree ensemble. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java index 1a0ae09deb0..42788fc3b7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -34,6 +36,10 @@ * Aggregates the summary of accumulated stats for the batch. * The summary stats contains gradients and hessians accumulated for each node, bucket and dimension id. */ +@OpMetadata( + opType = BoostedTreesSparseAggregateStats.OP_NAME, + inputsClass = BoostedTreesSparseAggregateStats.Inputs.class +) public final class BoostedTreesSparseAggregateStats extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class BoostedTreesSparseAggregateStats extends RawOp { private Output statsSummaryShape; - private BoostedTreesSparseAggregateStats(Operation operation) { - super(operation); + public BoostedTreesSparseAggregateStats(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; statsSummaryIndices = operation.output(outputIdx++); statsSummaryValues = operation.output(outputIdx++); @@ -127,6 +133,9 @@ public Output statsSummaryShape() { return statsSummaryShape; } + @OpInputsMetadata( + outputsClass = BoostedTreesSparseAggregateStats.class + ) public static class Inputs extends RawOpInputs { /** * int32; Rank 1 Tensor containing node ids for each example, shape [batch_size]. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java index a05b6234b12..235c28b624b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ *

    In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). *

    The output shapes are compatible in a way that the first dimension of all tensors are the same and equal to the number of possible split nodes for each feature. */ +@OpMetadata( + opType = BoostedTreesSparseCalculateBestFeatureSplit.OP_NAME, + inputsClass = BoostedTreesSparseCalculateBestFeatureSplit.Inputs.class +) public final class BoostedTreesSparseCalculateBestFeatureSplit extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -58,8 +64,8 @@ public final class BoostedTreesSparseCalculateBestFeatureSplit extends RawOp { private Output splitWithDefaultDirections; - private BoostedTreesSparseCalculateBestFeatureSplit(Operation operation) { - super(operation); + public BoostedTreesSparseCalculateBestFeatureSplit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; nodeIds = operation.output(outputIdx++); gains = operation.output(outputIdx++); @@ -211,6 +217,9 @@ public Options splitType(String splitType) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesSparseCalculateBestFeatureSplit.class + ) public static class Inputs extends RawOpInputs { /** * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java index e3114aacc16..9a9199ce224 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * It traverses the trees starting from cached tree id and cached node id and * calculates the updates to be pushed to the cache. */ +@OpMetadata( + opType = BoostedTreesTrainingPredict.OP_NAME, + inputsClass = BoostedTreesTrainingPredict.Inputs.class +) public final class BoostedTreesTrainingPredict extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class BoostedTreesTrainingPredict extends RawOp { private Output nodeIds; - private BoostedTreesTrainingPredict(Operation operation) { - super(operation); + public BoostedTreesTrainingPredict(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; partialLogits = operation.output(outputIdx++); treeIds = operation.output(outputIdx++); @@ -117,6 +123,9 @@ public Output nodeIds() { return nodeIds; } + @OpInputsMetadata( + outputsClass = BoostedTreesTrainingPredict.class + ) public static class Inputs extends RawOpInputs { /** * The treeEnsembleHandle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java index f4e436613e2..bc41e4ee4c7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -35,14 +37,18 @@ * Updates the tree ensemble by either adding a layer to the last tree being grown * or by starting a new tree. */ +@OpMetadata( + opType = BoostedTreesUpdateEnsemble.OP_NAME, + inputsClass = BoostedTreesUpdateEnsemble.Inputs.class +) public final class BoostedTreesUpdateEnsemble extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesUpdateEnsemble"; - private BoostedTreesUpdateEnsemble(Operation operation) { - super(operation); + public BoostedTreesUpdateEnsemble(Operation operation) { + super(operation, OP_NAME); } /** @@ -92,6 +98,9 @@ public static BoostedTreesUpdateEnsemble create(Scope scope, return new BoostedTreesUpdateEnsemble(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = BoostedTreesUpdateEnsemble.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the ensemble variable. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java index ba73b86fac6..0542c89a28b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -36,14 +38,18 @@ * Updates the tree ensemble by adding a layer to the last tree being grown * or by starting a new tree. */ +@OpMetadata( + opType = BoostedTreesUpdateEnsembleV2.OP_NAME, + inputsClass = BoostedTreesUpdateEnsembleV2.Inputs.class +) public final class BoostedTreesUpdateEnsembleV2 extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "BoostedTreesUpdateEnsembleV2"; - private BoostedTreesUpdateEnsembleV2(Operation operation) { - super(operation); + public BoostedTreesUpdateEnsembleV2(Operation operation) { + super(operation, OP_NAME); } /** @@ -167,6 +173,9 @@ public Options numGroups(Long numGroups) { } } + @OpInputsMetadata( + outputsClass = BoostedTreesUpdateEnsembleV2.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the ensemble variable. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java index 24e8f90f713..68964abb892 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; /** * Checks whether a tree ensemble has been initialized. */ +@OpMetadata( + opType = IsBoostedTreesEnsembleInitialized.OP_NAME, + inputsClass = IsBoostedTreesEnsembleInitialized.Inputs.class +) public final class IsBoostedTreesEnsembleInitialized extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class IsBoostedTreesEnsembleInitialized extends RawOp implements Op private Output isInitialized; - private IsBoostedTreesEnsembleInitialized(Operation operation) { - super(operation); + public IsBoostedTreesEnsembleInitialized(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; isInitialized = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return isInitialized; } + @OpInputsMetadata( + outputsClass = IsBoostedTreesEnsembleInitialized.class + ) public static class Inputs extends RawOpInputs { /** * Handle to the tree ensemble resource. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java index 9a15583349a..21db9a35455 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ * Checks whether a quantile stream has been initialized. * An Op that checks if quantile stream resource is initialized. */ +@OpMetadata( + opType = IsBoostedTreesQuantileStreamResourceInitialized.OP_NAME, + inputsClass = IsBoostedTreesQuantileStreamResourceInitialized.Inputs.class +) public final class IsBoostedTreesQuantileStreamResourceInitialized extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class IsBoostedTreesQuantileStreamResourceInitialized extends RawOp private Output isInitialized; - private IsBoostedTreesQuantileStreamResourceInitialized(Operation operation) { - super(operation); + public IsBoostedTreesQuantileStreamResourceInitialized(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; isInitialized = operation.output(outputIdx++); } @@ -79,6 +85,9 @@ public Output asOutput() { return isInitialized; } + @OpInputsMetadata( + outputsClass = IsBoostedTreesQuantileStreamResourceInitialized.class + ) public static class Inputs extends RawOpInputs { /** * resource; The reference to quantile stream resource handle. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustContrast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustContrast.java index 3e234b8c767..091153bfd77 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustContrast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustContrast.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AdjustContrast.OP_NAME, + inputsClass = AdjustContrast.Inputs.class +) @Operator( group = "image" ) @@ -55,8 +61,8 @@ public final class AdjustContrast extends RawOp implements Op private Output output; - private AdjustContrast(Operation operation) { - super(operation); + public AdjustContrast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = AdjustContrast.class + ) public static class Inputs extends RawOpInputs> { /** * Images to adjust. At least 3-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustHue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustHue.java index 99a4e8a28f6..b30f864511e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustHue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustHue.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AdjustHue.OP_NAME, + inputsClass = AdjustHue.Inputs.class +) @Operator( group = "image" ) @@ -53,8 +59,8 @@ public final class AdjustHue extends RawOp implements Operand private Output output; - private AdjustHue(Operation operation) { - super(operation); + public AdjustHue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = AdjustHue.class + ) public static class Inputs extends RawOpInputs> { /** * Images to adjust. At least 3-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustSaturation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustSaturation.java index 42949c86350..1bc2f095dcf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustSaturation.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/AdjustSaturation.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AdjustSaturation.OP_NAME, + inputsClass = AdjustSaturation.Inputs.class +) @Operator( group = "image" ) @@ -53,8 +59,8 @@ public final class AdjustSaturation extends RawOp implements private Output output; - private AdjustSaturation(Operation operation) { - super(operation); + public AdjustSaturation(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = AdjustSaturation.class + ) public static class Inputs extends RawOpInputs> { /** * Images to adjust. At least 3-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CombinedNonMaxSuppression.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CombinedNonMaxSuppression.java index ffe6982c45b..ea56fa18101 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CombinedNonMaxSuppression.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CombinedNonMaxSuppression.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -47,6 +49,10 @@ * The output of this operation is the final boxes, scores and classes tensor * returned after performing non_max_suppression. */ +@OpMetadata( + opType = CombinedNonMaxSuppression.OP_NAME, + inputsClass = CombinedNonMaxSuppression.Inputs.class +) @Operator( group = "image" ) @@ -64,8 +70,8 @@ public final class CombinedNonMaxSuppression extends RawOp { private Output validDetections; - private CombinedNonMaxSuppression(Operation operation) { - super(operation); + public CombinedNonMaxSuppression(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; nmsedBoxes = operation.output(outputIdx++); nmsedScores = operation.output(outputIdx++); @@ -228,6 +234,9 @@ public Options clipBoxes(Boolean clipBoxes) { } } + @OpInputsMetadata( + outputsClass = CombinedNonMaxSuppression.class + ) public static class Inputs extends RawOpInputs { /** * A 4-D float tensor of shape {@code [batch_size, num_boxes, q, 4]}. If {@code q} is 1 then diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResize.java index bafdeda6193..7ab253d6b3a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -50,6 +52,10 @@ * {@code tf.image.resize_nearest_neighbor()}(depends on the {@code method} argument) with * {@code align_corners=True}. */ +@OpMetadata( + opType = CropAndResize.OP_NAME, + inputsClass = CropAndResize.Inputs.class +) @Operator( group = "image" ) @@ -61,8 +67,8 @@ public final class CropAndResize extends RawOp implements Operand { private Output crops; - private CropAndResize(Operation operation) { - super(operation); + public CropAndResize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; crops = operation.output(outputIdx++); } @@ -188,6 +194,9 @@ public Options extrapolationValue(Float extrapolationValue) { } } + @OpInputsMetadata( + outputsClass = CropAndResize.class + ) public static class Inputs extends RawOpInputs { /** * A 4-D tensor of shape {@code [batch, image_height, image_width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradBoxes.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradBoxes.java index fb4da6130e8..69bda4c6326 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradBoxes.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradBoxes.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -36,6 +38,10 @@ /** * Computes the gradient of the crop_and_resize op wrt the input boxes tensor. */ +@OpMetadata( + opType = CropAndResizeGradBoxes.OP_NAME, + inputsClass = CropAndResizeGradBoxes.Inputs.class +) @Operator( group = "image" ) @@ -47,8 +53,8 @@ public final class CropAndResizeGradBoxes extends RawOp implements Operand output; - private CropAndResizeGradBoxes(Operation operation) { - super(operation); + public CropAndResizeGradBoxes(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -139,6 +145,9 @@ public Options method(String method) { } } + @OpInputsMetadata( + outputsClass = CropAndResizeGradBoxes.class + ) public static class Inputs extends RawOpInputs { /** * A 4-D tensor of shape {@code [num_boxes, crop_height, crop_width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradImage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradImage.java index 0f664e36fad..b754d75bc6e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradImage.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/CropAndResizeGradImage.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CropAndResizeGradImage.OP_NAME, + inputsClass = CropAndResizeGradImage.Inputs.class +) @Operator( group = "image" ) @@ -50,8 +56,8 @@ public final class CropAndResizeGradImage extends RawOp imple private Output output; - private CropAndResizeGradImage(Operation operation) { - super(operation); + public CropAndResizeGradImage(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options method(String method) { } } + @OpInputsMetadata( + outputsClass = CropAndResizeGradImage.class + ) public static class Inputs extends RawOpInputs> { /** * A 4-D tensor of shape {@code [num_boxes, crop_height, crop_width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeAndCropJpeg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeAndCropJpeg.java index 3ddd568feb4..35a36990a63 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeAndCropJpeg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeAndCropJpeg.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -50,6 +52,10 @@ *

    It is equivalent to a combination of decode and crop, but much faster by only * decoding partial jpeg image. */ +@OpMetadata( + opType = DecodeAndCropJpeg.OP_NAME, + inputsClass = DecodeAndCropJpeg.Inputs.class +) @Operator( group = "image" ) @@ -61,8 +67,8 @@ public final class DecodeAndCropJpeg extends RawOp implements Operand { private Output image; - private DecodeAndCropJpeg(Operation operation) { - super(operation); + public DecodeAndCropJpeg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; image = operation.output(outputIdx++); } @@ -283,6 +289,9 @@ public Options dctMethod(String dctMethod) { } } + @OpInputsMetadata( + outputsClass = DecodeAndCropJpeg.class + ) public static class Inputs extends RawOpInputs { /** * 0-D. The JPEG-encoded image. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeBmp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeBmp.java index 12995a44d7d..404059fba22 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeBmp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeBmp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.TUint8; @@ -42,6 +44,10 @@ *

  • 4: output an RGBA image.
  • * */ +@OpMetadata( + opType = DecodeBmp.OP_NAME, + inputsClass = DecodeBmp.Inputs.class +) @Operator( group = "image" ) @@ -53,8 +59,8 @@ public final class DecodeBmp extends RawOp implements Operand { private Output image; - private DecodeBmp(Operation operation) { - super(operation); + public DecodeBmp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; image = operation.output(outputIdx++); } @@ -128,6 +134,9 @@ public Options channels(Long channels) { } } + @OpInputsMetadata( + outputsClass = DecodeBmp.class + ) public static class Inputs extends RawOpInputs { /** * 0-D. The BMP-encoded image. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeGif.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeGif.java index f9ea2e04703..27d250f6e18 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeGif.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeGif.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.TUint8; @@ -42,6 +44,10 @@ *

    This op also supports decoding JPEGs and PNGs, though it is cleaner to use * {@code tf.io.decode_image}. */ +@OpMetadata( + opType = DecodeGif.OP_NAME, + inputsClass = DecodeGif.Inputs.class +) @Operator( group = "image" ) @@ -53,8 +59,8 @@ public final class DecodeGif extends RawOp implements Operand { private Output image; - private DecodeGif(Operation operation) { - super(operation); + public DecodeGif(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; image = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return image; } + @OpInputsMetadata( + outputsClass = DecodeGif.class + ) public static class Inputs extends RawOpInputs { /** * 0-D. The GIF-encoded image. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeImage.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeImage.java index 6a1720d52bf..a1eb37a26f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeImage.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeImage.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -54,6 +56,10 @@ * * @param data type for {@code image} output */ +@OpMetadata( + opType = DecodeImage.OP_NAME, + inputsClass = DecodeImage.Inputs.class +) @Operator( group = "image" ) @@ -65,8 +71,8 @@ public final class DecodeImage extends RawOp implements Opera private Output image; - private DecodeImage(Operation operation) { - super(operation); + public DecodeImage(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; image = operation.output(outputIdx++); } @@ -193,6 +199,9 @@ public Options expandAnimations(Boolean expandAnimations) { } } + @OpInputsMetadata( + outputsClass = DecodeImage.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D. The encoded image bytes. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeJpeg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeJpeg.java index ecf649a14c7..674522acfa3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeJpeg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodeJpeg.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.TUint8; @@ -49,6 +51,10 @@ *

    This op also supports decoding PNGs and non-animated GIFs since the interface is * the same, though it is cleaner to use {@code tf.io.decode_image}. */ +@OpMetadata( + opType = DecodeJpeg.OP_NAME, + inputsClass = DecodeJpeg.Inputs.class +) @Operator( group = "image" ) @@ -60,8 +66,8 @@ public final class DecodeJpeg extends RawOp implements Operand { private Output image; - private DecodeJpeg(Operation operation) { - super(operation); + public DecodeJpeg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; image = operation.output(outputIdx++); } @@ -279,6 +285,9 @@ public Options dctMethod(String dctMethod) { } } + @OpInputsMetadata( + outputsClass = DecodeJpeg.class + ) public static class Inputs extends RawOpInputs { /** * 0-D. The JPEG-encoded image. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodePng.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodePng.java index d0e58936125..bf75085ca17 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodePng.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DecodePng.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -52,6 +54,10 @@ * * @param data type for {@code image} output */ +@OpMetadata( + opType = DecodePng.OP_NAME, + inputsClass = DecodePng.Inputs.class +) @Operator( group = "image" ) @@ -63,8 +69,8 @@ public final class DecodePng extends RawOp implements Operand private Output image; - private DecodePng(Operation operation) { - super(operation); + public DecodePng(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; image = operation.output(outputIdx++); } @@ -158,6 +164,9 @@ public Options channels(Long channels) { } } + @OpInputsMetadata( + outputsClass = DecodePng.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D. The PNG-encoded image. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DrawBoundingBoxes.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DrawBoundingBoxes.java index 44bc9156c06..2621dd4f6e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DrawBoundingBoxes.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/DrawBoundingBoxes.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DrawBoundingBoxes.OP_NAME, + inputsClass = DrawBoundingBoxes.Inputs.class +) @Operator( group = "image" ) @@ -57,8 +63,8 @@ public final class DrawBoundingBoxes extends RawOp implements private Output output; - private DrawBoundingBoxes(Operation operation) { - super(operation); + public DrawBoundingBoxes(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = DrawBoundingBoxes.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, depth]}. A batch of images. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpeg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpeg.java index dd8f13b3189..45ffd3227c3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpeg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpeg.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.TUint8; @@ -50,6 +52,10 @@ *

  • 3: Output an RGB image.
  • * */ +@OpMetadata( + opType = EncodeJpeg.OP_NAME, + inputsClass = EncodeJpeg.Inputs.class +) @Operator( group = "image" ) @@ -61,8 +67,8 @@ public final class EncodeJpeg extends RawOp implements Operand { private Output contents; - private EncodeJpeg(Operation operation) { - super(operation); + public EncodeJpeg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; contents = operation.output(outputIdx++); } @@ -346,6 +352,9 @@ public Options xmpMetadata(String xmpMetadata) { } } + @OpInputsMetadata( + outputsClass = EncodeJpeg.class + ) public static class Inputs extends RawOpInputs { /** * 3-D with shape {@code [height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpegVariableQuality.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpegVariableQuality.java index 7b2b8faef21..0800236434d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpegVariableQuality.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodeJpegVariableQuality.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -37,6 +39,10 @@ * {@code image} is a 3-D uint8 Tensor of shape {@code [height, width, channels]}. * {@code quality} is an int32 jpeg compression quality value between 0 and 100. */ +@OpMetadata( + opType = EncodeJpegVariableQuality.OP_NAME, + inputsClass = EncodeJpegVariableQuality.Inputs.class +) @Operator( group = "image" ) @@ -48,8 +54,8 @@ public final class EncodeJpegVariableQuality extends RawOp implements Operand contents; - private EncodeJpegVariableQuality(Operation operation) { - super(operation); + public EncodeJpegVariableQuality(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; contents = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return contents; } + @OpInputsMetadata( + outputsClass = EncodeJpegVariableQuality.class + ) public static class Inputs extends RawOpInputs { /** * Images to adjust. At least 3-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodePng.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodePng.java index 81d56bc23a5..810617e128f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodePng.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/EncodePng.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -46,6 +48,10 @@ * default or a value from 0 to 9. 9 is the highest compression level, generating * the smallest output, but is slower. */ +@OpMetadata( + opType = EncodePng.OP_NAME, + inputsClass = EncodePng.Inputs.class +) @Operator( group = "image" ) @@ -57,8 +63,8 @@ public final class EncodePng extends RawOp implements Operand { private Output contents; - private EncodePng(Operation operation) { - super(operation); + public EncodePng(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; contents = operation.output(outputIdx++); } @@ -133,6 +139,9 @@ public Options compression(Long compression) { } } + @OpInputsMetadata( + outputsClass = EncodePng.class + ) public static class Inputs extends RawOpInputs { /** * 3-D with shape {@code [height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java index 45677ff83f1..fa8eba3e65b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -52,6 +54,10 @@ * numbers of pixels. * */ +@OpMetadata( + opType = ExtractGlimpse.OP_NAME, + inputsClass = ExtractGlimpse.Inputs.class +) public final class ExtractGlimpse extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -60,8 +66,8 @@ public final class ExtractGlimpse extends RawOp implements Operand { private Output glimpse; - private ExtractGlimpse(Operation operation) { - super(operation); + public ExtractGlimpse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; glimpse = operation.output(outputIdx++); } @@ -233,6 +239,9 @@ public Options noise(String noise) { } } + @OpInputsMetadata( + outputsClass = ExtractGlimpse.class + ) public static class Inputs extends RawOpInputs { /** * A 4-D float tensor of shape {@code [batch_size, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractImagePatches.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractImagePatches.java index 80353b0ab42..3c168cc0e87 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractImagePatches.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractImagePatches.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code patches} output */ +@OpMetadata( + opType = ExtractImagePatches.OP_NAME, + inputsClass = ExtractImagePatches.Inputs.class +) @Operator( group = "image" ) @@ -48,8 +54,8 @@ public final class ExtractImagePatches extends RawOp implements private Output patches; - private ExtractImagePatches(Operation operation) { - super(operation); + public ExtractImagePatches(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; patches = operation.output(outputIdx++); } @@ -114,6 +120,9 @@ public Output asOutput() { return patches; } + @OpInputsMetadata( + outputsClass = ExtractImagePatches.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D Tensor with shape {@code [batch, in_rows, in_cols, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractJpegShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractJpegShape.java index d5c7f09060c..15b168d66ad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractJpegShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractJpegShape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code image_shape} output */ +@OpMetadata( + opType = ExtractJpegShape.OP_NAME, + inputsClass = ExtractJpegShape.Inputs.class +) @Operator( group = "image" ) @@ -51,8 +57,8 @@ public final class ExtractJpegShape extends RawOp implements private Output imageShape; - private ExtractJpegShape(Operation operation) { - super(operation); + public ExtractJpegShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; imageShape = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return imageShape; } + @OpInputsMetadata( + outputsClass = ExtractJpegShape.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D. The JPEG-encoded image. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java index d0548098263..57643c3603a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * `roi_probabilities`: probability scores of each roi in 'rois', a 2D tensor of shape [Batch,post_nms_topn], padded with 0 if needed, sorted by scores. * */ +@OpMetadata( + opType = GenerateBoundingBoxProposals.OP_NAME, + inputsClass = GenerateBoundingBoxProposals.Inputs.class +) public final class GenerateBoundingBoxProposals extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -56,8 +62,8 @@ public final class GenerateBoundingBoxProposals extends RawOp { private Output roiProbabilities; - private GenerateBoundingBoxProposals(Operation operation) { - super(operation); + public GenerateBoundingBoxProposals(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; rois = operation.output(outputIdx++); roiProbabilities = operation.output(outputIdx++); @@ -154,6 +160,9 @@ public Options postNmsTopn(Long postNmsTopn) { } } + @OpInputsMetadata( + outputsClass = GenerateBoundingBoxProposals.class + ) public static class Inputs extends RawOpInputs { /** * A 4-D float tensor of shape {@code [num_images, height, width, num_achors]} containing scores of the boxes for given anchors, can be unsorted. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/HsvToRgb.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/HsvToRgb.java index 8bbb40efb55..5d2dcea1595 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/HsvToRgb.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/HsvToRgb.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = HsvToRgb.OP_NAME, + inputsClass = HsvToRgb.Inputs.class +) @Operator( group = "image" ) @@ -51,8 +57,8 @@ public final class HsvToRgb extends RawOp implements Operand< private Output output; - private HsvToRgb(Operation operation) { - super(operation); + public HsvToRgb(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = HsvToRgb.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D or higher rank. HSV data to convert. Last dimension must be size 3. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java index 9312eaed59f..99f87eb44dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * * @param data type for {@code transformed_images} output */ +@OpMetadata( + opType = ImageProjectiveTransformV2.OP_NAME, + inputsClass = ImageProjectiveTransformV2.Inputs.class +) public final class ImageProjectiveTransformV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class ImageProjectiveTransformV2 extends RawOp i private Output transformedImages; - private ImageProjectiveTransformV2(Operation operation) { - super(operation); + public ImageProjectiveTransformV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; transformedImages = operation.output(outputIdx++); } @@ -137,6 +143,9 @@ public Options fillMode(String fillMode) { } } + @OpInputsMetadata( + outputsClass = ImageProjectiveTransformV2.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java index 229c0eaf656..0704a533e4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * * @param data type for {@code transformed_images} output */ +@OpMetadata( + opType = ImageProjectiveTransformV3.OP_NAME, + inputsClass = ImageProjectiveTransformV3.Inputs.class +) public final class ImageProjectiveTransformV3 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class ImageProjectiveTransformV3 extends RawOp i private Output transformedImages; - private ImageProjectiveTransformV3(Operation operation) { - super(operation); + public ImageProjectiveTransformV3(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; transformedImages = operation.output(outputIdx++); } @@ -139,6 +145,9 @@ public Options fillMode(String fillMode) { } } + @OpInputsMetadata( + outputsClass = ImageProjectiveTransformV3.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java index d8b061bc045..8c6cf3b47fc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -36,6 +38,10 @@ * the list of candidate centers. For each point, the k centers that have least L2 * distance to it are computed. */ +@OpMetadata( + opType = NearestNeighbors.OP_NAME, + inputsClass = NearestNeighbors.Inputs.class +) public final class NearestNeighbors extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class NearestNeighbors extends RawOp { private Output nearestCenterDistances; - private NearestNeighbors(Operation operation) { - super(operation); + public NearestNeighbors(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; nearestCenterIndices = operation.output(outputIdx++); nearestCenterDistances = operation.output(outputIdx++); @@ -95,6 +101,9 @@ public Output nearestCenterDistances() { return nearestCenterDistances; } + @OpInputsMetadata( + outputsClass = NearestNeighbors.class + ) public static class Inputs extends RawOpInputs { /** * Matrix of shape (n, d). Rows are assumed to be input points. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppression.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppression.java index 910c2048d7f..704e99ea7df 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppression.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppression.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -59,6 +61,10 @@ * * @param data type for {@code selected_scores} output */ +@OpMetadata( + opType = NonMaxSuppression.OP_NAME, + inputsClass = NonMaxSuppression.Inputs.class +) @Operator( group = "image" ) @@ -74,8 +80,8 @@ public final class NonMaxSuppression extends RawOp { private Output validOutputs; - private NonMaxSuppression(Operation operation) { - super(operation); + public NonMaxSuppression(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; selectedIndices = operation.output(outputIdx++); selectedScores = operation.output(outputIdx++); @@ -190,6 +196,9 @@ public Options padToMaxOutputSize(Boolean padToMaxOutputSize) { } } + @OpInputsMetadata( + outputsClass = NonMaxSuppression.class + ) public static class Inputs extends RawOpInputs> { /** * A 2-D float tensor of shape {@code [num_boxes, 4]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppressionWithOverlaps.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppressionWithOverlaps.java index f12b180716f..9325d24b151 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppressionWithOverlaps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NonMaxSuppressionWithOverlaps.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * overlaps, scores, max_output_size, overlap_threshold, score_threshold) * selected_boxes = tf.gather(boxes, selected_indices) */ +@OpMetadata( + opType = NonMaxSuppressionWithOverlaps.OP_NAME, + inputsClass = NonMaxSuppressionWithOverlaps.Inputs.class +) @Operator( group = "image" ) @@ -57,8 +63,8 @@ public final class NonMaxSuppressionWithOverlaps extends RawOp implements Operan private Output selectedIndices; - private NonMaxSuppressionWithOverlaps(Operation operation) { - super(operation); + public NonMaxSuppressionWithOverlaps(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; selectedIndices = operation.output(outputIdx++); } @@ -109,6 +115,9 @@ public Output asOutput() { return selectedIndices; } + @OpInputsMetadata( + outputsClass = NonMaxSuppressionWithOverlaps.class + ) public static class Inputs extends RawOpInputs { /** * A 2-D float tensor of shape {@code [num_boxes, num_boxes]} representing diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/QuantizedResizeBilinear.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/QuantizedResizeBilinear.java index ef8936425db..5d0d7ba5c59 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/QuantizedResizeBilinear.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/QuantizedResizeBilinear.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -39,6 +41,10 @@ * * @param data type for {@code resized_images} output */ +@OpMetadata( + opType = QuantizedResizeBilinear.OP_NAME, + inputsClass = QuantizedResizeBilinear.Inputs.class +) @Operator( group = "image" ) @@ -54,8 +60,8 @@ public final class QuantizedResizeBilinear extends RawOp { private Output outMax; - private QuantizedResizeBilinear(Operation operation) { - super(operation); + public QuantizedResizeBilinear(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resizedImages = operation.output(outputIdx++); outMin = operation.output(outputIdx++); @@ -183,6 +189,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = QuantizedResizeBilinear.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RandomCrop.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RandomCrop.java index 580d7a893a1..1fb7066fe16 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RandomCrop.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RandomCrop.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomCrop.OP_NAME, + inputsClass = RandomCrop.Inputs.class +) @Operator( group = "image" ) @@ -53,8 +59,8 @@ public final class RandomCrop extends RawOp implements Operan private Output output; - private RandomCrop(Operation operation) { - super(operation); + public RandomCrop(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -162,6 +168,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomCrop.class + ) public static class Inputs extends RawOpInputs> { /** * 3-D of shape {@code [height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeArea.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeArea.java index fde981548d8..15f34355739 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeArea.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeArea.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -45,6 +47,10 @@ * input pixel's contribution to the average is weighted by the fraction of its * area that intersects the footprint. This is the same as OpenCV's INTER_AREA. */ +@OpMetadata( + opType = ResizeArea.OP_NAME, + inputsClass = ResizeArea.Inputs.class +) @Operator( group = "image" ) @@ -56,8 +62,8 @@ public final class ResizeArea extends RawOp implements Operand { private Output resizedImages; - private ResizeArea(Operation operation) { - super(operation); + public ResizeArea(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resizedImages = operation.output(outputIdx++); } @@ -138,6 +144,9 @@ public Options alignCorners(Boolean alignCorners) { } } + @OpInputsMetadata( + outputsClass = ResizeArea.class + ) public static class Inputs extends RawOpInputs { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubic.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubic.java index df9a6dd7977..3a4668dee5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubic.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubic.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -37,6 +39,10 @@ * Resize {@code images} to {@code size} using bicubic interpolation. * Input images can be of different types but output images are always float. */ +@OpMetadata( + opType = ResizeBicubic.OP_NAME, + inputsClass = ResizeBicubic.Inputs.class +) @Operator( group = "image" ) @@ -48,8 +54,8 @@ public final class ResizeBicubic extends RawOp implements Operand { private Output resizedImages; - private ResizeBicubic(Operation operation) { - super(operation); + public ResizeBicubic(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resizedImages = operation.output(outputIdx++); } @@ -156,6 +162,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = ResizeBicubic.class + ) public static class Inputs extends RawOpInputs { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java index 1e9aabf70d3..efa4aa46714 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ResizeBicubicGrad.OP_NAME, + inputsClass = ResizeBicubicGrad.Inputs.class +) public final class ResizeBicubicGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class ResizeBicubicGrad extends RawOp implements private Output output; - private ResizeBicubicGrad(Operation operation) { - super(operation); + public ResizeBicubicGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = ResizeBicubicGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinear.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinear.java index d42617f0aaa..761bb72ae82 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinear.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinear.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -37,6 +39,10 @@ * Resize {@code images} to {@code size} using bilinear interpolation. * Input images can be of different types but output images are always float. */ +@OpMetadata( + opType = ResizeBilinear.OP_NAME, + inputsClass = ResizeBilinear.Inputs.class +) @Operator( group = "image" ) @@ -48,8 +54,8 @@ public final class ResizeBilinear extends RawOp implements Operand { private Output resizedImages; - private ResizeBilinear(Operation operation) { - super(operation); + public ResizeBilinear(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resizedImages = operation.output(outputIdx++); } @@ -156,6 +162,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = ResizeBilinear.class + ) public static class Inputs extends RawOpInputs { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java index 25312aaccc1..c49e71c5998 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ResizeBilinearGrad.OP_NAME, + inputsClass = ResizeBilinearGrad.Inputs.class +) public final class ResizeBilinearGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class ResizeBilinearGrad extends RawOp implement private Output output; - private ResizeBilinearGrad(Operation operation) { - super(operation); + public ResizeBilinearGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = ResizeBilinearGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighbor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighbor.java index cc81189c0d6..0966a28bf36 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighbor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighbor.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ * * @param data type for {@code resized_images} output */ +@OpMetadata( + opType = ResizeNearestNeighbor.OP_NAME, + inputsClass = ResizeNearestNeighbor.Inputs.class +) @Operator( group = "image" ) @@ -48,8 +54,8 @@ public final class ResizeNearestNeighbor extends RawOp implem private Output resizedImages; - private ResizeNearestNeighbor(Operation operation) { - super(operation); + public ResizeNearestNeighbor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resizedImages = operation.output(outputIdx++); } @@ -157,6 +163,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = ResizeNearestNeighbor.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java index 0aafc8d0f5a..71d8dd28257 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ResizeNearestNeighborGrad.OP_NAME, + inputsClass = ResizeNearestNeighborGrad.Inputs.class +) public final class ResizeNearestNeighborGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class ResizeNearestNeighborGrad extends RawOp im private Output output; - private ResizeNearestNeighborGrad(Operation operation) { - super(operation); + public ResizeNearestNeighborGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -153,6 +159,9 @@ public Options halfPixelCenters(Boolean halfPixelCenters) { } } + @OpInputsMetadata( + outputsClass = ResizeNearestNeighborGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RgbToHsv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RgbToHsv.java index f06112cddb1..af5fe0fcbe4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RgbToHsv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/RgbToHsv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +59,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RgbToHsv.OP_NAME, + inputsClass = RgbToHsv.Inputs.class +) @Operator( group = "image" ) @@ -68,8 +74,8 @@ public final class RgbToHsv extends RawOp implements Operand< private Output output; - private RgbToHsv(Operation operation) { - super(operation); + public RgbToHsv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RgbToHsv.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D or higher rank. RGB data to convert. Last dimension must be size 3. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/SampleDistortedBoundingBox.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/SampleDistortedBoundingBox.java index bee6684766e..0d85ae63003 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/SampleDistortedBoundingBox.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/SampleDistortedBoundingBox.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -71,6 +73,10 @@ * * @param data type for {@code begin} output */ +@OpMetadata( + opType = SampleDistortedBoundingBox.OP_NAME, + inputsClass = SampleDistortedBoundingBox.Inputs.class +) @Operator( group = "image" ) @@ -86,8 +92,8 @@ public final class SampleDistortedBoundingBox extends RawOp { private Output bboxes; - private SampleDistortedBoundingBox(Operation operation) { - super(operation); + public SampleDistortedBoundingBox(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; begin = operation.output(outputIdx++); sizeOutput = operation.output(outputIdx++); @@ -390,6 +396,9 @@ public Options useImageIfNoBoundingBoxes(Boolean useImageIfNoBoundingBoxes) { } } + @OpInputsMetadata( + outputsClass = SampleDistortedBoundingBox.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D, containing {@code [height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslate.java index 5fd3d8f194c..3c1fe9bacf2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslate.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslate.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -36,6 +38,10 @@ /** * The ScaleAndTranslate operation */ +@OpMetadata( + opType = ScaleAndTranslate.OP_NAME, + inputsClass = ScaleAndTranslate.Inputs.class +) @Operator( group = "image" ) @@ -47,8 +53,8 @@ public final class ScaleAndTranslate extends RawOp implements Operand private Output resizedImages; - private ScaleAndTranslate(Operation operation) { - super(operation); + public ScaleAndTranslate(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resizedImages = operation.output(outputIdx++); } @@ -156,6 +162,9 @@ public Options antialias(Boolean antialias) { } } + @OpInputsMetadata( + outputsClass = ScaleAndTranslate.class + ) public static class Inputs extends RawOpInputs { /** * The images input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java index ad62f12fabe..b817b6640b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ScaleAndTranslateGrad.OP_NAME, + inputsClass = ScaleAndTranslateGrad.Inputs.class +) public final class ScaleAndTranslateGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class ScaleAndTranslateGrad extends RawOp implem private Output output; - private ScaleAndTranslateGrad(Operation operation) { - super(operation); + public ScaleAndTranslateGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options antialias(Boolean antialias) { } } + @OpInputsMetadata( + outputsClass = ScaleAndTranslateGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The grads input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/StatelessSampleDistortedBoundingBox.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/StatelessSampleDistortedBoundingBox.java index 9bdb6969f7d..d2a89901edc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/StatelessSampleDistortedBoundingBox.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/StatelessSampleDistortedBoundingBox.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -96,6 +98,10 @@ * * @param data type for {@code begin} output */ +@OpMetadata( + opType = StatelessSampleDistortedBoundingBox.OP_NAME, + inputsClass = StatelessSampleDistortedBoundingBox.Inputs.class +) @Operator( group = "image" ) @@ -111,8 +117,8 @@ public final class StatelessSampleDistortedBoundingBox extend private Output bboxes; - private StatelessSampleDistortedBoundingBox(Operation operation) { - super(operation); + public StatelessSampleDistortedBoundingBox(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; begin = operation.output(outputIdx++); sizeOutput = operation.output(outputIdx++); @@ -362,6 +368,9 @@ public Options useImageIfNoBoundingBoxes(Boolean useImageIfNoBoundingBoxes) { } } + @OpInputsMetadata( + outputsClass = StatelessSampleDistortedBoundingBox.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D, containing {@code [height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeBase64.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeBase64.java index 4ebc7750b91..473246f9290 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeBase64.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeBase64.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ * Input may or may not have padding at the end. See EncodeBase64 for padding. * Web-safe means that input must use - and _ instead of + and /. */ +@OpMetadata( + opType = DecodeBase64.OP_NAME, + inputsClass = DecodeBase64.Inputs.class +) @Operator( group = "io" ) @@ -46,8 +52,8 @@ public final class DecodeBase64 extends RawOp implements Operand { private Output output; - private DecodeBase64(Operation operation) { - super(operation); + public DecodeBase64(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = DecodeBase64.class + ) public static class Inputs extends RawOpInputs { /** * Base64 strings to decode. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCompressed.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCompressed.java index f732ce500eb..a194b362695 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCompressed.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCompressed.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * each element containing the decompressed data from the corresponding * element in {@code bytes}. */ +@OpMetadata( + opType = DecodeCompressed.OP_NAME, + inputsClass = DecodeCompressed.Inputs.class +) @Operator( group = "io" ) @@ -49,8 +55,8 @@ public final class DecodeCompressed extends RawOp implements Operand { private Output output; - private DecodeCompressed(Operation operation) { - super(operation); + public DecodeCompressed(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options compressionType(String compressionType) { } } + @OpInputsMetadata( + outputsClass = DecodeCompressed.class + ) public static class Inputs extends RawOpInputs { /** * A Tensor of string which is compressed. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCsv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCsv.java index 20073756dc6..3e3db5ec2d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCsv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeCsv.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -41,6 +43,10 @@ * (https://tools.ietf.org/html/rfc4180) * Note that we allow leading and trailing spaces with int or float field. */ +@OpMetadata( + opType = DecodeCsv.OP_NAME, + inputsClass = DecodeCsv.Inputs.class +) @Operator( group = "io" ) @@ -53,8 +59,8 @@ public final class DecodeCsv extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private DecodeCsv(Operation operation) { - super(operation); + public DecodeCsv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -244,6 +250,9 @@ public Options selectCols(Long... selectCols) { } } + @OpInputsMetadata( + outputsClass = DecodeCsv.class + ) public static class Inputs extends RawOpInputs { /** * Each string is a record/row in the csv and all records should have diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeJsonExample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeJsonExample.java index e27ec07fce8..84e4eb4ad97 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeJsonExample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeJsonExample.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -40,6 +42,10 @@ * {@code Example.SerializeToString()}) suitable for conversion to tensors with * {@code tf.io.parse_example}. */ +@OpMetadata( + opType = DecodeJsonExample.OP_NAME, + inputsClass = DecodeJsonExample.Inputs.class +) @Operator( group = "io" ) @@ -51,8 +57,8 @@ public final class DecodeJsonExample extends RawOp implements Operand { private Output binaryExamples; - private DecodeJsonExample(Operation operation) { - super(operation); + public DecodeJsonExample(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; binaryExamples = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return binaryExamples; } + @OpInputsMetadata( + outputsClass = DecodeJsonExample.class + ) public static class Inputs extends RawOpInputs { /** * Each string is a JSON object serialized according to the JSON diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodePaddedRaw.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodePaddedRaw.java index 693527476d2..35570618220 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodePaddedRaw.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodePaddedRaw.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DecodePaddedRaw.OP_NAME, + inputsClass = DecodePaddedRaw.Inputs.class +) @Operator( group = "io" ) @@ -50,8 +56,8 @@ public final class DecodePaddedRaw extends RawOp implements O private Output output; - private DecodePaddedRaw(Operation operation) { - super(operation); + public DecodePaddedRaw(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -137,6 +143,9 @@ public Options littleEndian(Boolean littleEndian) { } } + @OpInputsMetadata( + outputsClass = DecodePaddedRaw.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor of string to be decoded. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeRaw.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeRaw.java index 393cd50e94b..56f2c093b0b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeRaw.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DecodeRaw.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DecodeRaw.OP_NAME, + inputsClass = DecodeRaw.Inputs.class +) @Operator( group = "io" ) @@ -49,8 +55,8 @@ public final class DecodeRaw extends RawOp implements Operand output; - private DecodeRaw(Operation operation) { - super(operation); + public DecodeRaw(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -134,6 +140,9 @@ public Options littleEndian(Boolean littleEndian) { } } + @OpInputsMetadata( + outputsClass = DecodeRaw.class + ) public static class Inputs extends RawOpInputs> { /** * All the elements must have the same length. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DeserializeManySparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DeserializeManySparse.java index 9c900322d81..ae3949d3284 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DeserializeManySparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DeserializeManySparse.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -78,6 +80,10 @@ * * @param data type for {@code sparse_values} output */ +@OpMetadata( + opType = DeserializeManySparse.OP_NAME, + inputsClass = DeserializeManySparse.Inputs.class +) @Operator( group = "io" ) @@ -93,8 +99,8 @@ public final class DeserializeManySparse extends RawOp { private Output sparseShape; - private DeserializeManySparse(Operation operation) { - super(operation); + public DeserializeManySparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseIndices = operation.output(outputIdx++); sparseValues = operation.output(outputIdx++); @@ -149,6 +155,9 @@ public Output sparseShape() { return sparseShape; } + @OpInputsMetadata( + outputsClass = DeserializeManySparse.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D, The {@code N} serialized {@code SparseTensor} objects. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/EncodeBase64.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/EncodeBase64.java index 2f632b33937..daddfee308a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/EncodeBase64.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/EncodeBase64.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * link above. *

    Web-safe means that the encoder uses - and _ instead of + and /. */ +@OpMetadata( + opType = EncodeBase64.OP_NAME, + inputsClass = EncodeBase64.Inputs.class +) @Operator( group = "io" ) @@ -49,8 +55,8 @@ public final class EncodeBase64 extends RawOp implements Operand { private Output output; - private EncodeBase64(Operation operation) { - super(operation); + public EncodeBase64(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -124,6 +130,9 @@ public Options pad(Boolean pad) { } } + @OpInputsMetadata( + outputsClass = EncodeBase64.class + ) public static class Inputs extends RawOpInputs { /** * Strings to be encoded. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FifoQueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FifoQueue.java index 0a2a8f21957..36cff78458e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FifoQueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FifoQueue.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * A queue that produces elements in first-in first-out order. */ +@OpMetadata( + opType = FifoQueue.OP_NAME, + inputsClass = FifoQueue.Inputs.class +) @Operator( group = "io" ) @@ -49,8 +55,8 @@ public final class FifoQueue extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private FifoQueue(Operation operation) { - super(operation); + public FifoQueue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -247,6 +253,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = FifoQueue.class + ) public static class Inputs extends RawOpInputs { /** * The type of each component in a value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FixedLengthRecordReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FixedLengthRecordReader.java index 105ab72ea3d..ad16397019e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FixedLengthRecordReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FixedLengthRecordReader.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * A Reader that outputs fixed-length records from a file. */ +@OpMetadata( + opType = FixedLengthRecordReader.OP_NAME, + inputsClass = FixedLengthRecordReader.Inputs.class +) @Operator( group = "io" ) @@ -45,8 +51,8 @@ public final class FixedLengthRecordReader extends RawOp implements Operand readerHandle; @SuppressWarnings("unchecked") - private FixedLengthRecordReader(Operation operation) { - super(operation); + public FixedLengthRecordReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; readerHandle = operation.output(outputIdx++); } @@ -259,6 +265,9 @@ public Options encoding(String encoding) { } } + @OpInputsMetadata( + outputsClass = FixedLengthRecordReader.class + ) public static class Inputs extends RawOpInputs { /** * Number of bytes in the header, defaults to 0. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/IdentityReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/IdentityReader.java index a58cc80a838..453a17b7ccf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/IdentityReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/IdentityReader.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * To use, enqueue strings in a Queue. ReaderRead will take the front * work string and output (work, work). */ +@OpMetadata( + opType = IdentityReader.OP_NAME, + inputsClass = IdentityReader.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class IdentityReader extends RawOp implements Operand { private Output readerHandle; @SuppressWarnings("unchecked") - private IdentityReader(Operation operation) { - super(operation); + public IdentityReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; readerHandle = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = IdentityReader.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this reader is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/LmdbReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/LmdbReader.java index f90a320cf2b..9be472413d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/LmdbReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/LmdbReader.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** * A Reader that outputs the records from a LMDB file. */ +@OpMetadata( + opType = LmdbReader.OP_NAME, + inputsClass = LmdbReader.Inputs.class +) @Operator( group = "io" ) @@ -44,8 +50,8 @@ public final class LmdbReader extends RawOp implements Operand { private Output readerHandle; - private LmdbReader(Operation operation) { - super(operation); + public LmdbReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; readerHandle = operation.output(outputIdx++); } @@ -147,6 +153,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = LmdbReader.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this reader is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/MatchingFiles.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/MatchingFiles.java index 8c522a5ede2..217f5a80629 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/MatchingFiles.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/MatchingFiles.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -36,6 +38,10 @@ * basename portion of the pattern, not in the directory portion. * Note also that the order of filenames returned is deterministic. */ +@OpMetadata( + opType = MatchingFiles.OP_NAME, + inputsClass = MatchingFiles.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class MatchingFiles extends RawOp implements Operand { private Output filenames; - private MatchingFiles(Operation operation) { - super(operation); + public MatchingFiles(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; filenames = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return filenames; } + @OpInputsMetadata( + outputsClass = MatchingFiles.class + ) public static class Inputs extends RawOpInputs { /** * Shell wildcard pattern(s). Scalar or vector of type string. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PaddingFifoQueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PaddingFifoQueue.java index 82846004272..f07f71eb1d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PaddingFifoQueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PaddingFifoQueue.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * to 0 in the shape attr. In this case DequeueMany will pad up to the maximum * size of any given element in the minibatch. See below for details. */ +@OpMetadata( + opType = PaddingFifoQueue.OP_NAME, + inputsClass = PaddingFifoQueue.Inputs.class +) @Operator( group = "io" ) @@ -52,8 +58,8 @@ public final class PaddingFifoQueue extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private PaddingFifoQueue(Operation operation) { - super(operation); + public PaddingFifoQueue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -266,6 +272,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = PaddingFifoQueue.class + ) public static class Inputs extends RawOpInputs { /** * The type of each component in a value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseExample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseExample.java index e35ba099a79..6cd578647d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseExample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseExample.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -40,6 +42,10 @@ /** * Transforms a vector of tf.Example protos (as strings) into typed tensors. */ +@OpMetadata( + opType = ParseExample.OP_NAME, + inputsClass = ParseExample.Inputs.class +) @Operator( group = "io" ) @@ -62,8 +68,8 @@ public final class ParseExample extends RawOp { private List> raggedRowSplits; @SuppressWarnings("unchecked") - private ParseExample(Operation operation) { - super(operation); + public ParseExample(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int sparseIndicesLength = operation.outputListLength("sparse_indices"); sparseIndices = Arrays.asList((Output[]) operation.outputList(outputIdx, sparseIndicesLength)); @@ -222,6 +228,9 @@ public List> raggedRowSplits() { return raggedRowSplits; } + @OpInputsMetadata( + outputsClass = ParseExample.class + ) public static class Inputs extends RawOpInputs { /** * A scalar or vector containing binary serialized Example protos. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSequenceExample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSequenceExample.java index 27bd0d243d2..c01ca5f36b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSequenceExample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSequenceExample.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -42,6 +44,10 @@ * Transforms a vector of tf.io.SequenceExample protos (as strings) into * typed tensors. */ +@OpMetadata( + opType = ParseSequenceExample.OP_NAME, + inputsClass = ParseSequenceExample.Inputs.class +) @Operator( group = "io" ) @@ -80,8 +86,8 @@ public final class ParseSequenceExample extends RawOp { private List> featureListRaggedInnerSplits; @SuppressWarnings("unchecked") - private ParseSequenceExample(Operation operation) { - super(operation); + public ParseSequenceExample(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int contextSparseIndicesLength = operation.outputListLength("context_sparse_indices"); contextSparseIndices = Arrays.asList((Output[]) operation.outputList(outputIdx, contextSparseIndicesLength)); @@ -560,6 +566,9 @@ public Options featureListDenseShapes(Shape... featureListDenseShapes) { } } + @OpInputsMetadata( + outputsClass = ParseSequenceExample.class + ) public static class Inputs extends RawOpInputs { /** * A scalar or vector containing binary serialized SequenceExample protos. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleExample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleExample.java index 2c0b92b6acb..c316dd66768 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleExample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleExample.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * Transforms a tf.Example proto (as a string) into typed tensors. */ +@OpMetadata( + opType = ParseSingleExample.OP_NAME, + inputsClass = ParseSingleExample.Inputs.class +) @Operator( group = "io" ) @@ -57,8 +63,8 @@ public final class ParseSingleExample extends RawOp { private List> denseValues; @SuppressWarnings("unchecked") - private ParseSingleExample(Operation operation) { - super(operation); + public ParseSingleExample(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int sparseIndicesLength = operation.outputListLength("sparse_indices"); sparseIndices = Arrays.asList((Output[]) operation.outputList(outputIdx, sparseIndicesLength)); @@ -175,6 +181,9 @@ public List> denseValues() { return denseValues; } + @OpInputsMetadata( + outputsClass = ParseSingleExample.class + ) public static class Inputs extends RawOpInputs { /** * A vector containing a batch of binary serialized Example protos. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleSequenceExample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleSequenceExample.java index c5d98799d5a..884bf9f6b49 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleSequenceExample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseSingleSequenceExample.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ /** * Transforms a scalar brain.SequenceExample proto (as strings) into typed tensors. */ +@OpMetadata( + opType = ParseSingleSequenceExample.OP_NAME, + inputsClass = ParseSingleSequenceExample.Inputs.class +) @Operator( group = "io" ) @@ -65,8 +71,8 @@ public final class ParseSingleSequenceExample extends RawOp { private List> featureListDenseValues; @SuppressWarnings("unchecked") - private ParseSingleSequenceExample(Operation operation) { - super(operation); + public ParseSingleSequenceExample(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int contextSparseIndicesLength = operation.outputListLength("context_sparse_indices"); contextSparseIndices = Arrays.asList((Output[]) operation.outputList(outputIdx, contextSparseIndicesLength)); @@ -488,6 +494,9 @@ public Options featureListDenseShapes(Shape... featureListDenseShapes) { } } + @OpInputsMetadata( + outputsClass = ParseSingleSequenceExample.class + ) public static class Inputs extends RawOpInputs { /** * A scalar containing a binary serialized SequenceExample proto. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseTensor.java index cf71f17624d..e9819f92af6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ParseTensor.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ParseTensor.OP_NAME, + inputsClass = ParseTensor.Inputs.class +) @Operator( group = "io" ) @@ -49,8 +55,8 @@ public final class ParseTensor extends RawOp implements Operand private Output output; - private ParseTensor(Operation operation) { - super(operation); + public ParseTensor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ParseTensor.class + ) public static class Inputs extends RawOpInputs> { /** * A scalar string containing a serialized TensorProto proto. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PriorityQueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PriorityQueue.java index 3908003de08..a486a2c9983 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PriorityQueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/PriorityQueue.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * and DequeueMany) on a PriorityQueue will all require (resp. output) one extra * entry in their input (resp. output) lists. */ +@OpMetadata( + opType = PriorityQueue.OP_NAME, + inputsClass = PriorityQueue.Inputs.class +) @Operator( group = "io" ) @@ -54,8 +60,8 @@ public final class PriorityQueue extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private PriorityQueue(Operation operation) { - super(operation); + public PriorityQueue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -198,6 +204,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = PriorityQueue.class + ) public static class Inputs extends RawOpInputs { /** * The type of each component in a value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueClose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueClose.java index bc2bfa3fff4..6ad758b2def 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueClose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueClose.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * sufficient elements remain in the queue. Subsequent Dequeue(Many) * operations that would block will fail immediately. */ +@OpMetadata( + opType = QueueClose.OP_NAME, + inputsClass = QueueClose.Inputs.class +) @Operator( group = "io" ) @@ -46,8 +52,8 @@ public final class QueueClose extends RawOp { */ public static final String OP_NAME = "QueueCloseV2"; - private QueueClose(Operation operation) { - super(operation); + public QueueClose(Operation operation) { + super(operation, OP_NAME); } /** @@ -108,6 +114,9 @@ public Options cancelPendingEnqueues(Boolean cancelPendingEnqueues) { } } + @OpInputsMetadata( + outputsClass = QueueClose.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeue.java index 469de83d736..92f3d42bea7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeue.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ *

    N.B. If the queue is empty, this operation will block until an element * has been dequeued (or 'timeout_ms' elapses, if specified). */ +@OpMetadata( + opType = QueueDequeue.OP_NAME, + inputsClass = QueueDequeue.Inputs.class +) @Operator( group = "io" ) @@ -54,8 +60,8 @@ public final class QueueDequeue extends RawOp implements Iterable private List> components; @SuppressWarnings("unchecked") - private QueueDequeue(Operation operation) { - super(operation); + public QueueDequeue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -139,6 +145,9 @@ public Options timeoutMs(Long timeoutMs) { } } + @OpInputsMetadata( + outputsClass = QueueDequeue.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueMany.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueMany.java index ed61ce786c2..658875387f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueMany.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueMany.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ *

    N.B. If the queue is empty, this operation will block until {@code n} elements * have been dequeued (or 'timeout_ms' elapses, if specified). */ +@OpMetadata( + opType = QueueDequeueMany.OP_NAME, + inputsClass = QueueDequeueMany.Inputs.class +) @Operator( group = "io" ) @@ -60,8 +66,8 @@ public final class QueueDequeueMany extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private QueueDequeueMany(Operation operation) { - super(operation); + public QueueDequeueMany(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -147,6 +153,9 @@ public Options timeoutMs(Long timeoutMs) { } } + @OpInputsMetadata( + outputsClass = QueueDequeueMany.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueUpTo.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueUpTo.java index 91bec450c7b..8073d83c6f8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueUpTo.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueDequeueUpTo.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -52,6 +54,10 @@ * the tuples stored in the given queue, and output {@code i} is the ith * component of the dequeued tuple. */ +@OpMetadata( + opType = QueueDequeueUpTo.OP_NAME, + inputsClass = QueueDequeueUpTo.Inputs.class +) @Operator( group = "io" ) @@ -64,8 +70,8 @@ public final class QueueDequeueUpTo extends RawOp implements Iterable> components; @SuppressWarnings("unchecked") - private QueueDequeueUpTo(Operation operation) { - super(operation); + public QueueDequeueUpTo(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int componentsLength = operation.outputListLength("components"); components = Arrays.asList(operation.outputList(outputIdx, componentsLength)); @@ -151,6 +157,9 @@ public Options timeoutMs(Long timeoutMs) { } } + @OpInputsMetadata( + outputsClass = QueueDequeueUpTo.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueue.java index 1872c43e0a9..e712a9db04c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueue.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ *

    N.B. If the queue is full, this operation will block until the given * element has been enqueued (or 'timeout_ms' elapses, if specified). */ +@OpMetadata( + opType = QueueEnqueue.OP_NAME, + inputsClass = QueueEnqueue.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class QueueEnqueue extends RawOp { */ public static final String OP_NAME = "QueueEnqueueV2"; - private QueueEnqueue(Operation operation) { - super(operation); + public QueueEnqueue(Operation operation) { + super(operation, OP_NAME); } /** @@ -113,6 +119,9 @@ public Options timeoutMs(Long timeoutMs) { } } + @OpInputsMetadata( + outputsClass = QueueEnqueue.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueueMany.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueueMany.java index d0c17369fca..c93c37ca288 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueueMany.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueEnqueueMany.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ *

    N.B. If the queue is full, this operation will block until the given * elements have been enqueued (or 'timeout_ms' elapses, if specified). */ +@OpMetadata( + opType = QueueEnqueueMany.OP_NAME, + inputsClass = QueueEnqueueMany.Inputs.class +) @Operator( group = "io" ) @@ -50,8 +56,8 @@ public final class QueueEnqueueMany extends RawOp { */ public static final String OP_NAME = "QueueEnqueueManyV2"; - private QueueEnqueueMany(Operation operation) { - super(operation); + public QueueEnqueueMany(Operation operation) { + super(operation, OP_NAME); } /** @@ -117,6 +123,9 @@ public Options timeoutMs(Long timeoutMs) { } } + @OpInputsMetadata( + outputsClass = QueueEnqueueMany.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueIsClosed.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueIsClosed.java index 29fdf801889..422e3a0cea1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueIsClosed.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueIsClosed.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * This operation returns true if the queue is closed and false if the queue * is open. */ +@OpMetadata( + opType = QueueIsClosed.OP_NAME, + inputsClass = QueueIsClosed.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class QueueIsClosed extends RawOp implements Operand { private Output isClosed; - private QueueIsClosed(Operation operation) { - super(operation); + public QueueIsClosed(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; isClosed = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return isClosed; } + @OpInputsMetadata( + outputsClass = QueueIsClosed.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueSize.java index 2505096d056..b668f011af8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/QueueSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Computes the number of elements in the given queue. */ +@OpMetadata( + opType = QueueSize.OP_NAME, + inputsClass = QueueSize.Inputs.class +) @Operator( group = "io" ) @@ -45,8 +51,8 @@ public final class QueueSize extends RawOp implements Operand { private Output output; - private QueueSize(Operation operation) { - super(operation); + public QueueSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = QueueSize.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/RandomShuffleQueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/RandomShuffleQueue.java index c0f90c4cd81..ca4714bb4f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/RandomShuffleQueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/RandomShuffleQueue.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ /** * A queue that randomizes the order of elements. */ +@OpMetadata( + opType = RandomShuffleQueue.OP_NAME, + inputsClass = RandomShuffleQueue.Inputs.class +) @Operator( group = "io" ) @@ -49,8 +55,8 @@ public final class RandomShuffleQueue extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private RandomShuffleQueue(Operation operation) { - super(operation); + public RandomShuffleQueue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -331,6 +337,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = RandomShuffleQueue.class + ) public static class Inputs extends RawOpInputs { /** * The type of each component in a value. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReadFile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReadFile.java index f7d467d0cb7..ce25b5aedf8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReadFile.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReadFile.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** * Reads and outputs the entire contents of the input filename. */ +@OpMetadata( + opType = ReadFile.OP_NAME, + inputsClass = ReadFile.Inputs.class +) @Operator( group = "io" ) @@ -44,8 +50,8 @@ public final class ReadFile extends RawOp implements Operand { private Output contents; - private ReadFile(Operation operation) { - super(operation); + public ReadFile(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; contents = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return contents; } + @OpInputsMetadata( + outputsClass = ReadFile.class + ) public static class Inputs extends RawOpInputs { /** * The filename input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumRecordsProduced.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumRecordsProduced.java index 7f9c067dc03..40971f29991 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumRecordsProduced.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumRecordsProduced.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * This is the same as the number of ReaderRead executions that have * succeeded. */ +@OpMetadata( + opType = ReaderNumRecordsProduced.OP_NAME, + inputsClass = ReaderNumRecordsProduced.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class ReaderNumRecordsProduced extends RawOp implements Operand recordsProduced; - private ReaderNumRecordsProduced(Operation operation) { - super(operation); + public ReaderNumRecordsProduced(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; recordsProduced = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return recordsProduced; } + @OpInputsMetadata( + outputsClass = ReaderNumRecordsProduced.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a Reader. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumWorkUnitsCompleted.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumWorkUnitsCompleted.java index 0271bde551e..fe23ec6b424 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumWorkUnitsCompleted.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderNumWorkUnitsCompleted.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Returns the number of work units this Reader has finished processing. */ +@OpMetadata( + opType = ReaderNumWorkUnitsCompleted.OP_NAME, + inputsClass = ReaderNumWorkUnitsCompleted.Inputs.class +) @Operator( group = "io" ) @@ -45,8 +51,8 @@ public final class ReaderNumWorkUnitsCompleted extends RawOp implements Operand< private Output unitsCompleted; - private ReaderNumWorkUnitsCompleted(Operation operation) { - super(operation); + public ReaderNumWorkUnitsCompleted(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; unitsCompleted = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return unitsCompleted; } + @OpInputsMetadata( + outputsClass = ReaderNumWorkUnitsCompleted.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a Reader. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRead.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRead.java index c905794a387..ed71559ff04 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRead.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRead.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * Reader needs to start reading from a new file since it has finished * with the previous file). */ +@OpMetadata( + opType = ReaderRead.OP_NAME, + inputsClass = ReaderRead.Inputs.class +) @Operator( group = "io" ) @@ -50,8 +56,8 @@ public final class ReaderRead extends RawOp { private Output value; - private ReaderRead(Operation operation) { - super(operation); + public ReaderRead(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; key = operation.output(outputIdx++); value = operation.output(outputIdx++); @@ -94,6 +100,9 @@ public Output value() { return value; } + @OpInputsMetadata( + outputsClass = ReaderRead.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a Reader. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReadUpTo.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReadUpTo.java index 0fa533b15fe..be4b123e682 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReadUpTo.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReadUpTo.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ * with the previous file). * It may return less than {@code num_records} even before the last batch. */ +@OpMetadata( + opType = ReaderReadUpTo.OP_NAME, + inputsClass = ReaderReadUpTo.Inputs.class +) @Operator( group = "io" ) @@ -52,8 +58,8 @@ public final class ReaderReadUpTo extends RawOp { private Output values; - private ReaderReadUpTo(Operation operation) { - super(operation); + public ReaderReadUpTo(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; keys = operation.output(outputIdx++); values = operation.output(outputIdx++); @@ -98,6 +104,9 @@ public Output values() { return values; } + @OpInputsMetadata( + outputsClass = ReaderReadUpTo.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a {@code Reader}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReset.java index 2523ed24148..29abce1e9bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderReset.java @@ -26,12 +26,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * Restore a Reader to its initial clean state. */ +@OpMetadata( + opType = ReaderReset.OP_NAME, + inputsClass = ReaderReset.Inputs.class +) @Operator( group = "io" ) @@ -41,8 +47,8 @@ public final class ReaderReset extends RawOp { */ public static final String OP_NAME = "ReaderResetV2"; - private ReaderReset(Operation operation) { - super(operation); + public ReaderReset(Operation operation) { + super(operation, OP_NAME); } /** @@ -61,6 +67,9 @@ public static ReaderReset create(Scope scope, Operand readerHan return new ReaderReset(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ReaderReset.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a Reader. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRestoreState.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRestoreState.java index 0d9a421b5f9..ec8cd6b1e3b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRestoreState.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderRestoreState.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Not all Readers support being restored, so this can produce an * Unimplemented error. */ +@OpMetadata( + opType = ReaderRestoreState.OP_NAME, + inputsClass = ReaderRestoreState.Inputs.class +) @Operator( group = "io" ) @@ -44,8 +50,8 @@ public final class ReaderRestoreState extends RawOp { */ public static final String OP_NAME = "ReaderRestoreStateV2"; - private ReaderRestoreState(Operation operation) { - super(operation); + public ReaderRestoreState(Operation operation) { + super(operation, OP_NAME); } /** @@ -68,6 +74,9 @@ public static ReaderRestoreState create(Scope scope, Operand re return new ReaderRestoreState(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ReaderRestoreState.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a Reader. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderSerializeState.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderSerializeState.java index e4718641bcd..279c9011b5f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderSerializeState.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ReaderSerializeState.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * Not all Readers support being serialized, so this can produce an * Unimplemented error. */ +@OpMetadata( + opType = ReaderSerializeState.OP_NAME, + inputsClass = ReaderSerializeState.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class ReaderSerializeState extends RawOp implements Operand state; - private ReaderSerializeState(Operation operation) { - super(operation); + public ReaderSerializeState(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; state = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return state; } + @OpInputsMetadata( + outputsClass = ReaderSerializeState.class + ) public static class Inputs extends RawOpInputs { /** * Handle to a Reader. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeManySparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeManySparse.java index 482eb5554e4..3931de53007 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeManySparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeManySparse.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -45,6 +47,10 @@ * * @param data type for {@code serialized_sparse} output */ +@OpMetadata( + opType = SerializeManySparse.OP_NAME, + inputsClass = SerializeManySparse.Inputs.class +) @Operator( group = "io" ) @@ -56,8 +62,8 @@ public final class SerializeManySparse extends RawOp implements private Output serializedSparse; - private SerializeManySparse(Operation operation) { - super(operation); + public SerializeManySparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; serializedSparse = operation.output(outputIdx++); } @@ -119,6 +125,9 @@ public Output asOutput() { return serializedSparse; } + @OpInputsMetadata( + outputsClass = SerializeManySparse.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. The {@code indices} of the minibatch {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeSparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeSparse.java index b2adf565e6e..30f48d91bce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeSparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeSparse.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ * * @param data type for {@code serialized_sparse} output */ +@OpMetadata( + opType = SerializeSparse.OP_NAME, + inputsClass = SerializeSparse.Inputs.class +) @Operator( group = "io" ) @@ -50,8 +56,8 @@ public final class SerializeSparse extends RawOp implements Ope private Output serializedSparse; - private SerializeSparse(Operation operation) { - super(operation); + public SerializeSparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; serializedSparse = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return serializedSparse; } + @OpInputsMetadata( + outputsClass = SerializeSparse.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. The {@code indices} of the {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeTensor.java index 3321749b0b1..cb077bc90f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/SerializeTensor.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ /** * Transforms a Tensor into a serialized TensorProto proto. */ +@OpMetadata( + opType = SerializeTensor.OP_NAME, + inputsClass = SerializeTensor.Inputs.class +) @Operator( group = "io" ) @@ -46,8 +52,8 @@ public final class SerializeTensor extends RawOp implements Operand { private Output serialized; - private SerializeTensor(Operation operation) { - super(operation); + public SerializeTensor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; serialized = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return serialized; } + @OpInputsMetadata( + outputsClass = SerializeTensor.class + ) public static class Inputs extends RawOpInputs { /** * A Tensor of type {@code T}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilename.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilename.java index 0693814d39c..48c5d5aa871 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilename.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilename.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ * Generate a sharded filename. The filename is printf formatted as * %s-%05d-of-%05d, basename, shard, num_shards. */ +@OpMetadata( + opType = ShardedFilename.OP_NAME, + inputsClass = ShardedFilename.Inputs.class +) @Operator( group = "io" ) @@ -46,8 +52,8 @@ public final class ShardedFilename extends RawOp implements Operand { private Output filename; - private ShardedFilename(Operation operation) { - super(operation); + public ShardedFilename(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; filename = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return filename; } + @OpInputsMetadata( + outputsClass = ShardedFilename.class + ) public static class Inputs extends RawOpInputs { /** * The basename input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilespec.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilespec.java index f7aec950873..da2d98807b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilespec.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/ShardedFilespec.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -34,6 +36,10 @@ /** * Generate a glob pattern matching all sharded file names. */ +@OpMetadata( + opType = ShardedFilespec.OP_NAME, + inputsClass = ShardedFilespec.Inputs.class +) @Operator( group = "io" ) @@ -45,8 +51,8 @@ public final class ShardedFilespec extends RawOp implements Operand { private Output filename; - private ShardedFilespec(Operation operation) { - super(operation); + public ShardedFilespec(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; filename = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return filename; } + @OpInputsMetadata( + outputsClass = ShardedFilespec.class + ) public static class Inputs extends RawOpInputs { /** * The basename input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TextLineReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TextLineReader.java index 417b34c310c..ba98d273703 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TextLineReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TextLineReader.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * A Reader that outputs the lines of a file delimited by '\n'. */ +@OpMetadata( + opType = TextLineReader.OP_NAME, + inputsClass = TextLineReader.Inputs.class +) @Operator( group = "io" ) @@ -45,8 +51,8 @@ public final class TextLineReader extends RawOp implements Operand { private Output readerHandle; @SuppressWarnings("unchecked") - private TextLineReader(Operation operation) { - super(operation); + public TextLineReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; readerHandle = operation.output(outputIdx++); } @@ -175,6 +181,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = TextLineReader.class + ) public static class Inputs extends RawOpInputs { /** * Number of lines to skip from the beginning of every file. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TfRecordReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TfRecordReader.java index 7eea770b3f7..3fa20389961 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TfRecordReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/TfRecordReader.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * A Reader that outputs the records from a TensorFlow Records file. */ +@OpMetadata( + opType = TfRecordReader.OP_NAME, + inputsClass = TfRecordReader.Inputs.class +) @Operator( group = "io" ) @@ -45,8 +51,8 @@ public final class TfRecordReader extends RawOp implements Operand { private Output readerHandle; @SuppressWarnings("unchecked") - private TfRecordReader(Operation operation) { - super(operation); + public TfRecordReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; readerHandle = operation.output(outputIdx++); } @@ -175,6 +181,9 @@ public Options compressionType(String compressionType) { } } + @OpInputsMetadata( + outputsClass = TfRecordReader.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this reader is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WholeFileReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WholeFileReader.java index 919b46c7902..b71d8283cc8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WholeFileReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WholeFileReader.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * To use, enqueue filenames in a Queue. The output of ReaderRead will * be a filename (key) and the contents of that file (value). */ +@OpMetadata( + opType = WholeFileReader.OP_NAME, + inputsClass = WholeFileReader.Inputs.class +) @Operator( group = "io" ) @@ -47,8 +53,8 @@ public final class WholeFileReader extends RawOp implements Operand { private Output readerHandle; @SuppressWarnings("unchecked") - private WholeFileReader(Operation operation) { - super(operation); + public WholeFileReader(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; readerHandle = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = WholeFileReader.class + ) public static class Inputs extends RawOpInputs { /** * If non-empty, this reader is placed in the given container. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WriteFile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WriteFile.java index 7143f27744a..a2d86a01565 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WriteFile.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/WriteFile.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -33,6 +35,10 @@ * Writes contents to the file at input filename. Creates file and recursively * creates directory if not existing. */ +@OpMetadata( + opType = WriteFile.OP_NAME, + inputsClass = WriteFile.Inputs.class +) @Operator( group = "io" ) @@ -42,8 +48,8 @@ public final class WriteFile extends RawOp { */ public static final String OP_NAME = "WriteFile"; - private WriteFile(Operation operation) { - super(operation); + public WriteFile(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static WriteFile create(Scope scope, Operand filename, return new WriteFile(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = WriteFile.class + ) public static class Inputs extends RawOpInputs { /** * scalar. The name of the file to which we write the contents. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandPart.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandPart.java index ac43eb9df3d..c8fa49a410b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandPart.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandPart.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -66,6 +68,10 @@ * * @param data type for {@code band} output */ +@OpMetadata( + opType = BandPart.OP_NAME, + inputsClass = BandPart.Inputs.class +) @Operator( group = "linalg" ) @@ -77,8 +83,8 @@ public final class BandPart extends RawOp implements Operand private Output band; - private BandPart(Operation operation) { - super(operation); + public BandPart(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; band = operation.output(outputIdx++); } @@ -122,6 +128,9 @@ public Output asOutput() { return band; } + @OpInputsMetadata( + outputsClass = BandPart.class + ) public static class Inputs extends RawOpInputs> { /** * Rank {@code k} tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java index 64f76d4ee42..17b00d7b7d8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BandedTriangularSolve.OP_NAME, + inputsClass = BandedTriangularSolve.Inputs.class +) public final class BandedTriangularSolve extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BandedTriangularSolve extends RawOp implemen private Output output; - private BandedTriangularSolve(Operation operation) { - super(operation); + public BandedTriangularSolve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = BandedTriangularSolve.class + ) public static class Inputs extends RawOpInputs> { /** * The matrix input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholesky.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholesky.java index d7751ef99fb..6fa5ec46905 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholesky.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholesky.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchCholesky.OP_NAME, + inputsClass = BatchCholesky.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchCholesky extends RawOp implements Ope private Output output; - private BatchCholesky(Operation operation) { - super(operation); + public BatchCholesky(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchCholesky.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholeskyGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholeskyGrad.java index a7671e74726..31d533e3165 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholeskyGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchCholeskyGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchCholeskyGrad.OP_NAME, + inputsClass = BatchCholeskyGrad.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchCholeskyGrad extends RawOp implements private Output output; - private BatchCholeskyGrad(Operation operation) { - super(operation); + public BatchCholeskyGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchCholeskyGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The l input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixBandPart.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixBandPart.java index f8a440b50b8..c8eee9e13c8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixBandPart.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixBandPart.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -37,6 +39,10 @@ * * @param data type for {@code band} output */ +@OpMetadata( + opType = BatchMatrixBandPart.OP_NAME, + inputsClass = BatchMatrixBandPart.Inputs.class +) @Operator( group = "linalg" ) @@ -48,8 +54,8 @@ public final class BatchMatrixBandPart extends RawOp implements private Output band; - private BatchMatrixBandPart(Operation operation) { - super(operation); + public BatchMatrixBandPart(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; band = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return band; } + @OpInputsMetadata( + outputsClass = BatchMatrixBandPart.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDeterminant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDeterminant.java index b6824b065de..db89c57ff0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDeterminant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDeterminant.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixDeterminant.OP_NAME, + inputsClass = BatchMatrixDeterminant.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixDeterminant extends RawOp impleme private Output output; - private BatchMatrixDeterminant(Operation operation) { - super(operation); + public BatchMatrixDeterminant(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchMatrixDeterminant.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiag.java index 62ebc20bf8e..94fcdef153d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiag.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixDiag.OP_NAME, + inputsClass = BatchMatrixDiag.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixDiag extends RawOp implements Ope private Output output; - private BatchMatrixDiag(Operation operation) { - super(operation); + public BatchMatrixDiag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchMatrixDiag.class + ) public static class Inputs extends RawOpInputs> { /** * The diagonal input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiagPart.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiagPart.java index c749fed819d..3bc5e4f137d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiagPart.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixDiagPart.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code diagonal} output */ +@OpMetadata( + opType = BatchMatrixDiagPart.OP_NAME, + inputsClass = BatchMatrixDiagPart.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixDiagPart extends RawOp implements private Output diagonal; - private BatchMatrixDiagPart(Operation operation) { - super(operation); + public BatchMatrixDiagPart(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; diagonal = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return diagonal; } + @OpInputsMetadata( + outputsClass = BatchMatrixDiagPart.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixInverse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixInverse.java index 5f9145ff005..0310cfa4d24 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixInverse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixInverse.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixInverse.OP_NAME, + inputsClass = BatchMatrixInverse.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixInverse extends RawOp implement private Output output; - private BatchMatrixInverse(Operation operation) { - super(operation); + public BatchMatrixInverse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -124,6 +130,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = BatchMatrixInverse.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSetDiag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSetDiag.java index 30711550209..98999b3e89c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSetDiag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSetDiag.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixSetDiag.OP_NAME, + inputsClass = BatchMatrixSetDiag.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixSetDiag extends RawOp implements private Output output; - private BatchMatrixSetDiag(Operation operation) { - super(operation); + public BatchMatrixSetDiag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = BatchMatrixSetDiag.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolve.java index 47af37fb73b..034497a1b7c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixSolve.OP_NAME, + inputsClass = BatchMatrixSolve.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixSolve extends RawOp implements private Output output; - private BatchMatrixSolve(Operation operation) { - super(operation); + public BatchMatrixSolve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -126,6 +132,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = BatchMatrixSolve.class + ) public static class Inputs extends RawOpInputs> { /** * The matrix input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolveLs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolveLs.java index 537dbc51cda..e27017cdfd8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolveLs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixSolveLs.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat64; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixSolveLs.OP_NAME, + inputsClass = BatchMatrixSolveLs.Inputs.class +) @Operator( group = "linalg" ) @@ -48,8 +54,8 @@ public final class BatchMatrixSolveLs extends RawOp implement private Output output; - private BatchMatrixSolveLs(Operation operation) { - super(operation); + public BatchMatrixSolveLs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options fast(Boolean fast) { } } + @OpInputsMetadata( + outputsClass = BatchMatrixSolveLs.class + ) public static class Inputs extends RawOpInputs> { /** * The matrix input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixTriangularSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixTriangularSolve.java index b1a0fad24a9..c67b572576c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixTriangularSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchMatrixTriangularSolve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatrixTriangularSolve.OP_NAME, + inputsClass = BatchMatrixTriangularSolve.Inputs.class +) @Operator( group = "linalg" ) @@ -47,8 +53,8 @@ public final class BatchMatrixTriangularSolve extends RawOp i private Output output; - private BatchMatrixTriangularSolve(Operation operation) { - super(operation); + public BatchMatrixTriangularSolve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -152,6 +158,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = BatchMatrixTriangularSolve.class + ) public static class Inputs extends RawOpInputs> { /** * The matrix input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSelfAdjointEig.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSelfAdjointEig.java index 77cc608d34c..6819006c568 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSelfAdjointEig.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSelfAdjointEig.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code e} output */ +@OpMetadata( + opType = BatchSelfAdjointEig.OP_NAME, + inputsClass = BatchSelfAdjointEig.Inputs.class +) @Operator( group = "linalg" ) @@ -49,8 +55,8 @@ public final class BatchSelfAdjointEig extends RawOp { private Output v; - private BatchSelfAdjointEig(Operation operation) { - super(operation); + public BatchSelfAdjointEig(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; e = operation.output(outputIdx++); v = operation.output(outputIdx++); @@ -131,6 +137,9 @@ public Options computeV(Boolean computeV) { } } + @OpInputsMetadata( + outputsClass = BatchSelfAdjointEig.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSvd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSvd.java index 0eed0e8668a..86d30c34858 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSvd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BatchSvd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code s} output */ +@OpMetadata( + opType = BatchSvd.OP_NAME, + inputsClass = BatchSvd.Inputs.class +) @Operator( group = "linalg" ) @@ -51,8 +57,8 @@ public final class BatchSvd extends RawOp { private Output v; - private BatchSvd(Operation operation) { - super(operation); + public BatchSvd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; s = operation.output(outputIdx++); u = operation.output(outputIdx++); @@ -169,6 +175,9 @@ public Options fullMatrices(Boolean fullMatrices) { } } + @OpInputsMetadata( + outputsClass = BatchSvd.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cholesky.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cholesky.java index 994fe407425..b8859edf2c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cholesky.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cholesky.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Cholesky.OP_NAME, + inputsClass = Cholesky.Inputs.class +) @Operator( group = "linalg" ) @@ -57,8 +63,8 @@ public final class Cholesky extends RawOp implements Operand private Output output; - private Cholesky(Operation operation) { - super(operation); + public Cholesky(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Cholesky.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/CholeskyGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/CholeskyGrad.java index b52d99e7c91..ba1d4870f95 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/CholeskyGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/CholeskyGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CholeskyGrad.OP_NAME, + inputsClass = CholeskyGrad.Inputs.class +) @Operator( group = "linalg" ) @@ -49,8 +55,8 @@ public final class CholeskyGrad extends RawOp implements Oper private Output output; - private CholeskyGrad(Operation operation) { - super(operation); + public CholeskyGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = CholeskyGrad.class + ) public static class Inputs extends RawOpInputs> { /** * Output of batch Cholesky algorithm l = cholesky(A). Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/ConjugateTranspose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/ConjugateTranspose.java index 087147d3149..d65f3aa102c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/ConjugateTranspose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/ConjugateTranspose.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = ConjugateTranspose.OP_NAME, + inputsClass = ConjugateTranspose.Inputs.class +) @Operator( group = "linalg" ) @@ -51,8 +57,8 @@ public final class ConjugateTranspose extends RawOp implements private Output y; - private ConjugateTranspose(Operation operation) { - super(operation); + public ConjugateTranspose(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = ConjugateTranspose.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cross.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cross.java index 719266eeede..8c3a50c3080 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cross.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Cross.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code product} output */ +@OpMetadata( + opType = Cross.OP_NAME, + inputsClass = Cross.Inputs.class +) @Operator( group = "linalg" ) @@ -50,8 +56,8 @@ public final class Cross extends RawOp implements Operand private Output product; - private Cross(Operation operation) { - super(operation); + public Cross(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; product = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return product; } + @OpInputsMetadata( + outputsClass = Cross.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor containing 3-element vectors. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Det.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Det.java index 32aabbb3182..f92c8910ed5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Det.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Det.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Det.OP_NAME, + inputsClass = Det.Inputs.class +) @Operator( group = "linalg" ) @@ -50,8 +56,8 @@ public final class Det extends RawOp implements Operand { private Output output; - private Det(Operation operation) { - super(operation); + public Det(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Det.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Eig.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Eig.java index ea31072f390..0c9a90d97d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Eig.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Eig.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -47,6 +49,10 @@ * * @param data type for {@code e} output */ +@OpMetadata( + opType = Eig.OP_NAME, + inputsClass = Eig.Inputs.class +) @Operator( group = "linalg" ) @@ -60,8 +66,8 @@ public final class Eig extends RawOp { private Output v; - private Eig(Operation operation) { - super(operation); + public Eig(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; e = operation.output(outputIdx++); v = operation.output(outputIdx++); @@ -146,6 +152,9 @@ public Options computeV(Boolean computeV) { } } + @OpInputsMetadata( + outputsClass = Eig.class + ) public static class Inputs extends RawOpInputs> { /** * {@code Tensor} input of shape {@code [N, N]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Einsum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Einsum.java index c273cf2f9f0..c628f4d6205 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Einsum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Einsum.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -100,6 +102,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Einsum.OP_NAME, + inputsClass = Einsum.Inputs.class +) @Operator( group = "linalg" ) @@ -111,8 +117,8 @@ public final class Einsum extends RawOp implements Operand { private Output output; - private Einsum(Operation operation) { - super(operation); + public Einsum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Einsum.class + ) public static class Inputs extends RawOpInputs> { /** * List of 1 or 2 Tensors. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/EuclideanNorm.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/EuclideanNorm.java index a1fdf41273e..ce6dbf71bd9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/EuclideanNorm.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/EuclideanNorm.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = EuclideanNorm.OP_NAME, + inputsClass = EuclideanNorm.Inputs.class +) @Operator( group = "linalg" ) @@ -52,8 +58,8 @@ public final class EuclideanNorm extends RawOp implements Opera private Output output; - private EuclideanNorm(Operation operation) { - super(operation); + public EuclideanNorm(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -132,6 +138,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = EuclideanNorm.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Inv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Inv.java index 708bf232915..84a267bc666 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Inv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Inv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Inv.OP_NAME, + inputsClass = Inv.Inputs.class +) @Operator( group = "linalg" ) @@ -54,8 +60,8 @@ public final class Inv extends RawOp implements Operand { private Output output; - private Inv(Operation operation) { - super(operation); + public Inv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -133,6 +139,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = Inv.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LoadAndRemapMatrix.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LoadAndRemapMatrix.java index 31232e9c92b..6ad04c77b1a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LoadAndRemapMatrix.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LoadAndRemapMatrix.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -64,6 +66,10 @@ * [w(0, 0), w(0, 2), -0.5], * [0.25, -0.25, 42]] */ +@OpMetadata( + opType = LoadAndRemapMatrix.OP_NAME, + inputsClass = LoadAndRemapMatrix.Inputs.class +) @Operator( group = "linalg" ) @@ -75,8 +81,8 @@ public final class LoadAndRemapMatrix extends RawOp implements Operand private Output outputMatrix; - private LoadAndRemapMatrix(Operation operation) { - super(operation); + public LoadAndRemapMatrix(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputMatrix = operation.output(outputIdx++); } @@ -177,6 +183,9 @@ public Options maxRowsInMemory(Long maxRowsInMemory) { } } + @OpInputsMetadata( + outputsClass = LoadAndRemapMatrix.class + ) public static class Inputs extends RawOpInputs { /** * Path to the TensorFlow checkpoint (version 2, {@code TensorBundle}) from diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LogMatrixDeterminant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LogMatrixDeterminant.java index 05456f98e52..c9aafe2e9f8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LogMatrixDeterminant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/LogMatrixDeterminant.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code sign} output */ +@OpMetadata( + opType = LogMatrixDeterminant.OP_NAME, + inputsClass = LogMatrixDeterminant.Inputs.class +) @Operator( group = "linalg" ) @@ -57,8 +63,8 @@ public final class LogMatrixDeterminant extends RawOp { private Output logAbsDeterminant; - private LogMatrixDeterminant(Operation operation) { - super(operation); + public LogMatrixDeterminant(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sign = operation.output(outputIdx++); logAbsDeterminant = operation.output(outputIdx++); @@ -100,6 +106,9 @@ public Output logAbsDeterminant() { return logAbsDeterminant; } + @OpInputsMetadata( + outputsClass = LogMatrixDeterminant.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [N, M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Lu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Lu.java index e61e8627340..392b184c72d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Lu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Lu.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -54,6 +56,10 @@ * * @param data type for {@code p} output */ +@OpMetadata( + opType = Lu.OP_NAME, + inputsClass = Lu.Inputs.class +) @Operator( group = "linalg" ) @@ -67,8 +73,8 @@ public final class Lu extends RawOp { private Output p; - private Lu(Operation operation) { - super(operation); + public Lu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; lu = operation.output(outputIdx++); p = operation.output(outputIdx++); @@ -139,6 +145,9 @@ public Output p() { return p; } + @OpInputsMetadata( + outputsClass = Lu.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor of shape {@code [..., M, M]} whose inner-most 2 dimensions form matrices of diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatMul.java index 139cbbb831e..e538d714e41 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code product} output */ +@OpMetadata( + opType = MatMul.OP_NAME, + inputsClass = MatMul.Inputs.class +) @Operator( group = "linalg" ) @@ -53,8 +59,8 @@ public final class MatMul extends RawOp implements Operand { private Output product; - private MatMul(Operation operation) { - super(operation); + public MatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; product = operation.output(outputIdx++); } @@ -158,6 +164,9 @@ public Options transposeB(Boolean transposeB) { } } + @OpInputsMetadata( + outputsClass = MatMul.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiag.java index 9c6f94274bd..056b3ec173c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiag.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -117,6 +119,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MatrixDiag.OP_NAME, + inputsClass = MatrixDiag.Inputs.class +) @Operator( group = "linalg" ) @@ -128,8 +134,8 @@ public final class MatrixDiag extends RawOp implements Operand< private Output output; - private MatrixDiag(Operation operation) { - super(operation); + public MatrixDiag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -183,6 +189,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = MatrixDiag.class + ) public static class Inputs extends RawOpInputs> { /** * Rank {@code r}, where {@code r >= 1} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPart.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPart.java index d98d54862e4..8f82271074a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPart.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPart.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -97,6 +99,10 @@ * * @param data type for {@code diagonal} output */ +@OpMetadata( + opType = MatrixDiagPart.OP_NAME, + inputsClass = MatrixDiagPart.Inputs.class +) @Operator( group = "linalg" ) @@ -108,8 +114,8 @@ public final class MatrixDiagPart extends RawOp implements Oper private Output diagonal; - private MatrixDiagPart(Operation operation) { - super(operation); + public MatrixDiagPart(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; diagonal = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Output asOutput() { return diagonal; } + @OpInputsMetadata( + outputsClass = MatrixDiagPart.class + ) public static class Inputs extends RawOpInputs> { /** * Rank {@code r} tensor where {@code r >= 2}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPartV3.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPartV3.java index 038c744a651..62cc8403b8e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPartV3.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagPartV3.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -127,6 +129,10 @@ * * @param data type for {@code diagonal} output */ +@OpMetadata( + opType = MatrixDiagPartV3.OP_NAME, + inputsClass = MatrixDiagPartV3.Inputs.class +) @Operator( group = "linalg" ) @@ -138,8 +144,8 @@ public final class MatrixDiagPartV3 extends RawOp implements Op private Output diagonal; - private MatrixDiagPartV3(Operation operation) { - super(operation); + public MatrixDiagPartV3(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; diagonal = operation.output(outputIdx++); } @@ -235,6 +241,9 @@ public Options align(String align) { } } + @OpInputsMetadata( + outputsClass = MatrixDiagPartV3.class + ) public static class Inputs extends RawOpInputs> { /** * Rank {@code r} tensor where {@code r >= 2}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagV3.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagV3.java index 814a25b4058..4c1f9c5649a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagV3.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixDiagV3.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -145,6 +147,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MatrixDiagV3.OP_NAME, + inputsClass = MatrixDiagV3.Inputs.class +) @Operator( group = "linalg" ) @@ -156,8 +162,8 @@ public final class MatrixDiagV3 extends RawOp implements Operan private Output output; - private MatrixDiagV3(Operation operation) { - super(operation); + public MatrixDiagV3(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -262,6 +268,9 @@ public Options align(String align) { } } + @OpInputsMetadata( + outputsClass = MatrixDiagV3.class + ) public static class Inputs extends RawOpInputs> { /** * Rank {@code r}, where {@code r >= 1} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java index d0c74cd771b..19d87ac7b57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MatrixLogarithm.OP_NAME, + inputsClass = MatrixLogarithm.Inputs.class +) public final class MatrixLogarithm extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -54,8 +60,8 @@ public final class MatrixLogarithm extends RawOp implements Ope private Output output; - private MatrixLogarithm(Operation operation) { - super(operation); + public MatrixLogarithm(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = MatrixLogarithm.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSetDiag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSetDiag.java index 89ebedfe50f..726a383702a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSetDiag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSetDiag.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -133,6 +135,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MatrixSetDiag.OP_NAME, + inputsClass = MatrixSetDiag.Inputs.class +) @Operator( group = "linalg" ) @@ -144,8 +150,8 @@ public final class MatrixSetDiag extends RawOp implements Opera private Output output; - private MatrixSetDiag(Operation operation) { - super(operation); + public MatrixSetDiag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -241,6 +247,9 @@ public Options align(String align) { } } + @OpInputsMetadata( + outputsClass = MatrixSetDiag.class + ) public static class Inputs extends RawOpInputs> { /** * Rank {@code r+1}, where {@code r >= 1}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSolveLs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSolveLs.java index a1a0e24aea4..c3e0a51ac2a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSolveLs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixSolveLs.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat64; @@ -67,6 +69,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MatrixSolveLs.OP_NAME, + inputsClass = MatrixSolveLs.Inputs.class +) @Operator( group = "linalg" ) @@ -78,8 +84,8 @@ public final class MatrixSolveLs extends RawOp implements Opera private Output output; - private MatrixSolveLs(Operation operation) { - super(operation); + public MatrixSolveLs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -162,6 +168,9 @@ public Options fast(Boolean fast) { } } + @OpInputsMetadata( + outputsClass = MatrixSolveLs.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, N]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Qr.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Qr.java index 618f33cc405..2c86bec5b9f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Qr.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Qr.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +50,10 @@ * * @param data type for {@code q} output */ +@OpMetadata( + opType = Qr.OP_NAME, + inputsClass = Qr.Inputs.class +) @Operator( group = "linalg" ) @@ -61,8 +67,8 @@ public final class Qr extends RawOp { private Output r; - private Qr(Operation operation) { - super(operation); + public Qr(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; q = operation.output(outputIdx++); r = operation.output(outputIdx++); @@ -148,6 +154,9 @@ public Options fullMatrices(Boolean fullMatrices) { } } + @OpInputsMetadata( + outputsClass = Qr.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor of shape {@code [..., M, N]} whose inner-most 2 dimensions diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMul.java index 151f366432c..4eb611641db 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMul.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -42,6 +44,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = QuantizedMatMul.OP_NAME, + inputsClass = QuantizedMatMul.Inputs.class +) @Operator( group = "linalg" ) @@ -57,8 +63,8 @@ public final class QuantizedMatMul extends RawOp { private Output maxOut; - private QuantizedMatMul(Operation operation) { - super(operation); + public QuantizedMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); minOut = operation.output(outputIdx++); @@ -193,6 +199,9 @@ public Options transposeB(Boolean transposeB) { } } + @OpInputsMetadata( + outputsClass = QuantizedMatMul.class + ) public static class Inputs extends RawOpInputs> { /** * Must be a two-dimensional tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java index e55131173ff..ac098c6bbc0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -43,6 +45,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = QuantizedMatMulWithBias.OP_NAME, + inputsClass = QuantizedMatMulWithBias.Inputs.class +) public final class QuantizedMatMulWithBias extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -55,8 +61,8 @@ public final class QuantizedMatMulWithBias extends RawOp { private Output maxOut; - private QuantizedMatMulWithBias(Operation operation) { - super(operation); + public QuantizedMatMulWithBias(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); minOut = operation.output(outputIdx++); @@ -216,6 +222,9 @@ public Options inputQuantMode(String inputQuantMode) { } } + @OpInputsMetadata( + outputsClass = QuantizedMatMulWithBias.class + ) public static class Inputs extends RawOpInputs> { /** * A matrix to be multiplied. Must be a two-dimensional tensor of type {@code quint8}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java index 89adf2b6063..4643805d6d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +46,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = QuantizedMatMulWithBiasAndRelu.OP_NAME, + inputsClass = QuantizedMatMulWithBiasAndRelu.Inputs.class +) public final class QuantizedMatMulWithBiasAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -56,8 +62,8 @@ public final class QuantizedMatMulWithBiasAndRelu extends Raw private Output maxOut; - private QuantizedMatMulWithBiasAndRelu(Operation operation) { - super(operation); + public QuantizedMatMulWithBiasAndRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); minOut = operation.output(outputIdx++); @@ -217,6 +223,9 @@ public Options inputQuantMode(String inputQuantMode) { } } + @OpInputsMetadata( + outputsClass = QuantizedMatMulWithBiasAndRelu.class + ) public static class Inputs extends RawOpInputs> { /** * A matrix to be multiplied. Must be a two-dimensional tensor of type {@code quint8}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java index adf556ae8d3..deb84874e4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -45,6 +47,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = QuantizedMatMulWithBiasAndReluAndRequantize.OP_NAME, + inputsClass = QuantizedMatMulWithBiasAndReluAndRequantize.Inputs.class +) public final class QuantizedMatMulWithBiasAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -57,8 +63,8 @@ public final class QuantizedMatMulWithBiasAndReluAndRequantize maxOut; - private QuantizedMatMulWithBiasAndReluAndRequantize(Operation operation) { - super(operation); + public QuantizedMatMulWithBiasAndReluAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); minOut = operation.output(outputIdx++); @@ -223,6 +229,9 @@ public Options inputQuantMode(String inputQuantMode) { } } + @OpInputsMetadata( + outputsClass = QuantizedMatMulWithBiasAndReluAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * A matrix to be multiplied. Must be a two-dimensional tensor of type {@code quint8}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/SelfAdjointEig.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/SelfAdjointEig.java index adde5950814..2f7bba28e22 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/SelfAdjointEig.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/SelfAdjointEig.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +48,10 @@ * * @param data type for {@code e} output */ +@OpMetadata( + opType = SelfAdjointEig.OP_NAME, + inputsClass = SelfAdjointEig.Inputs.class +) @Operator( group = "linalg" ) @@ -59,8 +65,8 @@ public final class SelfAdjointEig extends RawOp { private Output v; - private SelfAdjointEig(Operation operation) { - super(operation); + public SelfAdjointEig(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; e = operation.output(outputIdx++); v = operation.output(outputIdx++); @@ -143,6 +149,9 @@ public Options computeV(Boolean computeV) { } } + @OpInputsMetadata( + outputsClass = SelfAdjointEig.class + ) public static class Inputs extends RawOpInputs> { /** * {@code Tensor} input of shape {@code [N, N]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Solve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Solve.java index f07de1dcd02..7da5179d0f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Solve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Solve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Solve.OP_NAME, + inputsClass = Solve.Inputs.class +) @Operator( group = "linalg" ) @@ -53,8 +59,8 @@ public final class Solve extends RawOp implements Operand { private Output output; - private Solve(Operation operation) { - super(operation); + public Solve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -134,6 +140,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = Solve.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Sqrtm.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Sqrtm.java index cb633920385..c426849f57f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Sqrtm.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Sqrtm.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Sqrtm.OP_NAME, + inputsClass = Sqrtm.Inputs.class +) @Operator( group = "linalg" ) @@ -59,8 +65,8 @@ public final class Sqrtm extends RawOp implements Operand { private Output output; - private Sqrtm(Operation operation) { - super(operation); + public Sqrtm(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Sqrtm.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Svd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Svd.java index a3b388794cc..5d515f65548 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Svd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Svd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +48,10 @@ * * @param data type for {@code s} output */ +@OpMetadata( + opType = Svd.OP_NAME, + inputsClass = Svd.Inputs.class +) @Operator( group = "linalg" ) @@ -61,8 +67,8 @@ public final class Svd extends RawOp { private Output v; - private Svd(Operation operation) { - super(operation); + public Svd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; s = operation.output(outputIdx++); u = operation.output(outputIdx++); @@ -191,6 +197,9 @@ public Options fullMatrices(Boolean fullMatrices) { } } + @OpInputsMetadata( + outputsClass = Svd.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor of shape {@code [..., M, N]} whose inner-most 2 dimensions diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiag.java index 4dac3f09451..2e1b99481cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiag.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -49,6 +51,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TensorDiag.OP_NAME, + inputsClass = TensorDiag.Inputs.class +) @Operator( group = "linalg" ) @@ -60,8 +66,8 @@ public final class TensorDiag extends RawOp implements Operand< private Output output; - private TensorDiag(Operation operation) { - super(operation); + public TensorDiag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TensorDiag.class + ) public static class Inputs extends RawOpInputs> { /** * Rank k tensor where k is at most 1. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiagPart.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiagPart.java index 6cc286ea41b..e7f24e7e5c7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiagPart.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TensorDiagPart.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -50,6 +52,10 @@ * * @param data type for {@code diagonal} output */ +@OpMetadata( + opType = TensorDiagPart.OP_NAME, + inputsClass = TensorDiagPart.Inputs.class +) @Operator( group = "linalg" ) @@ -61,8 +67,8 @@ public final class TensorDiagPart extends RawOp implements Oper private Output diagonal; - private TensorDiagPart(Operation operation) { - super(operation); + public TensorDiagPart(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; diagonal = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return diagonal; } + @OpInputsMetadata( + outputsClass = TensorDiagPart.class + ) public static class Inputs extends RawOpInputs> { /** * Rank k tensor where k is even and not zero. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Transpose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Transpose.java index cb3430f85d4..504f4ba3d81 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Transpose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/Transpose.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Transpose.OP_NAME, + inputsClass = Transpose.Inputs.class +) @Operator( group = "linalg" ) @@ -50,8 +56,8 @@ public final class Transpose extends RawOp implements Operand y; - private Transpose(Operation operation) { - super(operation); + public Transpose(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Transpose.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TriangularSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TriangularSolve.java index 65f0269ce1d..9fbc4979644 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TriangularSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TriangularSolve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -78,6 +80,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TriangularSolve.OP_NAME, + inputsClass = TriangularSolve.Inputs.class +) @Operator( group = "linalg" ) @@ -89,8 +95,8 @@ public final class TriangularSolve extends RawOp implements Ope private Output output; - private TriangularSolve(Operation operation) { - super(operation); + public TriangularSolve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -204,6 +210,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = TriangularSolve.class + ) public static class Inputs extends RawOpInputs> { /** * Shape is {@code [..., M, M]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java index 544a71da27d..e27f5e67b08 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TridiagonalMatMul.OP_NAME, + inputsClass = TridiagonalMatMul.Inputs.class +) public final class TridiagonalMatMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class TridiagonalMatMul extends RawOp implements O private Output output; - private TridiagonalMatMul(Operation operation) { - super(operation); + public TridiagonalMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TridiagonalMatMul.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor of shape {@code [..., 1, M]}, representing superdiagonals of diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java index a62030705f6..d9fccfaebe6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TridiagonalSolve.OP_NAME, + inputsClass = TridiagonalSolve.Inputs.class +) public final class TridiagonalSolve extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class TridiagonalSolve extends RawOp implements Op private Output output; - private TridiagonalSolve(Operation operation) { - super(operation); + public TridiagonalSolve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -161,6 +167,9 @@ public Options perturbSingular(Boolean perturbSingular) { } } + @OpInputsMetadata( + outputsClass = TridiagonalSolve.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor of shape {@code [..., 3, M]} whose innermost 2 dimensions represent the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java index 441df3a3e05..2251ead4b68 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = CSRSparseMatrixComponents.OP_NAME, + inputsClass = CSRSparseMatrixComponents.Inputs.class +) public final class CSRSparseMatrixComponents extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -51,8 +57,8 @@ public final class CSRSparseMatrixComponents extends RawOp { private Output values; - private CSRSparseMatrixComponents(Operation operation) { - super(operation); + public CSRSparseMatrixComponents(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; rowPtrs = operation.output(outputIdx++); colInds = operation.output(outputIdx++); @@ -108,6 +114,9 @@ public Output values() { return values; } + @OpInputsMetadata( + outputsClass = CSRSparseMatrixComponents.class + ) public static class Inputs extends RawOpInputs> { /** * A batched CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java index 032c356bb97..3d0a5b37604 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code dense_output} output */ +@OpMetadata( + opType = CSRSparseMatrixToDense.OP_NAME, + inputsClass = CSRSparseMatrixToDense.Inputs.class +) public final class CSRSparseMatrixToDense extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class CSRSparseMatrixToDense extends RawOp impleme private Output denseOutput; - private CSRSparseMatrixToDense(Operation operation) { - super(operation); + public CSRSparseMatrixToDense(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; denseOutput = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return denseOutput; } + @OpInputsMetadata( + outputsClass = CSRSparseMatrixToDense.class + ) public static class Inputs extends RawOpInputs> { /** * A batched CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java index e34a47fad2c..4cb3929785f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = CSRSparseMatrixToSparseTensor.OP_NAME, + inputsClass = CSRSparseMatrixToSparseTensor.Inputs.class +) public final class CSRSparseMatrixToSparseTensor extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class CSRSparseMatrixToSparseTensor extends RawOp private Output denseShape; - private CSRSparseMatrixToSparseTensor(Operation operation) { - super(operation); + public CSRSparseMatrixToSparseTensor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; indices = operation.output(outputIdx++); values = operation.output(outputIdx++); @@ -104,6 +110,9 @@ public Output denseShape() { return denseShape; } + @OpInputsMetadata( + outputsClass = CSRSparseMatrixToSparseTensor.class + ) public static class Inputs extends RawOpInputs> { /** * A (possibly batched) CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java index b1edd569963..403903fa74a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Converts a dense tensor to a (possibly batched) CSRSparseMatrix. */ +@OpMetadata( + opType = DenseToCSRSparseMatrix.OP_NAME, + inputsClass = DenseToCSRSparseMatrix.Inputs.class +) public final class DenseToCSRSparseMatrix extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class DenseToCSRSparseMatrix extends RawOp implements Operand sparseOutput; @SuppressWarnings("unchecked") - private DenseToCSRSparseMatrix(Operation operation) { - super(operation); + public DenseToCSRSparseMatrix(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseOutput = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return (Output) sparseOutput; } + @OpInputsMetadata( + outputsClass = DenseToCSRSparseMatrix.class + ) public static class Inputs extends RawOpInputs { /** * A Dense tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java index 36a84e7f892..13b704ce778 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * The gradients of SparseMatrixAdd outputs with respect to alpha and beta are not * currently defined (TensorFlow will return zeros for these entries). */ +@OpMetadata( + opType = SparseMatrixAdd.OP_NAME, + inputsClass = SparseMatrixAdd.Inputs.class +) public final class SparseMatrixAdd extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class SparseMatrixAdd extends RawOp implements Operand { private Output c; @SuppressWarnings("unchecked") - private SparseMatrixAdd(Operation operation) { - super(operation); + public SparseMatrixAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; c = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return (Output) c; } + @OpInputsMetadata( + outputsClass = SparseMatrixAdd.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java index 11b776289df..5ad55a2bd3d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -56,6 +58,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseMatrixMatMul.OP_NAME, + inputsClass = SparseMatrixMatMul.Inputs.class +) public final class SparseMatrixMatMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -64,8 +70,8 @@ public final class SparseMatrixMatMul extends RawOp implements private Output output; - private SparseMatrixMatMul(Operation operation) { - super(operation); + public SparseMatrixMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -273,6 +279,9 @@ public Options conjugateOutput(Boolean conjugateOutput) { } } + @OpInputsMetadata( + outputsClass = SparseMatrixMatMul.class + ) public static class Inputs extends RawOpInputs> { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java index ce8d51f00a6..cb7b25cea40 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ *

    NOTE even if {@code b} is zero, the sparsity structure of the output does not * change. */ +@OpMetadata( + opType = SparseMatrixMul.OP_NAME, + inputsClass = SparseMatrixMul.Inputs.class +) public final class SparseMatrixMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class SparseMatrixMul extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private SparseMatrixMul(Operation operation) { - super(operation); + public SparseMatrixMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = SparseMatrixMul.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java index a6c19832dd2..1d92de3aee1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; /** * Returns the number of nonzeroes of {@code sparse_matrix}. */ +@OpMetadata( + opType = SparseMatrixNNZ.OP_NAME, + inputsClass = SparseMatrixNNZ.Inputs.class +) public final class SparseMatrixNNZ extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class SparseMatrixNNZ extends RawOp implements Operand { private Output nnz; - private SparseMatrixNNZ(Operation operation) { - super(operation); + public SparseMatrixNNZ(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; nnz = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return nnz; } + @OpInputsMetadata( + outputsClass = SparseMatrixNNZ.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java index 9b35eac22c9..7a766931103 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -74,6 +76,10 @@ *

    {@code ordering_amd_value} stores the AMD ordering: {@code [1 2 3 0]}. *

    input: A {@code CSRSparseMatrix}. */ +@OpMetadata( + opType = SparseMatrixOrderingAMD.OP_NAME, + inputsClass = SparseMatrixOrderingAMD.Inputs.class +) public final class SparseMatrixOrderingAMD extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -82,8 +88,8 @@ public final class SparseMatrixOrderingAMD extends RawOp implements Operand output; - private SparseMatrixOrderingAMD(Operation operation) { - super(operation); + public SparseMatrixOrderingAMD(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -118,6 +124,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseMatrixOrderingAMD.class + ) public static class Inputs extends RawOpInputs { /** * A {@code CSRSparseMatrix}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java index 5dd3f79012c..a11eaaf2e98 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * the output has the same sparsity structure as the input (though missing values * in the output may now be treated as having probability zero). */ +@OpMetadata( + opType = SparseMatrixSoftmax.OP_NAME, + inputsClass = SparseMatrixSoftmax.Inputs.class +) public final class SparseMatrixSoftmax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class SparseMatrixSoftmax extends RawOp implements Operand { private Output softmax; @SuppressWarnings("unchecked") - private SparseMatrixSoftmax(Operation operation) { - super(operation); + public SparseMatrixSoftmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; softmax = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return (Output) softmax; } + @OpInputsMetadata( + outputsClass = SparseMatrixSoftmax.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java index 05ae75273bc..3a2d3f1fffa 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ /** * Calculates the gradient of the SparseMatrixSoftmax op. */ +@OpMetadata( + opType = SparseMatrixSoftmaxGrad.OP_NAME, + inputsClass = SparseMatrixSoftmaxGrad.Inputs.class +) public final class SparseMatrixSoftmaxGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class SparseMatrixSoftmaxGrad extends RawOp implements Operand gradient; @SuppressWarnings("unchecked") - private SparseMatrixSoftmaxGrad(Operation operation) { - super(operation); + public SparseMatrixSoftmaxGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; gradient = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return (Output) gradient; } + @OpInputsMetadata( + outputsClass = SparseMatrixSoftmaxGrad.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java index bd8d2d0fcc4..c5b31229d45 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -99,6 +101,10 @@ * permutation: A {@code Tensor}. * type: The type of {@code input}. */ +@OpMetadata( + opType = SparseMatrixSparseCholesky.OP_NAME, + inputsClass = SparseMatrixSparseCholesky.Inputs.class +) public final class SparseMatrixSparseCholesky extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -108,8 +114,8 @@ public final class SparseMatrixSparseCholesky extends RawOp implements Operand output; @SuppressWarnings("unchecked") - private SparseMatrixSparseCholesky(Operation operation) { - super(operation); + public SparseMatrixSparseCholesky(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = SparseMatrixSparseCholesky.class + ) public static class Inputs extends RawOpInputs { /** * A {@code CSRSparseMatrix}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java index 5c4564e1067..4f3e5d718ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -100,6 +102,10 @@ * adjoint_a: If True, {@code a} adjointed before multiplication. * adjoint_b: If True, {@code b} adjointed before multiplication. */ +@OpMetadata( + opType = SparseMatrixSparseMatMul.OP_NAME, + inputsClass = SparseMatrixSparseMatMul.Inputs.class +) public final class SparseMatrixSparseMatMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -109,8 +115,8 @@ public final class SparseMatrixSparseMatMul extends RawOp implements Operand c; @SuppressWarnings("unchecked") - private SparseMatrixSparseMatMul(Operation operation) { - super(operation); + public SparseMatrixSparseMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; c = operation.output(outputIdx++); } @@ -269,6 +275,9 @@ public Options adjointB(Boolean adjointB) { } } + @OpInputsMetadata( + outputsClass = SparseMatrixSparseMatMul.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java index e8a0beceddb..2219a63bdfc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * Transposes the inner (matrix) dimensions of a SparseMatrix and optionally * conjugates its values. */ +@OpMetadata( + opType = SparseMatrixTranspose.OP_NAME, + inputsClass = SparseMatrixTranspose.Inputs.class +) public final class SparseMatrixTranspose extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class SparseMatrixTranspose extends RawOp implements Operand private Output output; @SuppressWarnings("unchecked") - private SparseMatrixTranspose(Operation operation) { - super(operation); + public SparseMatrixTranspose(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -125,6 +131,9 @@ public Options conjugate(Boolean conjugate) { } } + @OpInputsMetadata( + outputsClass = SparseMatrixTranspose.class + ) public static class Inputs extends RawOpInputs { /** * A CSRSparseMatrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java index 6502ad7d100..91080f1f093 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ /** * Creates an all-zeros CSRSparseMatrix with shape {@code dense_shape}. */ +@OpMetadata( + opType = SparseMatrixZeros.OP_NAME, + inputsClass = SparseMatrixZeros.Inputs.class +) public final class SparseMatrixZeros extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class SparseMatrixZeros extends RawOp implements Operand { private Output sparseMatrix; @SuppressWarnings("unchecked") - private SparseMatrixZeros(Operation operation) { - super(operation); + public SparseMatrixZeros(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseMatrix = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return (Output) sparseMatrix; } + @OpInputsMetadata( + outputsClass = SparseMatrixZeros.class + ) public static class Inputs extends RawOpInputs { /** * The desired matrix shape. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java index 36d6fb8013e..d27421d7375 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * Converts a SparseTensor to a (possibly batched) CSRSparseMatrix. */ +@OpMetadata( + opType = SparseTensorToCSRSparseMatrix.OP_NAME, + inputsClass = SparseTensorToCSRSparseMatrix.Inputs.class +) public final class SparseTensorToCSRSparseMatrix extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class SparseTensorToCSRSparseMatrix extends RawOp implements Operan private Output sparseMatrix; @SuppressWarnings("unchecked") - private SparseTensorToCSRSparseMatrix(Operation operation) { - super(operation); + public SparseTensorToCSRSparseMatrix(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseMatrix = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return (Output) sparseMatrix; } + @OpInputsMetadata( + outputsClass = SparseTensorToCSRSparseMatrix.class + ) public static class Inputs extends RawOpInputs { /** * SparseTensor indices. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Abs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Abs.java index c78b34ed12f..c19b0c28eed 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Abs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Abs.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Abs.OP_NAME, + inputsClass = Abs.Inputs.class +) @Operator( group = "math" ) @@ -50,8 +56,8 @@ public final class Abs extends RawOp implements Operand { private Output y; - private Abs(Operation operation) { - super(operation); + public Abs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Abs.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AccumulateN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AccumulateN.java index 7a531d623e8..a958c7f6938 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AccumulateN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AccumulateN.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code sum} output */ +@OpMetadata( + opType = AccumulateN.OP_NAME, + inputsClass = AccumulateN.Inputs.class +) @Operator( group = "math" ) @@ -55,8 +61,8 @@ public final class AccumulateN extends RawOp implements Operand private Output sum; - private AccumulateN(Operation operation) { - super(operation); + public AccumulateN(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sum = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return sum; } + @OpInputsMetadata( + outputsClass = AccumulateN.class + ) public static class Inputs extends RawOpInputs> { /** * A list of {@code Tensor} objects, each with same shape and type. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acos.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acos.java index bc2d95bfe5c..977099004bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acos.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acos.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Acos.OP_NAME, + inputsClass = Acos.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Acos extends RawOp implements Operand { private Output y; - private Acos(Operation operation) { - super(operation); + public Acos(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Acos.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acosh.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acosh.java index d273129375d..39da836345c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acosh.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Acosh.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Acosh.OP_NAME, + inputsClass = Acosh.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class Acosh extends RawOp implements Operand { private Output y; - private Acosh(Operation operation) { - super(operation); + public Acosh(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Acosh.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Add.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Add.java index 19bb9502f1d..ec17877eed3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Add.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Add.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Add.OP_NAME, + inputsClass = Add.Inputs.class +) @Operator( group = "math" ) @@ -51,8 +57,8 @@ public final class Add extends RawOp implements Operand { private Output z; - private Add(Operation operation) { - super(operation); + public Add(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Add.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AddN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AddN.java index 04d7c198272..27f572e53a3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AddN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/AddN.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code sum} output */ +@OpMetadata( + opType = AddN.OP_NAME, + inputsClass = AddN.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class AddN extends RawOp implements Operand { private Output sum; - private AddN(Operation operation) { - super(operation); + public AddN(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sum = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return sum; } + @OpInputsMetadata( + outputsClass = AddN.class + ) public static class Inputs extends RawOpInputs> { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Angle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Angle.java index afb2b79ab48..d0e971a9a7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Angle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Angle.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -52,6 +54,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Angle.OP_NAME, + inputsClass = Angle.Inputs.class +) @Operator( group = "math" ) @@ -63,8 +69,8 @@ public final class Angle extends RawOp implements Operand private Output output; - private Angle(Operation operation) { - super(operation); + public Angle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -117,6 +123,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Angle.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ApproximateEqual.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ApproximateEqual.java index 5a8173def8e..0700aea3d11 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ApproximateEqual.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ApproximateEqual.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -35,6 +37,10 @@ /** * Returns the truth value of abs(x-y) < tolerance element-wise. */ +@OpMetadata( + opType = ApproximateEqual.OP_NAME, + inputsClass = ApproximateEqual.Inputs.class +) @Operator( group = "math" ) @@ -46,8 +52,8 @@ public final class ApproximateEqual extends RawOp implements Operand { private Output z; - private ApproximateEqual(Operation operation) { - super(operation); + public ApproximateEqual(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -125,6 +131,9 @@ public Options tolerance(Float tolerance) { } } + @OpInputsMetadata( + outputsClass = ApproximateEqual.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMax.java index 91b94a54d2b..0d0212311e2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMax.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ArgMax.OP_NAME, + inputsClass = ArgMax.Inputs.class +) @Operator( group = "math" ) @@ -60,8 +66,8 @@ public final class ArgMax extends RawOp implements Operand private Output output; - private ArgMax(Operation operation) { - super(operation); + public ArgMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -122,6 +128,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ArgMax.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMin.java index a5742d5c542..f5c7c4a058f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ArgMin.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ArgMin.OP_NAME, + inputsClass = ArgMin.Inputs.class +) @Operator( group = "math" ) @@ -60,8 +66,8 @@ public final class ArgMin extends RawOp implements Operand private Output output; - private ArgMin(Operation operation) { - super(operation); + public ArgMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -122,6 +128,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ArgMin.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asin.java index 0ab2256368d..764b633187e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +50,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Asin.OP_NAME, + inputsClass = Asin.Inputs.class +) @Operator( group = "math" ) @@ -59,8 +65,8 @@ public final class Asin extends RawOp implements Operand { private Output y; - private Asin(Operation operation) { - super(operation); + public Asin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Asin.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asinh.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asinh.java index bd402f3f550..3483c91a719 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asinh.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Asinh.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Asinh.OP_NAME, + inputsClass = Asinh.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Asinh extends RawOp implements Operand { private Output y; - private Asinh(Operation operation) { - super(operation); + public Asinh(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Asinh.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan.java index 06edc00e74a..1973f102d39 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +50,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Atan.OP_NAME, + inputsClass = Atan.Inputs.class +) @Operator( group = "math" ) @@ -59,8 +65,8 @@ public final class Atan extends RawOp implements Operand { private Output y; - private Atan(Operation operation) { - super(operation); + public Atan(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Atan.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan2.java index 675120effab..3458cf38200 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atan2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -52,6 +54,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Atan2.OP_NAME, + inputsClass = Atan2.Inputs.class +) @Operator( group = "math" ) @@ -63,8 +69,8 @@ public final class Atan2 extends RawOp implements Operand private Output z; - private Atan2(Operation operation) { - super(operation); + public Atan2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -102,6 +108,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Atan2.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atanh.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atanh.java index c9aff2cf3a3..bd0c86a60db 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atanh.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Atanh.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -45,6 +47,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Atanh.OP_NAME, + inputsClass = Atanh.Inputs.class +) @Operator( group = "math" ) @@ -56,8 +62,8 @@ public final class Atanh extends RawOp implements Operand { private Output y; - private Atanh(Operation operation) { - super(operation); + public Atanh(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Atanh.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java index c7824375420..62505b7e4e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselI0.OP_NAME, + inputsClass = BesselI0.Inputs.class +) public final class BesselI0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselI0 extends RawOp implements Operand< private Output y; - private BesselI0(Operation operation) { - super(operation); + public BesselI0(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselI0.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java index a557e9f294e..1386e6ceefb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselI0e.OP_NAME, + inputsClass = BesselI0e.Inputs.class +) public final class BesselI0e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselI0e extends RawOp implements Operand private Output y; - private BesselI0e(Operation operation) { - super(operation); + public BesselI0e(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselI0e.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java index 3b5362106b7..16a2a62d0b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselI1.OP_NAME, + inputsClass = BesselI1.Inputs.class +) public final class BesselI1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselI1 extends RawOp implements Operand< private Output y; - private BesselI1(Operation operation) { - super(operation); + public BesselI1(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselI1.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java index 00461637760..4390f61cde7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselI1e.OP_NAME, + inputsClass = BesselI1e.Inputs.class +) public final class BesselI1e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselI1e extends RawOp implements Operand private Output y; - private BesselI1e(Operation operation) { - super(operation); + public BesselI1e(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselI1e.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Betainc.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Betainc.java index 3e873075268..b4c5bdf1a77 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Betainc.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Betainc.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Betainc.OP_NAME, + inputsClass = Betainc.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class Betainc extends RawOp implements Operand z; - private Betainc(Operation operation) { - super(operation); + public Betainc(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Betainc.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Bincount.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Bincount.java index ae7d1475d31..e7f6bc53875 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Bincount.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Bincount.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -43,6 +45,10 @@ * * @param data type for {@code bins} output */ +@OpMetadata( + opType = Bincount.OP_NAME, + inputsClass = Bincount.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Bincount extends RawOp implements Operand< private Output bins; - private Bincount(Operation operation) { - super(operation); + public Bincount(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; bins = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return bins; } + @OpInputsMetadata( + outputsClass = Bincount.class + ) public static class Inputs extends RawOpInputs> { /** * int32 {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ceil.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ceil.java index f27874e0c69..58dd4718ee7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ceil.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ceil.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Ceil.OP_NAME, + inputsClass = Ceil.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Ceil extends RawOp implements Operand { private Output y; - private Ceil(Operation operation) { - super(operation); + public Ceil(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Ceil.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ComplexAbs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ComplexAbs.java index c03e3d836d1..b01366920ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ComplexAbs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ComplexAbs.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -53,6 +55,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = ComplexAbs.OP_NAME, + inputsClass = ComplexAbs.Inputs.class +) @Operator( group = "math" ) @@ -64,8 +70,8 @@ public final class ComplexAbs extends RawOp implements Operan private Output y; - private ComplexAbs(Operation operation) { - super(operation); + public ComplexAbs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -118,6 +124,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = ComplexAbs.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Conj.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Conj.java index c07c7b6d9dd..9878f1cbc55 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Conj.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Conj.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conj.OP_NAME, + inputsClass = Conj.Inputs.class +) @Operator( group = "math" ) @@ -57,8 +63,8 @@ public final class Conj extends RawOp implements Operand { private Output output; - private Conj(Operation operation) { - super(operation); + public Conj(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Conj.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cos.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cos.java index ddfe40b21dd..ebae2e0ab01 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cos.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cos.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Cos.OP_NAME, + inputsClass = Cos.Inputs.class +) @Operator( group = "math" ) @@ -55,8 +61,8 @@ public final class Cos extends RawOp implements Operand { private Output y; - private Cos(Operation operation) { - super(operation); + public Cos(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Cos.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cosh.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cosh.java index 4cebf838097..ff272324595 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cosh.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cosh.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Cosh.OP_NAME, + inputsClass = Cosh.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Cosh extends RawOp implements Operand { private Output y; - private Cosh(Operation operation) { - super(operation); + public Cosh(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Cosh.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumprod.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumprod.java index fa818c7aa06..19b5ff33e5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumprod.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumprod.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +59,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = Cumprod.OP_NAME, + inputsClass = Cumprod.Inputs.class +) @Operator( group = "math" ) @@ -68,8 +74,8 @@ public final class Cumprod extends RawOp implements Operand private Output out; - private Cumprod(Operation operation) { - super(operation); + public Cumprod(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -176,6 +182,9 @@ public Options reverse(Boolean reverse) { } } + @OpInputsMetadata( + outputsClass = Cumprod.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor}. Must be one of the following types: {@code float32}, {@code float64}, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumsum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumsum.java index f0b4b9b2e45..a5ace804137 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumsum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Cumsum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +59,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = Cumsum.OP_NAME, + inputsClass = Cumsum.Inputs.class +) @Operator( group = "math" ) @@ -68,8 +74,8 @@ public final class Cumsum extends RawOp implements Operand { private Output out; - private Cumsum(Operation operation) { - super(operation); + public Cumsum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -176,6 +182,9 @@ public Options reverse(Boolean reverse) { } } + @OpInputsMetadata( + outputsClass = Cumsum.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor}. Must be one of the following types: {@code float32}, {@code float64}, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java index ca5448a9960..b727b2aa70c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +53,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = CumulativeLogsumexp.OP_NAME, + inputsClass = CumulativeLogsumexp.Inputs.class +) public final class CumulativeLogsumexp extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -59,8 +65,8 @@ public final class CumulativeLogsumexp extends RawOp implemen private Output out; - private CumulativeLogsumexp(Operation operation) { - super(operation); + public CumulativeLogsumexp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -165,6 +171,9 @@ public Options reverse(Boolean reverse) { } } + @OpInputsMetadata( + outputsClass = CumulativeLogsumexp.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor}. Must be one of the following types: {@code float16}, {@code float32}, {@code float64}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DenseBincount.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DenseBincount.java index 556c582b9d6..24949d906ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DenseBincount.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DenseBincount.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DenseBincount.OP_NAME, + inputsClass = DenseBincount.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class DenseBincount extends RawOp implements Ope private Output output; - private DenseBincount(Operation operation) { - super(operation); + public DenseBincount(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -138,6 +144,9 @@ public Options binaryOutput(Boolean binaryOutput) { } } + @OpInputsMetadata( + outputsClass = DenseBincount.class + ) public static class Inputs extends RawOpInputs> { /** * 1D or 2D int {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Digamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Digamma.java index 7d50c38999a..3c4b64d60e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Digamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Digamma.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Digamma.OP_NAME, + inputsClass = Digamma.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Digamma extends RawOp implements Operand y; - private Digamma(Operation operation) { - super(operation); + public Digamma(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Digamma.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Div.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Div.java index 7a8a0ca06de..e95a4b3da69 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Div.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Div.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Div.OP_NAME, + inputsClass = Div.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Div extends RawOp implements Operand { private Output z; - private Div(Operation operation) { - super(operation); + public Div(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Div.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DivNoNan.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DivNoNan.java index d21762c3127..ffac20c160b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DivNoNan.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/DivNoNan.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = DivNoNan.OP_NAME, + inputsClass = DivNoNan.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class DivNoNan extends RawOp implements Operand private Output z; - private DivNoNan(Operation operation) { - super(operation); + public DivNoNan(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = DivNoNan.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Equal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Equal.java index 625624b428c..2695c76a5e9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Equal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Equal.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -46,6 +48,10 @@ * tf.math.equal(x, y) ==> array([True, True]) * */ +@OpMetadata( + opType = Equal.OP_NAME, + inputsClass = Equal.Inputs.class +) @Operator( group = "math" ) @@ -57,8 +63,8 @@ public final class Equal extends RawOp implements Operand { private Output z; - private Equal(Operation operation) { - super(operation); + public Equal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -136,6 +142,9 @@ public Options incompatibleShapeError(Boolean incompatibleShapeError) { } } + @OpInputsMetadata( + outputsClass = Equal.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erf.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erf.java index 8960e9851f5..fe8a0ed2afc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erf.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erf.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Erf.OP_NAME, + inputsClass = Erf.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Erf extends RawOp implements Operand { private Output y; - private Erf(Operation operation) { - super(operation); + public Erf(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Erf.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erfc.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erfc.java index c24a60a10a4..cbc503be446 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erfc.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Erfc.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Erfc.OP_NAME, + inputsClass = Erfc.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Erfc extends RawOp implements Operand { private Output y; - private Erfc(Operation operation) { - super(operation); + public Erfc(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Erfc.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Exp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Exp.java index edc260dd2ae..0ddf4078f0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Exp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Exp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -57,6 +59,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Exp.OP_NAME, + inputsClass = Exp.Inputs.class +) @Operator( group = "math" ) @@ -68,8 +74,8 @@ public final class Exp extends RawOp implements Operand { private Output y; - private Exp(Operation operation) { - super(operation); + public Exp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Exp.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Expm1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Expm1.java index 29e89e11ae6..d15593d2071 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Expm1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Expm1.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +50,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Expm1.OP_NAME, + inputsClass = Expm1.Inputs.class +) @Operator( group = "math" ) @@ -59,8 +65,8 @@ public final class Expm1 extends RawOp implements Operand { private Output y; - private Expm1(Operation operation) { - super(operation); + public Expm1(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Expm1.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Fact.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Fact.java index 95aa9162307..91e819c29a3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Fact.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Fact.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** * Output a fact about factorials. */ +@OpMetadata( + opType = Fact.OP_NAME, + inputsClass = Fact.Inputs.class +) @Operator( group = "math" ) @@ -44,8 +50,8 @@ public final class Fact extends RawOp implements Operand { private Output fact; - private Fact(Operation operation) { - super(operation); + public Fact(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; fact = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return fact; } + @OpInputsMetadata( + outputsClass = Fact.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new Fact(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Floor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Floor.java index a83480aa9b9..5d539cfe495 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Floor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Floor.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Floor.OP_NAME, + inputsClass = Floor.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Floor extends RawOp implements Operand private Output y; - private Floor(Operation operation) { - super(operation); + public Floor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Floor.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorDiv.java index bb8adeeb63d..619ad282bc9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorDiv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = FloorDiv.OP_NAME, + inputsClass = FloorDiv.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class FloorDiv extends RawOp implements Operand private Output z; - private FloorDiv(Operation operation) { - super(operation); + public FloorDiv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = FloorDiv.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorMod.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorMod.java index 61d7d5b5cfa..63963d5b3f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorMod.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/FloorMod.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = FloorMod.OP_NAME, + inputsClass = FloorMod.Inputs.class +) @Operator( group = "math" ) @@ -51,8 +57,8 @@ public final class FloorMod extends RawOp implements Operand< private Output z; - private FloorMod(Operation operation) { - super(operation); + public FloorMod(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = FloorMod.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Greater.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Greater.java index 57504b1d508..c566f003533 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Greater.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Greater.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -47,6 +49,10 @@ * tf.math.greater(x, y) ==> [False, False, True] * */ +@OpMetadata( + opType = Greater.OP_NAME, + inputsClass = Greater.Inputs.class +) @Operator( group = "math" ) @@ -58,8 +64,8 @@ public final class Greater extends RawOp implements Operand { private Output z; - private Greater(Operation operation) { - super(operation); + public Greater(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Greater.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/GreaterEqual.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/GreaterEqual.java index 9a38c6b5fa8..64589450029 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/GreaterEqual.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/GreaterEqual.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -47,6 +49,10 @@ * tf.math.greater_equal(x, y) ==> [True, False, True, True] * */ +@OpMetadata( + opType = GreaterEqual.OP_NAME, + inputsClass = GreaterEqual.Inputs.class +) @Operator( group = "math" ) @@ -58,8 +64,8 @@ public final class GreaterEqual extends RawOp implements Operand { private Output z; - private GreaterEqual(Operation operation) { - super(operation); + public GreaterEqual(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = GreaterEqual.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igamma.java index 5bb9bc21881..383c0d95434 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igamma.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -43,6 +45,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Igamma.OP_NAME, + inputsClass = Igamma.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Igamma extends RawOp implements Operand private Output z; - private Igamma(Operation operation) { - super(operation); + public Igamma(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Igamma.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java index f223cb5a1a1..058b27f96d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = IgammaGradA.OP_NAME, + inputsClass = IgammaGradA.Inputs.class +) public final class IgammaGradA extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class IgammaGradA extends RawOp implements Opera private Output z; - private IgammaGradA(Operation operation) { - super(operation); + public IgammaGradA(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = IgammaGradA.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igammac.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igammac.java index 356f1129187..92ded48b24e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igammac.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Igammac.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -43,6 +45,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Igammac.OP_NAME, + inputsClass = Igammac.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Igammac extends RawOp implements Operand z; - private Igammac(Operation operation) { - super(operation); + public Igammac(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Igammac.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Imag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Imag.java index 10eb163fabe..8f1eedb2865 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Imag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Imag.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Imag.OP_NAME, + inputsClass = Imag.Inputs.class +) @Operator( group = "math" ) @@ -59,8 +65,8 @@ public final class Imag extends RawOp implements Operand { private Output output; - private Imag(Operation operation) { - super(operation); + public Imag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Imag.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/InvertPermutation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/InvertPermutation.java index 570c0408231..b83e3c7687b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/InvertPermutation.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/InvertPermutation.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -47,6 +49,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = InvertPermutation.OP_NAME, + inputsClass = InvertPermutation.Inputs.class +) @Operator( group = "math" ) @@ -58,8 +64,8 @@ public final class InvertPermutation extends RawOp implements private Output y; - private InvertPermutation(Operation operation) { - super(operation); + public InvertPermutation(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = InvertPermutation.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsFinite.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsFinite.java index b134a350c34..db8c3cfac33 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsFinite.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsFinite.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -43,6 +45,10 @@ * tf.math.is_finite(x) ==> [True, True, True, False, False] * */ +@OpMetadata( + opType = IsFinite.OP_NAME, + inputsClass = IsFinite.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class IsFinite extends RawOp implements Operand { private Output y; - private IsFinite(Operation operation) { - super(operation); + public IsFinite(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = IsFinite.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsInf.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsInf.java index 1b111be0a5d..6d7a74a693c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsInf.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsInf.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -43,6 +45,10 @@ * tf.math.is_inf(x) ==> [False, True, False, True] * */ +@OpMetadata( + opType = IsInf.OP_NAME, + inputsClass = IsInf.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class IsInf extends RawOp implements Operand { private Output y; - private IsInf(Operation operation) { - super(operation); + public IsInf(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = IsInf.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsNan.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsNan.java index 5d3bbc59140..2313dc981f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsNan.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IsNan.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -43,6 +45,10 @@ * tf.math.is_nan(x) ==> [False, True, False, True, False] * */ +@OpMetadata( + opType = IsNan.OP_NAME, + inputsClass = IsNan.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class IsNan extends RawOp implements Operand { private Output y; - private IsNan(Operation operation) { - super(operation); + public IsNan(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = IsNan.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Less.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Less.java index ab28b8e688d..ed99ef4e7dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Less.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Less.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -47,6 +49,10 @@ * tf.math.less(x, y) ==> [False, True, True] * */ +@OpMetadata( + opType = Less.OP_NAME, + inputsClass = Less.Inputs.class +) @Operator( group = "math" ) @@ -58,8 +64,8 @@ public final class Less extends RawOp implements Operand { private Output z; - private Less(Operation operation) { - super(operation); + public Less(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Less.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LessEqual.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LessEqual.java index de450dd1f93..b090bd12112 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LessEqual.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LessEqual.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -47,6 +49,10 @@ * tf.math.less_equal(x, y) ==> [True, True, True] * */ +@OpMetadata( + opType = LessEqual.OP_NAME, + inputsClass = LessEqual.Inputs.class +) @Operator( group = "math" ) @@ -58,8 +64,8 @@ public final class LessEqual extends RawOp implements Operand { private Output z; - private LessEqual(Operation operation) { - super(operation); + public LessEqual(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = LessEqual.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Lgamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Lgamma.java index 4881d6ca779..ee474789a0a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Lgamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Lgamma.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -43,6 +45,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Lgamma.OP_NAME, + inputsClass = Lgamma.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Lgamma extends RawOp implements Operand private Output y; - private Lgamma(Operation operation) { - super(operation); + public Lgamma(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Lgamma.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log.java index 098a1d0ff6a..ee2e976f3ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Log.OP_NAME, + inputsClass = Log.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class Log extends RawOp implements Operand { private Output y; - private Log(Operation operation) { - super(operation); + public Log(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Log.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log1p.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log1p.java index 2c4a27b0be2..78e42d010d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log1p.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Log1p.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Log1p.OP_NAME, + inputsClass = Log1p.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class Log1p extends RawOp implements Operand { private Output y; - private Log1p(Operation operation) { - super(operation); + public Log1p(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Log1p.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalAnd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalAnd.java index b2b1329ef87..455bc3db20c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalAnd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalAnd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; @@ -35,6 +37,10 @@ * NOTE: {@code math.LogicalAnd} supports broadcasting. More about broadcasting * here */ +@OpMetadata( + opType = LogicalAnd.OP_NAME, + inputsClass = LogicalAnd.Inputs.class +) @Operator( group = "math" ) @@ -46,8 +52,8 @@ public final class LogicalAnd extends RawOp implements Operand { private Output z; - private LogicalAnd(Operation operation) { - super(operation); + public LogicalAnd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = LogicalAnd.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalNot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalNot.java index 65de99bbfcb..38e71e4cb83 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalNot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalNot.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; /** * Returns the truth value of {@code NOT x} element-wise. */ +@OpMetadata( + opType = LogicalNot.OP_NAME, + inputsClass = LogicalNot.Inputs.class +) @Operator( group = "math" ) @@ -44,8 +50,8 @@ public final class LogicalNot extends RawOp implements Operand { private Output y; - private LogicalNot(Operation operation) { - super(operation); + public LogicalNot(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = LogicalNot.class + ) public static class Inputs extends RawOpInputs { /** * A {@code Tensor} of type {@code bool}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalOr.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalOr.java index 6c7b7c127db..d066fc83024 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalOr.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/LogicalOr.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; @@ -35,6 +37,10 @@ * NOTE: {@code math.LogicalOr} supports broadcasting. More about broadcasting * here */ +@OpMetadata( + opType = LogicalOr.OP_NAME, + inputsClass = LogicalOr.Inputs.class +) @Operator( group = "math" ) @@ -46,8 +52,8 @@ public final class LogicalOr extends RawOp implements Operand { private Output z; - private LogicalOr(Operation operation) { - super(operation); + public LogicalOr(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = LogicalOr.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Maximum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Maximum.java index f467d171afc..0f94fa56c93 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Maximum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Maximum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Maximum.OP_NAME, + inputsClass = Maximum.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Maximum extends RawOp implements Operand z; - private Maximum(Operation operation) { - super(operation); + public Maximum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Maximum.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mean.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mean.java index 122567fd9ae..abc6db045a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mean.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mean.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Mean.OP_NAME, + inputsClass = Mean.Inputs.class +) @Operator( group = "math" ) @@ -52,8 +58,8 @@ public final class Mean extends RawOp implements Operand { private Output output; - private Mean(Operation operation) { - super(operation); + public Mean(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -132,6 +138,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = Mean.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to reduce. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Minimum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Minimum.java index facf4cb1560..ebea75921af 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Minimum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Minimum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Minimum.OP_NAME, + inputsClass = Minimum.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Minimum extends RawOp implements Operand z; - private Minimum(Operation operation) { - super(operation); + public Minimum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Minimum.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mod.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mod.java index 8838ae84a78..2221ecce333 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mod.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mod.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Mod.OP_NAME, + inputsClass = Mod.Inputs.class +) @Operator( group = "math" ) @@ -51,8 +57,8 @@ public final class Mod extends RawOp implements Operand { private Output z; - private Mod(Operation operation) { - super(operation); + public Mod(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Mod.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mul.java index 28cd75c9810..864e3d3fd6a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Mul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Mul.OP_NAME, + inputsClass = Mul.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Mul extends RawOp implements Operand { private Output z; - private Mul(Operation operation) { - super(operation); + public Mul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Mul.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/MulNoNan.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/MulNoNan.java index a5a904d2bd4..845947687f7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/MulNoNan.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/MulNoNan.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = MulNoNan.OP_NAME, + inputsClass = MulNoNan.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class MulNoNan extends RawOp implements Operand private Output z; - private MulNoNan(Operation operation) { - super(operation); + public MulNoNan(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = MulNoNan.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ndtri.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ndtri.java index 832e0e067ea..651438a8523 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ndtri.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Ndtri.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Ndtri.OP_NAME, + inputsClass = Ndtri.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Ndtri extends RawOp implements Operand private Output y; - private Ndtri(Operation operation) { - super(operation); + public Ndtri(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Ndtri.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Neg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Neg.java index 961355ca9af..2124815e5c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Neg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Neg.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Neg.OP_NAME, + inputsClass = Neg.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Neg extends RawOp implements Operand { private Output y; - private Neg(Operation operation) { - super(operation); + public Neg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Neg.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NextAfter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NextAfter.java index 3f96ce897d7..c53812ad539 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NextAfter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NextAfter.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = NextAfter.OP_NAME, + inputsClass = NextAfter.Inputs.class +) @Operator( group = "math" ) @@ -52,8 +58,8 @@ public final class NextAfter extends RawOp implements Operand private Output output; - private NextAfter(Operation operation) { - super(operation); + public NextAfter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = NextAfter.class + ) public static class Inputs extends RawOpInputs> { /** * The x1 input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NotEqual.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NotEqual.java index b579d881c1e..60ad0e8b773 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NotEqual.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/NotEqual.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -37,6 +39,10 @@ * NOTE: {@code math.NotEqual} supports broadcasting. More about broadcasting * here */ +@OpMetadata( + opType = NotEqual.OP_NAME, + inputsClass = NotEqual.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class NotEqual extends RawOp implements Operand { private Output z; - private NotEqual(Operation operation) { - super(operation); + public NotEqual(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options incompatibleShapeError(Boolean incompatibleShapeError) { } } + @OpInputsMetadata( + outputsClass = NotEqual.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Polygamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Polygamma.java index 2b7aacb5503..a3d26a80452 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Polygamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Polygamma.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Polygamma.OP_NAME, + inputsClass = Polygamma.Inputs.class +) @Operator( group = "math" ) @@ -51,8 +57,8 @@ public final class Polygamma extends RawOp implements Operand private Output z; - private Polygamma(Operation operation) { - super(operation); + public Polygamma(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Polygamma.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/PopulationCount.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/PopulationCount.java index c49751ed2ae..bb751df6597 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/PopulationCount.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/PopulationCount.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TUint8; @@ -40,6 +42,10 @@ * {@code int32} or {@code int64} and perform the bitcount on the result, than to feed in * 8- or 16-bit inputs and then aggregate the resulting counts. */ +@OpMetadata( + opType = PopulationCount.OP_NAME, + inputsClass = PopulationCount.Inputs.class +) @Operator( group = "math" ) @@ -51,8 +57,8 @@ public final class PopulationCount extends RawOp implements Operand { private Output y; - private PopulationCount(Operation operation) { - super(operation); + public PopulationCount(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = PopulationCount.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Pow.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Pow.java index d7a3059dc65..dca15c8146d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Pow.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Pow.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Pow.OP_NAME, + inputsClass = Pow.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Pow extends RawOp implements Operand { private Output z; - private Pow(Operation operation) { - super(operation); + public Pow(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Pow.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedAdd.java index 49e68db93ce..d120d55bf02 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedAdd.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = QuantizedAdd.OP_NAME, + inputsClass = QuantizedAdd.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class QuantizedAdd extends RawOp { private Output maxZ; - private QuantizedAdd(Operation operation) { - super(operation); + public QuantizedAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); minZ = operation.output(outputIdx++); @@ -121,6 +127,9 @@ public Output maxZ() { return maxZ; } + @OpInputsMetadata( + outputsClass = QuantizedAdd.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedMul.java index 3861db85f3a..6042eb2e048 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/QuantizedMul.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = QuantizedMul.OP_NAME, + inputsClass = QuantizedMul.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class QuantizedMul extends RawOp { private Output maxZ; - private QuantizedMul(Operation operation) { - super(operation); + public QuantizedMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); minZ = operation.output(outputIdx++); @@ -121,6 +127,9 @@ public Output maxZ() { return maxZ; } + @OpInputsMetadata( + outputsClass = QuantizedMul.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Real.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Real.java index b4d443f9083..e00013a7f4c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Real.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Real.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Real.OP_NAME, + inputsClass = Real.Inputs.class +) @Operator( group = "math" ) @@ -59,8 +65,8 @@ public final class Real extends RawOp implements Operand { private Output output; - private Real(Operation operation) { - super(operation); + public Real(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Real.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RealDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RealDiv.java index 3b301133fa1..e3f48312429 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RealDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RealDiv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RealDiv.OP_NAME, + inputsClass = RealDiv.Inputs.class +) @Operator( group = "math" ) @@ -50,8 +56,8 @@ public final class RealDiv extends RawOp implements Operand private Output z; - private RealDiv(Operation operation) { - super(operation); + public RealDiv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RealDiv.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Reciprocal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Reciprocal.java index 637d25d959b..7518dd7bd6b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Reciprocal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Reciprocal.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Reciprocal.OP_NAME, + inputsClass = Reciprocal.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Reciprocal extends RawOp implements Operand< private Output y; - private Reciprocal(Operation operation) { - super(operation); + public Reciprocal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Reciprocal.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java index 579fc948e26..24b41cfb5e0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = ReciprocalGrad.OP_NAME, + inputsClass = ReciprocalGrad.Inputs.class +) public final class ReciprocalGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class ReciprocalGrad extends RawOp implements Oper private Output z; - private ReciprocalGrad(Operation operation) { - super(operation); + public ReciprocalGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = ReciprocalGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java index 121255c20a4..3d5a6435e08 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * Computes requantization range per channel. */ +@OpMetadata( + opType = RequantizationRangePerChannel.OP_NAME, + inputsClass = RequantizationRangePerChannel.Inputs.class +) public final class RequantizationRangePerChannel extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RequantizationRangePerChannel extends RawOp { private Output outputMax; - private RequantizationRangePerChannel(Operation operation) { - super(operation); + public RequantizationRangePerChannel(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputMin = operation.output(outputIdx++); outputMax = operation.output(outputIdx++); @@ -93,6 +99,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = RequantizationRangePerChannel.class + ) public static class Inputs extends RawOpInputs { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java index f3a513f4164..d8913c86743 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RequantizePerChannel.OP_NAME, + inputsClass = RequantizePerChannel.Inputs.class +) public final class RequantizePerChannel extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RequantizePerChannel extends RawOp { private Output outputMax; - private RequantizePerChannel(Operation operation) { - super(operation); + public RequantizePerChannel(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputMin = operation.output(outputIdx++); @@ -114,6 +120,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = RequantizePerChannel.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rint.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rint.java index 7014e1f8921..1886c3b06a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rint.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rint.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +46,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Rint.OP_NAME, + inputsClass = Rint.Inputs.class +) @Operator( group = "math" ) @@ -55,8 +61,8 @@ public final class Rint extends RawOp implements Operand { private Output y; - private Rint(Operation operation) { - super(operation); + public Rint(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Rint.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Round.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Round.java index 1d079601ebd..ebde5c2358c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Round.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Round.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Round.OP_NAME, + inputsClass = Round.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Round extends RawOp implements Operand { private Output y; - private Round(Operation operation) { - super(operation); + public Round(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Round.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rsqrt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rsqrt.java index 9bf061a5192..081368e6f02 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rsqrt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Rsqrt.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Rsqrt.OP_NAME, + inputsClass = Rsqrt.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Rsqrt extends RawOp implements Operand { private Output y; - private Rsqrt(Operation operation) { - super(operation); + public Rsqrt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Rsqrt.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java index 8874d5a8b00..16e114dc591 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RsqrtGrad.OP_NAME, + inputsClass = RsqrtGrad.Inputs.class +) public final class RsqrtGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RsqrtGrad extends RawOp implements Operand z; - private RsqrtGrad(Operation operation) { - super(operation); + public RsqrtGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RsqrtGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMax.java index 57098d1f87e..167e9ecf905 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SegmentMax.OP_NAME, + inputsClass = SegmentMax.Inputs.class +) @Operator( group = "math" ) @@ -64,8 +70,8 @@ public final class SegmentMax extends RawOp implements Operan private Output output; - private SegmentMax(Operation operation) { - super(operation); + public SegmentMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SegmentMax.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMean.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMean.java index afd9e32a207..715c0098dce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMean.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMean.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -55,6 +57,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SegmentMean.OP_NAME, + inputsClass = SegmentMean.Inputs.class +) @Operator( group = "math" ) @@ -66,8 +72,8 @@ public final class SegmentMean extends RawOp implements Operand private Output output; - private SegmentMean(Operation operation) { - super(operation); + public SegmentMean(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -108,6 +114,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SegmentMean.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMin.java index a32e370e3d3..b867b9fd972 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SegmentMin.OP_NAME, + inputsClass = SegmentMin.Inputs.class +) @Operator( group = "math" ) @@ -64,8 +70,8 @@ public final class SegmentMin extends RawOp implements Operan private Output output; - private SegmentMin(Operation operation) { - super(operation); + public SegmentMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SegmentMin.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentProd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentProd.java index 192a5367e3c..f2857db52cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentProd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentProd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -54,6 +56,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SegmentProd.OP_NAME, + inputsClass = SegmentProd.Inputs.class +) @Operator( group = "math" ) @@ -65,8 +71,8 @@ public final class SegmentProd extends RawOp implements Operand private Output output; - private SegmentProd(Operation operation) { - super(operation); + public SegmentProd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SegmentProd.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentSum.java index 10f814320f8..bfdd2182153 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SegmentSum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -54,6 +56,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SegmentSum.OP_NAME, + inputsClass = SegmentSum.Inputs.class +) @Operator( group = "math" ) @@ -65,8 +71,8 @@ public final class SegmentSum extends RawOp implements Operand< private Output output; - private SegmentSum(Operation operation) { - super(operation); + public SegmentSum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SegmentSum.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sigmoid.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sigmoid.java index 2716b3cf8b8..d9ca11b4d48 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sigmoid.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sigmoid.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Sigmoid.OP_NAME, + inputsClass = Sigmoid.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Sigmoid extends RawOp implements Operand private Output y; - private Sigmoid(Operation operation) { - super(operation); + public Sigmoid(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Sigmoid.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java index 55d174c2eba..d89e6603f98 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = SigmoidGrad.OP_NAME, + inputsClass = SigmoidGrad.Inputs.class +) public final class SigmoidGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class SigmoidGrad extends RawOp implements Operand private Output z; - private SigmoidGrad(Operation operation) { - super(operation); + public SigmoidGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = SigmoidGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sign.java index 4acd2234734..e9673f31954 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sign.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -47,6 +49,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Sign.OP_NAME, + inputsClass = Sign.Inputs.class +) @Operator( group = "math" ) @@ -58,8 +64,8 @@ public final class Sign extends RawOp implements Operand { private Output y; - private Sign(Operation operation) { - super(operation); + public Sign(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Sign.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sin.java index 2bf226a61ad..a174e6bfcb8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Sin.OP_NAME, + inputsClass = Sin.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Sin extends RawOp implements Operand { private Output y; - private Sin(Operation operation) { - super(operation); + public Sin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Sin.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sinh.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sinh.java index 4486f6576fb..4d817c8cba8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sinh.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sinh.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Sinh.OP_NAME, + inputsClass = Sinh.Inputs.class +) @Operator( group = "math" ) @@ -54,8 +60,8 @@ public final class Sinh extends RawOp implements Operand { private Output y; - private Sinh(Operation operation) { - super(operation); + public Sinh(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Sinh.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java index 4e31c283009..3301f457584 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code samples} output */ +@OpMetadata( + opType = SobolSample.OP_NAME, + inputsClass = SobolSample.Inputs.class +) public final class SobolSample extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class SobolSample extends RawOp implements Opera private Output samples; - private SobolSample(Operation operation) { - super(operation); + public SobolSample(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; samples = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return samples; } + @OpInputsMetadata( + outputsClass = SobolSample.class + ) public static class Inputs extends RawOpInputs> { /** * Positive scalar {@code Tensor} representing each sample's dimension. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Softplus.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Softplus.java index 75a59a3389e..081ba524a08 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Softplus.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Softplus.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = Softplus.OP_NAME, + inputsClass = Softplus.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Softplus extends RawOp implements Operand< private Output activations; - private Softplus(Operation operation) { - super(operation); + public Softplus(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return activations; } + @OpInputsMetadata( + outputsClass = Softplus.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java index dd9f8637498..76caf9ca811 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = SoftplusGrad.OP_NAME, + inputsClass = SoftplusGrad.Inputs.class +) public final class SoftplusGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class SoftplusGrad extends RawOp implements Oper private Output backprops; - private SoftplusGrad(Operation operation) { - super(operation); + public SoftplusGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return backprops; } + @OpInputsMetadata( + outputsClass = SoftplusGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding softplus operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sqrt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sqrt.java index ffbc55ab620..93ce7497ba3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sqrt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sqrt.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Sqrt.OP_NAME, + inputsClass = Sqrt.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Sqrt extends RawOp implements Operand { private Output y; - private Sqrt(Operation operation) { - super(operation); + public Sqrt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Sqrt.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java index 8bdef3778d0..44150536e1b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = SqrtGrad.OP_NAME, + inputsClass = SqrtGrad.Inputs.class +) public final class SqrtGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class SqrtGrad extends RawOp implements Operand private Output z; - private SqrtGrad(Operation operation) { - super(operation); + public SqrtGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = SqrtGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Square.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Square.java index 52f321ef0d5..36b966a671d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Square.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Square.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Square.OP_NAME, + inputsClass = Square.Inputs.class +) @Operator( group = "math" ) @@ -48,8 +54,8 @@ public final class Square extends RawOp implements Operand { private Output y; - private Square(Operation operation) { - super(operation); + public Square(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Square.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SquaredDifference.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SquaredDifference.java index a71e97d7c5f..46b06c4daa0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SquaredDifference.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SquaredDifference.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = SquaredDifference.OP_NAME, + inputsClass = SquaredDifference.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class SquaredDifference extends RawOp implements O private Output z; - private SquaredDifference(Operation operation) { - super(operation); + public SquaredDifference(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = SquaredDifference.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sub.java index c415449aefc..147016b4975 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Sub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Sub.OP_NAME, + inputsClass = Sub.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Sub extends RawOp implements Operand { private Output z; - private Sub(Operation operation) { - super(operation); + public Sub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Sub.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tan.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tan.java index 80f4bb51b53..9c8f23a2e60 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tan.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tan.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Tan.OP_NAME, + inputsClass = Tan.Inputs.class +) @Operator( group = "math" ) @@ -55,8 +61,8 @@ public final class Tan extends RawOp implements Operand { private Output y; - private Tan(Operation operation) { - super(operation); + public Tan(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Tan.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tanh.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tanh.java index b4a3cdbe76e..c9e2f43ca94 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tanh.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Tanh.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -50,6 +52,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Tanh.OP_NAME, + inputsClass = Tanh.Inputs.class +) @Operator( group = "math" ) @@ -61,8 +67,8 @@ public final class Tanh extends RawOp implements Operand { private Output y; - private Tanh(Operation operation) { - super(operation); + public Tanh(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Tanh.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java index 287eda24c21..7dd6d1ef4d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = TanhGrad.OP_NAME, + inputsClass = TanhGrad.Inputs.class +) public final class TanhGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class TanhGrad extends RawOp implements Operand private Output z; - private TanhGrad(Operation operation) { - super(operation); + public TanhGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = TanhGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateDiv.java index 04461701a1e..ece8ab49c44 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateDiv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = TruncateDiv.OP_NAME, + inputsClass = TruncateDiv.Inputs.class +) @Operator( group = "math" ) @@ -53,8 +59,8 @@ public final class TruncateDiv extends RawOp implements Operand private Output z; - private TruncateDiv(Operation operation) { - super(operation); + public TruncateDiv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = TruncateDiv.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateMod.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateMod.java index 5e0aeadeaae..3fa76d9f178 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateMod.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TruncateMod.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = TruncateMod.OP_NAME, + inputsClass = TruncateMod.Inputs.class +) @Operator( group = "math" ) @@ -50,8 +56,8 @@ public final class TruncateMod extends RawOp implements Opera private Output z; - private TruncateMod(Operation operation) { - super(operation); + public TruncateMod(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = TruncateMod.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMax.java index fd872df8540..785503c0af5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -59,6 +61,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = UnsortedSegmentMax.OP_NAME, + inputsClass = UnsortedSegmentMax.Inputs.class +) @Operator( group = "math" ) @@ -70,8 +76,8 @@ public final class UnsortedSegmentMax extends RawOp implement private Output output; - private UnsortedSegmentMax(Operation operation) { - super(operation); + public UnsortedSegmentMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -114,6 +120,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UnsortedSegmentMax.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMin.java index 5a010ae3b5f..541a0f15b47 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -56,6 +58,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = UnsortedSegmentMin.OP_NAME, + inputsClass = UnsortedSegmentMin.Inputs.class +) @Operator( group = "math" ) @@ -67,8 +73,8 @@ public final class UnsortedSegmentMin extends RawOp implement private Output output; - private UnsortedSegmentMin(Operation operation) { - super(operation); + public UnsortedSegmentMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -111,6 +117,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UnsortedSegmentMin.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentProd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentProd.java index e54296d0a47..4d80fde16df 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentProd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentProd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -56,6 +58,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = UnsortedSegmentProd.OP_NAME, + inputsClass = UnsortedSegmentProd.Inputs.class +) @Operator( group = "math" ) @@ -67,8 +73,8 @@ public final class UnsortedSegmentProd extends RawOp implements private Output output; - private UnsortedSegmentProd(Operation operation) { - super(operation); + public UnsortedSegmentProd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -111,6 +117,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UnsortedSegmentProd.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentSum.java index 037711460b0..8c271e284e9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UnsortedSegmentSum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -58,6 +60,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = UnsortedSegmentSum.OP_NAME, + inputsClass = UnsortedSegmentSum.Inputs.class +) @Operator( group = "math" ) @@ -69,8 +75,8 @@ public final class UnsortedSegmentSum extends RawOp implements private Output output; - private UnsortedSegmentSum(Operation operation) { - super(operation); + public UnsortedSegmentSum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -113,6 +119,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UnsortedSegmentSum.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xdivy.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xdivy.java index c0cea7d21f7..8640e808f4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xdivy.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xdivy.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Xdivy.OP_NAME, + inputsClass = Xdivy.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Xdivy extends RawOp implements Operand { private Output z; - private Xdivy(Operation operation) { - super(operation); + public Xdivy(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Xdivy.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlog1py.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlog1py.java index 75936967d7d..d69db380541 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlog1py.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlog1py.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Xlog1py.OP_NAME, + inputsClass = Xlog1py.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Xlog1py extends RawOp implements Operand private Output z; - private Xlog1py(Operation operation) { - super(operation); + public Xlog1py(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Xlog1py.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlogy.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlogy.java index da05a27ba3a..1eae3d1e39d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlogy.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Xlogy.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Xlogy.OP_NAME, + inputsClass = Xlogy.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class Xlogy extends RawOp implements Operand { private Output z; - private Xlogy(Operation operation) { - super(operation); + public Xlogy(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Xlogy.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Zeta.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Zeta.java index 8a7fb4ebcbc..8c04a80e767 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Zeta.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/Zeta.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = Zeta.OP_NAME, + inputsClass = Zeta.Inputs.class +) @Operator( group = "math" ) @@ -49,8 +55,8 @@ public final class Zeta extends RawOp implements Operand { private Output z; - private Zeta(Operation operation) { - super(operation); + public Zeta(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = Zeta.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/erfinv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/erfinv.java index 81c9afb6a49..a5cbcdecd88 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/erfinv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/erfinv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = erfinv.OP_NAME, + inputsClass = erfinv.Inputs.class +) @Operator( group = "math" ) @@ -47,8 +53,8 @@ public final class erfinv extends RawOp implements Operand private Output y; - private erfinv(Operation operation) { - super(operation); + public erfinv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = erfinv.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java index a9b0a239f7b..8c600b907c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselJ0.OP_NAME, + inputsClass = BesselJ0.Inputs.class +) public final class BesselJ0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselJ0 extends RawOp implements Operand< private Output y; - private BesselJ0(Operation operation) { - super(operation); + public BesselJ0(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselJ0.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java index 2f4b8d6b71e..52a0694d668 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselJ1.OP_NAME, + inputsClass = BesselJ1.Inputs.class +) public final class BesselJ1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselJ1 extends RawOp implements Operand< private Output y; - private BesselJ1(Operation operation) { - super(operation); + public BesselJ1(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselJ1.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java index 13bf915e021..60a575f23f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselK0.OP_NAME, + inputsClass = BesselK0.Inputs.class +) public final class BesselK0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselK0 extends RawOp implements Operand< private Output y; - private BesselK0(Operation operation) { - super(operation); + public BesselK0(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselK0.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java index 0f611baf8df..0cb07b3796c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselK0e.OP_NAME, + inputsClass = BesselK0e.Inputs.class +) public final class BesselK0e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselK0e extends RawOp implements Operand private Output y; - private BesselK0e(Operation operation) { - super(operation); + public BesselK0e(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselK0e.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java index 5696e8ca364..10ca43255fb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselK1.OP_NAME, + inputsClass = BesselK1.Inputs.class +) public final class BesselK1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselK1 extends RawOp implements Operand< private Output y; - private BesselK1(Operation operation) { - super(operation); + public BesselK1(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselK1.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java index ce0de01e5fc..f0b189f96a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselK1e.OP_NAME, + inputsClass = BesselK1e.Inputs.class +) public final class BesselK1e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselK1e extends RawOp implements Operand private Output y; - private BesselK1e(Operation operation) { - super(operation); + public BesselK1e(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselK1e.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java index 79bdc102e59..bc9f338820b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselY0.OP_NAME, + inputsClass = BesselY0.Inputs.class +) public final class BesselY0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselY0 extends RawOp implements Operand< private Output y; - private BesselY0(Operation operation) { - super(operation); + public BesselY0(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselY0.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java index 25c1343ccf2..0e0c7550d90 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = BesselY1.OP_NAME, + inputsClass = BesselY1.Inputs.class +) public final class BesselY1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class BesselY1 extends RawOp implements Operand< private Output y; - private BesselY1(Operation operation) { - super(operation); + public BesselY1(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = BesselY1.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java index f801b241a46..c7a1c1c80bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Dawsn.OP_NAME, + inputsClass = Dawsn.Inputs.class +) public final class Dawsn extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class Dawsn extends RawOp implements Operand private Output y; - private Dawsn(Operation operation) { - super(operation); + public Dawsn(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Dawsn.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java index ed852574a64..01ae1c2b6b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Expint.OP_NAME, + inputsClass = Expint.Inputs.class +) public final class Expint extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class Expint extends RawOp implements Operand private Output y; - private Expint(Operation operation) { - super(operation); + public Expint(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Expint.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java index b7433ee47b5..b43796ba4c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = FresnelCos.OP_NAME, + inputsClass = FresnelCos.Inputs.class +) public final class FresnelCos extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class FresnelCos extends RawOp implements Operan private Output y; - private FresnelCos(Operation operation) { - super(operation); + public FresnelCos(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = FresnelCos.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java index 2c3ac5e9469..ccf8573fb13 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = FresnelSin.OP_NAME, + inputsClass = FresnelSin.Inputs.class +) public final class FresnelSin extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class FresnelSin extends RawOp implements Operan private Output y; - private FresnelSin(Operation operation) { - super(operation); + public FresnelSin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = FresnelSin.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java index 64128da2664..48ded64c429 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = Spence.OP_NAME, + inputsClass = Spence.Inputs.class +) public final class Spence extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class Spence extends RawOp implements Operand private Output y; - private Spence(Operation operation) { - super(operation); + public Spence(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = Spence.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool.java index 4496333ee52..464436c169f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AvgPool.OP_NAME, + inputsClass = AvgPool.Inputs.class +) @Operator( group = "nn" ) @@ -50,8 +56,8 @@ public final class AvgPool extends RawOp implements Operand output; - private AvgPool(Operation operation) { - super(operation); + public AvgPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -149,6 +155,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = AvgPool.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3d.java index 6a42d550dfa..4add48fb958 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AvgPool3d.OP_NAME, + inputsClass = AvgPool3d.Inputs.class +) @Operator( group = "nn" ) @@ -50,8 +56,8 @@ public final class AvgPool3d extends RawOp implements Operand private Output output; - private AvgPool3d(Operation operation) { - super(operation); + public AvgPool3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = AvgPool3d.class + ) public static class Inputs extends RawOpInputs> { /** * Shape {@code [batch, depth, rows, cols, channels]} tensor to pool over. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3dGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3dGrad.java index 359c151cf8a..c5f30586478 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3dGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPool3dGrad.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AvgPool3dGrad.OP_NAME, + inputsClass = AvgPool3dGrad.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class AvgPool3dGrad extends RawOp implements Ope private Output output; - private AvgPool3dGrad(Operation operation) { - super(operation); + public AvgPool3dGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -153,6 +159,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = AvgPool3dGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The original input dimensions. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java index 980664e35e1..ed699ac9669 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AvgPoolGrad.OP_NAME, + inputsClass = AvgPoolGrad.Inputs.class +) public final class AvgPoolGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class AvgPoolGrad extends RawOp implements Opera private Output output; - private AvgPoolGrad(Operation operation) { - super(operation); + public AvgPoolGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = AvgPoolGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. Shape of the original input to {@code avg_pool}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalization.java index bf1a0975705..3188b4698c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalization.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code result} output */ +@OpMetadata( + opType = BatchNormWithGlobalNormalization.OP_NAME, + inputsClass = BatchNormWithGlobalNormalization.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class BatchNormWithGlobalNormalization extends Raw private Output result; - private BatchNormWithGlobalNormalization(Operation operation) { - super(operation); + public BatchNormWithGlobalNormalization(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; result = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return result; } + @OpInputsMetadata( + outputsClass = BatchNormWithGlobalNormalization.class + ) public static class Inputs extends RawOpInputs> { /** * A 4D input Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalizationGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalizationGrad.java index 86a0ac56fa1..b14ed684364 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalizationGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BatchNormWithGlobalNormalizationGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code dx} output */ +@OpMetadata( + opType = BatchNormWithGlobalNormalizationGrad.OP_NAME, + inputsClass = BatchNormWithGlobalNormalizationGrad.Inputs.class +) @Operator( group = "nn" ) @@ -56,8 +62,8 @@ public final class BatchNormWithGlobalNormalizationGrad extends private Output dg; - private BatchNormWithGlobalNormalizationGrad(Operation operation) { - super(operation); + public BatchNormWithGlobalNormalizationGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; dx = operation.output(outputIdx++); dm = operation.output(outputIdx++); @@ -149,6 +155,9 @@ public Output dg() { return dg; } + @OpInputsMetadata( + outputsClass = BatchNormWithGlobalNormalizationGrad.class + ) public static class Inputs extends RawOpInputs> { /** * A 4D input Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAdd.java index 0e0917f0422..7f7617a1114 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BiasAdd.OP_NAME, + inputsClass = BiasAdd.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class BiasAdd extends RawOp implements Operand private Output output; - private BiasAdd(Operation operation) { - super(operation); + public BiasAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = BiasAdd.class + ) public static class Inputs extends RawOpInputs> { /** * Any number of dimensions. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAddGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAddGrad.java index b9eda9a3c68..100f71935c0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAddGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BiasAddGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BiasAddGrad.OP_NAME, + inputsClass = BiasAddGrad.Inputs.class +) @Operator( group = "nn" ) @@ -50,8 +56,8 @@ public final class BiasAddGrad extends RawOp implements Operand private Output output; - private BiasAddGrad(Operation operation) { - super(operation); + public BiasAddGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -139,6 +145,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = BiasAddGrad.class + ) public static class Inputs extends RawOpInputs> { /** * Any number of dimensions. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java index 187ad0505ed..bb71ef9c7b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -56,6 +58,10 @@ * * @param data type for {@code i} output */ +@OpMetadata( + opType = BlockLSTM.OP_NAME, + inputsClass = BlockLSTM.Inputs.class +) public final class BlockLSTM extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -76,8 +82,8 @@ public final class BlockLSTM extends RawOp { private Output h; - private BlockLSTM(Operation operation) { - super(operation); + public BlockLSTM(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; i = operation.output(outputIdx++); cs = operation.output(outputIdx++); @@ -252,6 +258,9 @@ public Options usePeephole(Boolean usePeephole) { } } + @OpInputsMetadata( + outputsClass = BlockLSTM.class + ) public static class Inputs extends RawOpInputs> { /** * Maximum time length actually used by this input. Outputs are padded diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java index f794390c585..5cecd1948c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code x_grad} output */ +@OpMetadata( + opType = BlockLSTMGrad.OP_NAME, + inputsClass = BlockLSTMGrad.Inputs.class +) public final class BlockLSTMGrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -59,8 +65,8 @@ public final class BlockLSTMGrad extends RawOp { private Output bGrad; - private BlockLSTMGrad(Operation operation) { - super(operation); + public BlockLSTMGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; xGrad = operation.output(outputIdx++); csPrevGrad = operation.output(outputIdx++); @@ -202,6 +208,9 @@ public Output bGrad() { return bGrad; } + @OpInputsMetadata( + outputsClass = BlockLSTMGrad.class + ) public static class Inputs extends RawOpInputs> { /** * Maximum time length actually used by this input. Outputs are padded diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java index cab3b859657..e19e16773df 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -36,6 +38,10 @@ * the gradient. This class performs the softmax operation for you, so inputs * should be e.g. linear projections of outputs by an LSTM. */ +@OpMetadata( + opType = CTCLossV2.OP_NAME, + inputsClass = CTCLossV2.Inputs.class +) public final class CTCLossV2 extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class CTCLossV2 extends RawOp { private Output gradient; - private CTCLossV2(Operation operation) { - super(operation); + public CTCLossV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; loss = operation.output(outputIdx++); gradient = operation.output(outputIdx++); @@ -200,6 +206,9 @@ public Options ignoreLongerOutputsThanInputs(Boolean ignoreLongerOutputsThanInpu } } + @OpInputsMetadata( + outputsClass = CTCLossV2.class + ) public static class Inputs extends RawOpInputs { /** * 3-D, shape: {@code (max_time x batch_size x num_classes)}, the logits. Default blank diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ComputeAccidentalHits.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ComputeAccidentalHits.java index 55e64bbc19f..080a5b71d85 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ComputeAccidentalHits.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ComputeAccidentalHits.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * the effect of 'removing' the sampled labels that match the true labels by * making the classifier sure that they are sampled labels. */ +@OpMetadata( + opType = ComputeAccidentalHits.OP_NAME, + inputsClass = ComputeAccidentalHits.Inputs.class +) @Operator( group = "nn" ) @@ -54,8 +60,8 @@ public final class ComputeAccidentalHits extends RawOp { private Output weights; - private ComputeAccidentalHits(Operation operation) { - super(operation); + public ComputeAccidentalHits(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; indices = operation.output(outputIdx++); ids = operation.output(outputIdx++); @@ -181,6 +187,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = ComputeAccidentalHits.class + ) public static class Inputs extends RawOpInputs { /** * The true_classes output of UnpackSparseLabels. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2d.java index a769c324177..8649cc0b54c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +59,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv2d.OP_NAME, + inputsClass = Conv2d.Inputs.class +) @Operator( group = "nn" ) @@ -68,8 +74,8 @@ public final class Conv2d extends RawOp implements Operand private Output output; - private Conv2d(Operation operation) { - super(operation); + public Conv2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -324,6 +330,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = Conv2d.class + ) public static class Inputs extends RawOpInputs> { /** * A 4-D tensor. The dimension order is interpreted according to the value diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropFilter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropFilter.java index b5718f2ca8a..6fb5caec5ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropFilter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropFilter.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv2dBackpropFilter.OP_NAME, + inputsClass = Conv2dBackpropFilter.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class Conv2dBackpropFilter extends RawOp impleme private Output output; - private Conv2dBackpropFilter(Operation operation) { - super(operation); + public Conv2dBackpropFilter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -310,6 +316,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = Conv2dBackpropFilter.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, in_height, in_width, in_channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropInput.java index f629ba8e03a..400d03fa9f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv2dBackpropInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv2dBackpropInput.OP_NAME, + inputsClass = Conv2dBackpropInput.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class Conv2dBackpropInput extends RawOp implemen private Output output; - private Conv2dBackpropInput(Operation operation) { - super(operation); + public Conv2dBackpropInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -309,6 +315,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = Conv2dBackpropInput.class + ) public static class Inputs extends RawOpInputs> { /** * An integer vector representing the shape of {@code input}, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3d.java index aeefc06db8e..e9c3ffb04af 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv3d.OP_NAME, + inputsClass = Conv3d.Inputs.class +) @Operator( group = "nn" ) @@ -52,8 +58,8 @@ public final class Conv3d extends RawOp implements Operand private Output output; - private Conv3d(Operation operation) { - super(operation); + public Conv3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -215,6 +221,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = Conv3d.class + ) public static class Inputs extends RawOpInputs> { /** * Shape {@code [batch, in_depth, in_height, in_width, in_channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropFilter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropFilter.java index a788a51fb92..b79bc9181ff 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropFilter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropFilter.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv3dBackpropFilter.OP_NAME, + inputsClass = Conv3dBackpropFilter.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class Conv3dBackpropFilter extends RawOp impleme private Output output; - private Conv3dBackpropFilter(Operation operation) { - super(operation); + public Conv3dBackpropFilter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -218,6 +224,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = Conv3dBackpropFilter.class + ) public static class Inputs extends RawOpInputs> { /** * Shape {@code [batch, depth, rows, cols, in_channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropInput.java index 0cda7f2020c..5e9324aaf0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Conv3dBackpropInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv3dBackpropInput.OP_NAME, + inputsClass = Conv3dBackpropInput.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class Conv3dBackpropInput extends RawOp implemen private Output output; - private Conv3dBackpropInput(Operation operation) { - super(operation); + public Conv3dBackpropInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -217,6 +223,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = Conv3dBackpropInput.class + ) public static class Inputs extends RawOpInputs> { /** * An integer vector representing the tensor shape of {@code input}, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcBeamSearchDecoder.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcBeamSearchDecoder.java index b9cab99e0a5..ffee6939d9f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcBeamSearchDecoder.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcBeamSearchDecoder.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -44,6 +46,10 @@ * * @param data type for {@code log_probability} output */ +@OpMetadata( + opType = CtcBeamSearchDecoder.OP_NAME, + inputsClass = CtcBeamSearchDecoder.Inputs.class +) @Operator( group = "nn" ) @@ -62,8 +68,8 @@ public final class CtcBeamSearchDecoder extends RawOp { private Output logProbability; @SuppressWarnings("unchecked") - private CtcBeamSearchDecoder(Operation operation) { - super(operation); + public CtcBeamSearchDecoder(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int decodedIndicesLength = operation.outputListLength("decoded_indices"); decodedIndices = Arrays.asList((Output[]) operation.outputList(outputIdx, decodedIndicesLength)); @@ -183,6 +189,9 @@ public Options mergeRepeated(Boolean mergeRepeated) { } } + @OpInputsMetadata( + outputsClass = CtcBeamSearchDecoder.class + ) public static class Inputs extends RawOpInputs> { /** * 3-D, shape: {@code (max_time x batch_size x num_classes)}, the logits. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcGreedyDecoder.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcGreedyDecoder.java index 3975061991c..6ed1bfb1fac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcGreedyDecoder.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcGreedyDecoder.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * * @param data type for {@code log_probability} output */ +@OpMetadata( + opType = CtcGreedyDecoder.OP_NAME, + inputsClass = CtcGreedyDecoder.Inputs.class +) @Operator( group = "nn" ) @@ -63,8 +69,8 @@ public final class CtcGreedyDecoder extends RawOp { private Output logProbability; - private CtcGreedyDecoder(Operation operation) { - super(operation); + public CtcGreedyDecoder(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; decodedIndices = operation.output(outputIdx++); decodedValues = operation.output(outputIdx++); @@ -197,6 +203,9 @@ public Options blankIndex(Long blankIndex) { } } + @OpInputsMetadata( + outputsClass = CtcGreedyDecoder.class + ) public static class Inputs extends RawOpInputs> { /** * 3-D, shape: {@code (max_time x batch_size x num_classes)}, the logits. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcLoss.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcLoss.java index f70a3763d9e..ca47ae6d968 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcLoss.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CtcLoss.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code loss} output */ +@OpMetadata( + opType = CtcLoss.OP_NAME, + inputsClass = CtcLoss.Inputs.class +) @Operator( group = "nn" ) @@ -53,8 +59,8 @@ public final class CtcLoss extends RawOp { private Output gradient; - private CtcLoss(Operation operation) { - super(operation); + public CtcLoss(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; loss = operation.output(outputIdx++); gradient = operation.output(outputIdx++); @@ -207,6 +213,9 @@ public Options ignoreLongerOutputsThanInputs(Boolean ignoreLongerOutputsThanInpu } } + @OpInputsMetadata( + outputsClass = CtcLoss.class + ) public static class Inputs extends RawOpInputs> { /** * 3-D, shape: {@code (max_time x batch_size x num_classes)}, the logits. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java index 68e42dfd916..22a652fed1c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -73,6 +75,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CudnnRNN.OP_NAME, + inputsClass = CudnnRNN.Inputs.class +) public final class CudnnRNN extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -90,8 +96,8 @@ public final class CudnnRNN extends RawOp { private Output hostReserved; @SuppressWarnings("unchecked") - private CudnnRNN(Operation operation) { - super(operation); + public CudnnRNN(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputH = operation.output(outputIdx++); @@ -419,6 +425,9 @@ public Options timeMajor(Boolean timeMajor) { } } + @OpInputsMetadata( + outputsClass = CudnnRNN.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java index 13372cc1bbc..fb56c1f0f07 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -83,6 +85,10 @@ * * @param data type for {@code input_backprop} output */ +@OpMetadata( + opType = CudnnRNNBackprop.OP_NAME, + inputsClass = CudnnRNNBackprop.Inputs.class +) public final class CudnnRNNBackprop extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -97,8 +103,8 @@ public final class CudnnRNNBackprop extends RawOp { private Output paramsBackprop; - private CudnnRNNBackprop(Operation operation) { - super(operation); + public CudnnRNNBackprop(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; inputBackprop = operation.output(outputIdx++); inputHBackprop = operation.output(outputIdx++); @@ -408,6 +414,9 @@ public Options timeMajor(Boolean timeMajor) { } } + @OpInputsMetadata( + outputsClass = CudnnRNNBackprop.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNCanonicalToParams.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNCanonicalToParams.java index eeda8931a7b..f720ab18d24 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNCanonicalToParams.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNCanonicalToParams.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -66,6 +68,10 @@ * * @param data type for {@code params} output */ +@OpMetadata( + opType = CudnnRNNCanonicalToParams.OP_NAME, + inputsClass = CudnnRNNCanonicalToParams.Inputs.class +) @Operator( group = "nn" ) @@ -77,8 +83,8 @@ public final class CudnnRNNCanonicalToParams extends RawOp im private Output params; - private CudnnRNNCanonicalToParams(Operation operation) { - super(operation); + public CudnnRNNCanonicalToParams(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; params = operation.output(outputIdx++); } @@ -319,6 +325,9 @@ public Options numProj(Long numProj) { } } + @OpInputsMetadata( + outputsClass = CudnnRNNCanonicalToParams.class + ) public static class Inputs extends RawOpInputs> { /** * The numLayers input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNParamsToCanonical.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNParamsToCanonical.java index 4cfdf128902..f3c981ae1ca 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNParamsToCanonical.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNParamsToCanonical.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -66,6 +68,10 @@ * * @param data type for {@code weights} output */ +@OpMetadata( + opType = CudnnRNNParamsToCanonical.OP_NAME, + inputsClass = CudnnRNNParamsToCanonical.Inputs.class +) @Operator( group = "nn" ) @@ -80,8 +86,8 @@ public final class CudnnRNNParamsToCanonical extends RawOp { private List> biases; @SuppressWarnings("unchecked") - private CudnnRNNParamsToCanonical(Operation operation) { - super(operation); + public CudnnRNNParamsToCanonical(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int weightsLength = operation.outputListLength("weights"); weights = Arrays.asList((Output[]) operation.outputList(outputIdx, weightsLength)); @@ -333,6 +339,9 @@ public Options numProj(Long numProj) { } } + @OpInputsMetadata( + outputsClass = CudnnRNNParamsToCanonical.class + ) public static class Inputs extends RawOpInputs> { /** * The numLayers input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRnnParamsSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRnnParamsSize.java index 2dd40e8cfad..76e7384ef36 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRnnParamsSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRnnParamsSize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -58,6 +60,10 @@ * * @param data type for {@code params_size} output */ +@OpMetadata( + opType = CudnnRnnParamsSize.OP_NAME, + inputsClass = CudnnRnnParamsSize.Inputs.class +) @Operator( group = "nn" ) @@ -69,8 +75,8 @@ public final class CudnnRnnParamsSize extends RawOp implement private Output paramsSize; - private CudnnRnnParamsSize(Operation operation) { - super(operation); + public CudnnRnnParamsSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; paramsSize = operation.output(outputIdx++); } @@ -312,6 +318,9 @@ public Options numProj(Long numProj) { } } + @OpInputsMetadata( + outputsClass = CudnnRnnParamsSize.class + ) public static class Inputs extends RawOpInputs> { /** * The numLayers input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatDimMap.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatDimMap.java index 95cf7c41dd0..cb39229fe25 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatDimMap.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatDimMap.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = DataFormatDimMap.OP_NAME, + inputsClass = DataFormatDimMap.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class DataFormatDimMap extends RawOp implements private Output y; - private DataFormatDimMap(Operation operation) { - super(operation); + public DataFormatDimMap(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -152,6 +158,9 @@ public Options dstFormat(String dstFormat) { } } + @OpInputsMetadata( + outputsClass = DataFormatDimMap.class + ) public static class Inputs extends RawOpInputs> { /** * A Tensor with each element as a dimension index in source data format. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatVecPermute.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatVecPermute.java index df1a3825d4b..208a59c355d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatVecPermute.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DataFormatVecPermute.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -55,6 +57,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = DataFormatVecPermute.OP_NAME, + inputsClass = DataFormatVecPermute.Inputs.class +) @Operator( group = "nn" ) @@ -66,8 +72,8 @@ public final class DataFormatVecPermute extends RawOp impleme private Output y; - private DataFormatVecPermute(Operation operation) { - super(operation); + public DataFormatVecPermute(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -169,6 +175,9 @@ public Options dstFormat(String dstFormat) { } } + @OpInputsMetadata( + outputsClass = DataFormatVecPermute.class + ) public static class Inputs extends RawOpInputs> { /** * Vector of size 4 or Tensor of shape (4, 2) in source data format. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthToSpace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthToSpace.java index 2d2d46b500b..73440f2a0f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthToSpace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthToSpace.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -110,6 +112,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DepthToSpace.OP_NAME, + inputsClass = DepthToSpace.Inputs.class +) @Operator( group = "nn" ) @@ -121,8 +127,8 @@ public final class DepthToSpace extends RawOp implements Operan private Output output; - private DepthToSpace(Operation operation) { - super(operation); + public DepthToSpace(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -200,6 +206,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = DepthToSpace.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNative.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNative.java index 929cd0737ae..2abafeda8d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNative.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNative.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DepthwiseConv2dNative.OP_NAME, + inputsClass = DepthwiseConv2dNative.Inputs.class +) @Operator( group = "nn" ) @@ -64,8 +70,8 @@ public final class DepthwiseConv2dNative extends RawOp implem private Output output; - private DepthwiseConv2dNative(Operation operation) { - super(operation); + public DepthwiseConv2dNative(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -278,6 +284,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = DepthwiseConv2dNative.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropFilter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropFilter.java index 9e80ec91386..2389f0dc99a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropFilter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropFilter.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DepthwiseConv2dNativeBackpropFilter.OP_NAME, + inputsClass = DepthwiseConv2dNativeBackpropFilter.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class DepthwiseConv2dNativeBackpropFilter extend private Output output; - private DepthwiseConv2dNativeBackpropFilter(Operation operation) { - super(operation); + public DepthwiseConv2dNativeBackpropFilter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -274,6 +280,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = DepthwiseConv2dNativeBackpropFilter.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape based on {@code data_format}. For example, if diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropInput.java index 57baa455eaf..cf8f88d0be2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/DepthwiseConv2dNativeBackpropInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DepthwiseConv2dNativeBackpropInput.OP_NAME, + inputsClass = DepthwiseConv2dNativeBackpropInput.Inputs.class +) @Operator( group = "nn" ) @@ -49,8 +55,8 @@ public final class DepthwiseConv2dNativeBackpropInput extends private Output output; - private DepthwiseConv2dNativeBackpropInput(Operation operation) { - super(operation); + public DepthwiseConv2dNativeBackpropInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -274,6 +280,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = DepthwiseConv2dNativeBackpropInput.class + ) public static class Inputs extends RawOpInputs> { /** * An integer vector representing the shape of {@code input}, based diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2d.java index 2eced92585f..e17b172eb20 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -58,6 +60,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Dilation2d.OP_NAME, + inputsClass = Dilation2d.Inputs.class +) @Operator( group = "nn" ) @@ -69,8 +75,8 @@ public final class Dilation2d extends RawOp implements Operan private Output output; - private Dilation2d(Operation operation) { - super(operation); + public Dilation2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -125,6 +131,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Dilation2d.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, in_height, in_width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropFilter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropFilter.java index baa0e803e8a..8c8c3baebc0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropFilter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropFilter.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code filter_backprop} output */ +@OpMetadata( + opType = Dilation2dBackpropFilter.OP_NAME, + inputsClass = Dilation2dBackpropFilter.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class Dilation2dBackpropFilter extends RawOp imp private Output filterBackprop; - private Dilation2dBackpropFilter(Operation operation) { - super(operation); + public Dilation2dBackpropFilter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; filterBackprop = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return filterBackprop; } + @OpInputsMetadata( + outputsClass = Dilation2dBackpropFilter.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, in_height, in_width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropInput.java index 2a11f34c905..ef317847166 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Dilation2dBackpropInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code in_backprop} output */ +@OpMetadata( + opType = Dilation2dBackpropInput.OP_NAME, + inputsClass = Dilation2dBackpropInput.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class Dilation2dBackpropInput extends RawOp impl private Output inBackprop; - private Dilation2dBackpropInput(Operation operation) { - super(operation); + public Dilation2dBackpropInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; inBackprop = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return inBackprop; } + @OpInputsMetadata( + outputsClass = Dilation2dBackpropInput.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, in_height, in_width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Elu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Elu.java index 0060e99d5f3..4995249e8d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Elu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Elu.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -56,6 +58,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = Elu.OP_NAME, + inputsClass = Elu.Inputs.class +) @Operator( group = "nn" ) @@ -67,8 +73,8 @@ public final class Elu extends RawOp implements Operand { private Output activations; - private Elu(Operation operation) { - super(operation); + public Elu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -104,6 +110,9 @@ public Output asOutput() { return activations; } + @OpInputsMetadata( + outputsClass = Elu.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java index 7d3af9d8535..b62f09ade82 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = EluGrad.OP_NAME, + inputsClass = EluGrad.Inputs.class +) public final class EluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class EluGrad extends RawOp implements Operand backprops; - private EluGrad(Operation operation) { - super(operation); + public EluGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return backprops; } + @OpInputsMetadata( + outputsClass = EluGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding Elu operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FixedUnigramCandidateSampler.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FixedUnigramCandidateSampler.java index d488683a39b..f3f1204f574 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FixedUnigramCandidateSampler.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FixedUnigramCandidateSampler.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -46,6 +48,10 @@ * the sampled candidates must be chosen independently of the context and of the * true labels. */ +@OpMetadata( + opType = FixedUnigramCandidateSampler.OP_NAME, + inputsClass = FixedUnigramCandidateSampler.Inputs.class +) @Operator( group = "nn" ) @@ -61,8 +67,8 @@ public final class FixedUnigramCandidateSampler extends RawOp { private Output sampledExpectedCount; - private FixedUnigramCandidateSampler(Operation operation) { - super(operation); + public FixedUnigramCandidateSampler(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sampledCandidates = operation.output(outputIdx++); trueExpectedCount = operation.output(outputIdx++); @@ -416,6 +422,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = FixedUnigramCandidateSampler.class + ) public static class Inputs extends RawOpInputs { /** * A batch_size * num_true matrix, in which each row contains the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPool.java index ce678467253..362e0e58796 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPool.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = FractionalAvgPool.OP_NAME, + inputsClass = FractionalAvgPool.Inputs.class +) @Operator( group = "nn" ) @@ -57,8 +63,8 @@ public final class FractionalAvgPool extends RawOp { private Output colPoolingSequence; - private FractionalAvgPool(Operation operation) { - super(operation); + public FractionalAvgPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); rowPoolingSequence = operation.output(outputIdx++); @@ -288,6 +294,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = FractionalAvgPool.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java index 0cfd7f8e7d2..26f45e63842 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = FractionalAvgPoolGrad.OP_NAME, + inputsClass = FractionalAvgPoolGrad.Inputs.class +) public final class FractionalAvgPoolGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class FractionalAvgPoolGrad extends RawOp implem private Output output; - private FractionalAvgPoolGrad(Operation operation) { - super(operation); + public FractionalAvgPoolGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options overlapping(Boolean overlapping) { } } + @OpInputsMetadata( + outputsClass = FractionalAvgPoolGrad.class + ) public static class Inputs extends RawOpInputs> { /** * Original input tensor shape for {@code fractional_avg_pool} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPool.java index f1733ed28cc..9764fcbc919 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPool.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -64,6 +66,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = FractionalMaxPool.OP_NAME, + inputsClass = FractionalMaxPool.Inputs.class +) @Operator( group = "nn" ) @@ -79,8 +85,8 @@ public final class FractionalMaxPool extends RawOp { private Output colPoolingSequence; - private FractionalMaxPool(Operation operation) { - super(operation); + public FractionalMaxPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); rowPoolingSequence = operation.output(outputIdx++); @@ -310,6 +316,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = FractionalMaxPool.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java index 5233a6b19f1..2b0623418d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = FractionalMaxPoolGrad.OP_NAME, + inputsClass = FractionalMaxPoolGrad.Inputs.class +) public final class FractionalMaxPoolGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class FractionalMaxPoolGrad extends RawOp implem private Output output; - private FractionalMaxPoolGrad(Operation operation) { - super(operation); + public FractionalMaxPoolGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -143,6 +149,9 @@ public Options overlapping(Boolean overlapping) { } } + @OpInputsMetadata( + outputsClass = FractionalMaxPoolGrad.class + ) public static class Inputs extends RawOpInputs> { /** * Original input for {@code fractional_max_pool} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNorm.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNorm.java index a216c600ff7..08d83f5a099 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNorm.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNorm.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code batch_mean} output */ +@OpMetadata( + opType = FusedBatchNorm.OP_NAME, + inputsClass = FusedBatchNorm.Inputs.class +) @Operator( group = "nn" ) @@ -61,8 +67,8 @@ public final class FusedBatchNorm extends private Output reserveSpace3; - private FusedBatchNorm(Operation operation) { - super(operation); + public FusedBatchNorm(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); batchMean = operation.output(outputIdx++); @@ -280,6 +286,9 @@ public Options isTraining(Boolean isTraining) { } } + @OpInputsMetadata( + outputsClass = FusedBatchNorm.class + ) public static class Inputs extends RawOpInputs> { /** * A 4D Tensor for input data. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNormGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNormGrad.java index a423903c502..f1b7dc8eb91 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNormGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedBatchNormGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -41,6 +43,10 @@ * * @param data type for {@code scale_backprop} output */ +@OpMetadata( + opType = FusedBatchNormGrad.OP_NAME, + inputsClass = FusedBatchNormGrad.Inputs.class +) @Operator( group = "nn" ) @@ -60,8 +66,8 @@ public final class FusedBatchNormGrad exte private Output reserveSpace5; - private FusedBatchNormGrad(Operation operation) { - super(operation); + public FusedBatchNormGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; xBackprop = operation.output(outputIdx++); scaleBackprop = operation.output(outputIdx++); @@ -250,6 +256,9 @@ public Options isTraining(Boolean isTraining) { } } + @OpInputsMetadata( + outputsClass = FusedBatchNormGrad.class + ) public static class Inputs extends RawOpInputs> { /** * A 4D Tensor for the gradient with respect to y. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedPadConv2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedPadConv2d.java index 3d938544bd8..0d7a0589f98 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedPadConv2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedPadConv2d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -49,6 +51,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = FusedPadConv2d.OP_NAME, + inputsClass = FusedPadConv2d.Inputs.class +) @Operator( group = "nn" ) @@ -60,8 +66,8 @@ public final class FusedPadConv2d extends RawOp implements Op private Output output; - private FusedPadConv2d(Operation operation) { - super(operation); + public FusedPadConv2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -116,6 +122,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = FusedPadConv2d.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, in_height, in_width, in_channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedResizeAndPadConv2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedResizeAndPadConv2d.java index 1d072dbf388..f0253a8ddd5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedResizeAndPadConv2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FusedResizeAndPadConv2d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = FusedResizeAndPadConv2d.OP_NAME, + inputsClass = FusedResizeAndPadConv2d.Inputs.class +) @Operator( group = "nn" ) @@ -59,8 +65,8 @@ public final class FusedResizeAndPadConv2d extends RawOp impl private Output output; - private FusedResizeAndPadConv2d(Operation operation) { - super(operation); + public FusedResizeAndPadConv2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -159,6 +165,9 @@ public Options resizeAlignCorners(Boolean resizeAlignCorners) { } } + @OpInputsMetadata( + outputsClass = FusedResizeAndPadConv2d.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, in_height, in_width, in_channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java index a78808f3977..b122981e655 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -73,6 +75,10 @@ * * @param data type for {@code r} output */ +@OpMetadata( + opType = GRUBlockCell.OP_NAME, + inputsClass = GRUBlockCell.Inputs.class +) public final class GRUBlockCell extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -87,8 +93,8 @@ public final class GRUBlockCell extends RawOp { private Output h; - private GRUBlockCell(Operation operation) { - super(operation); + public GRUBlockCell(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; r = operation.output(outputIdx++); u = operation.output(outputIdx++); @@ -160,6 +166,9 @@ public Output h() { return h; } + @OpInputsMetadata( + outputsClass = GRUBlockCell.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java index b2a324662d1..e46ece87c86 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -108,6 +110,10 @@ * * @param data type for {@code d_x} output */ +@OpMetadata( + opType = GRUBlockCellGrad.OP_NAME, + inputsClass = GRUBlockCellGrad.Inputs.class +) public final class GRUBlockCellGrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -122,8 +128,8 @@ public final class GRUBlockCellGrad extends RawOp { private Output dRBarUBar; - private GRUBlockCellGrad(Operation operation) { - super(operation); + public GRUBlockCellGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; dX = operation.output(outputIdx++); dHPrev = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Output dRBarUBar() { return dRBarUBar; } + @OpInputsMetadata( + outputsClass = GRUBlockCellGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InTopK.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InTopK.java index 43b2d73ac08..11641113983 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InTopK.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InTopK.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -47,6 +49,10 @@ * \(out_i\) be the output for example {@code i}, *

    $$out_i = predictions_{i, targets_i} \in TopKIncludingTies(predictions_i)$$ */ +@OpMetadata( + opType = InTopK.OP_NAME, + inputsClass = InTopK.Inputs.class +) @Operator( group = "nn" ) @@ -58,8 +64,8 @@ public final class InTopK extends RawOp implements Operand { private Output precision; - private InTopK(Operation operation) { - super(operation); + public InTopK(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; precision = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return precision; } + @OpInputsMetadata( + outputsClass = InTopK.class + ) public static class Inputs extends RawOpInputs { /** * A {@code batch_size} x {@code classes} tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java index bb095a412bb..5f541190f38 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = InvGrad.OP_NAME, + inputsClass = InvGrad.Inputs.class +) public final class InvGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class InvGrad extends RawOp implements Operand private Output z; - private InvGrad(Operation operation) { - super(operation); + public InvGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = InvGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The y input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java index 2da787ebe0b..bc03d392173 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = IsotonicRegression.OP_NAME, + inputsClass = IsotonicRegression.Inputs.class +) public final class IsotonicRegression extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class IsotonicRegression extends RawOp { private Output segments; - private IsotonicRegression(Operation operation) { - super(operation); + public IsotonicRegression(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); segments = operation.output(outputIdx++); @@ -107,6 +113,9 @@ public Output segments() { return segments; } + @OpInputsMetadata( + outputsClass = IsotonicRegression.class + ) public static class Inputs extends RawOpInputs> { /** * A (batch_size, dim)-tensor holding a batch of inputs. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/L2Loss.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/L2Loss.java index e4b2fcf5d47..7e59f9de758 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/L2Loss.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/L2Loss.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = L2Loss.OP_NAME, + inputsClass = L2Loss.Inputs.class +) @Operator( group = "nn" ) @@ -51,8 +57,8 @@ public final class L2Loss extends RawOp implements Operand private Output output; - private L2Loss(Operation operation) { - super(operation); + public L2Loss(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = L2Loss.class + ) public static class Inputs extends RawOpInputs> { /** * Typically 2-D, but may have any dimensions. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java index 4f3067b63e7..2c302edcfb2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +59,10 @@ * * @param data type for {@code i} output */ +@OpMetadata( + opType = LSTMBlockCell.OP_NAME, + inputsClass = LSTMBlockCell.Inputs.class +) public final class LSTMBlockCell extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -77,8 +83,8 @@ public final class LSTMBlockCell extends RawOp { private Output h; - private LSTMBlockCell(Operation operation) { - super(operation); + public LSTMBlockCell(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; i = operation.output(outputIdx++); cs = operation.output(outputIdx++); @@ -276,6 +282,9 @@ public Options usePeephole(Boolean usePeephole) { } } + @OpInputsMetadata( + outputsClass = LSTMBlockCell.class + ) public static class Inputs extends RawOpInputs> { /** * The input to the LSTM cell, shape (batch_size, num_inputs). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java index b20e2482520..b8cda9d4399 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code cs_prev_grad} output */ +@OpMetadata( + opType = LSTMBlockCellGrad.OP_NAME, + inputsClass = LSTMBlockCellGrad.Inputs.class +) public final class LSTMBlockCellGrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -52,8 +58,8 @@ public final class LSTMBlockCellGrad extends RawOp { private Output wcoGrad; - private LSTMBlockCellGrad(Operation operation) { - super(operation); + public LSTMBlockCellGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; csPrevGrad = operation.output(outputIdx++); dicfo = operation.output(outputIdx++); @@ -159,6 +165,9 @@ public Output wcoGrad() { return wcoGrad; } + @OpInputsMetadata( + outputsClass = LSTMBlockCellGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The input to the LSTM cell, shape (batch_size, num_inputs). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LeakyRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LeakyRelu.java index b02c71ea25a..67b86b50c6b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LeakyRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LeakyRelu.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = LeakyRelu.OP_NAME, + inputsClass = LeakyRelu.Inputs.class +) @Operator( group = "nn" ) @@ -47,8 +53,8 @@ public final class LeakyRelu extends RawOp implements Operand private Output activations; - private LeakyRelu(Operation operation) { - super(operation); + public LeakyRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -124,6 +130,9 @@ public Options alpha(Float alpha) { } } + @OpInputsMetadata( + outputsClass = LeakyRelu.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LearnedUnigramCandidateSampler.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LearnedUnigramCandidateSampler.java index 0e0c5633f34..009e3d7f0ac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LearnedUnigramCandidateSampler.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LearnedUnigramCandidateSampler.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * the sampled candidates must be chosen independently of the context and of the * true labels. */ +@OpMetadata( + opType = LearnedUnigramCandidateSampler.OP_NAME, + inputsClass = LearnedUnigramCandidateSampler.Inputs.class +) @Operator( group = "nn" ) @@ -56,8 +62,8 @@ public final class LearnedUnigramCandidateSampler extends RawOp { private Output sampledExpectedCount; - private LearnedUnigramCandidateSampler(Operation operation) { - super(operation); + public LearnedUnigramCandidateSampler(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sampledCandidates = operation.output(outputIdx++); trueExpectedCount = operation.output(outputIdx++); @@ -194,6 +200,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = LearnedUnigramCandidateSampler.class + ) public static class Inputs extends RawOpInputs { /** * A batch_size * num_true matrix, in which each row contains the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalization.java index 826c443cf64..04bced1cffe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalization.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -47,6 +49,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = LocalResponseNormalization.OP_NAME, + inputsClass = LocalResponseNormalization.Inputs.class +) @Operator( group = "nn" ) @@ -58,8 +64,8 @@ public final class LocalResponseNormalization extends RawOp i private Output output; - private LocalResponseNormalization(Operation operation) { - super(operation); + public LocalResponseNormalization(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -213,6 +219,9 @@ public Options beta(Float beta) { } } + @OpInputsMetadata( + outputsClass = LocalResponseNormalization.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java index d3ef3925776..1aad5086f85 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = LocalResponseNormalizationGrad.OP_NAME, + inputsClass = LocalResponseNormalizationGrad.Inputs.class +) public final class LocalResponseNormalizationGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class LocalResponseNormalizationGrad extends Raw private Output output; - private LocalResponseNormalizationGrad(Operation operation) { - super(operation); + public LocalResponseNormalizationGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -202,6 +208,9 @@ public Options beta(Float beta) { } } + @OpInputsMetadata( + outputsClass = LocalResponseNormalizationGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LogSoftmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LogSoftmax.java index 82f178f0e1a..0773cdb9edb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LogSoftmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LogSoftmax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code logsoftmax} output */ +@OpMetadata( + opType = LogSoftmax.OP_NAME, + inputsClass = LogSoftmax.Inputs.class +) @Operator( group = "nn" ) @@ -51,8 +57,8 @@ public final class LogSoftmax extends RawOp implements Operan private Output logsoftmax; - private LogSoftmax(Operation operation) { - super(operation); + public LogSoftmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; logsoftmax = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return logsoftmax; } + @OpInputsMetadata( + outputsClass = LogSoftmax.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D with shape {@code [batch_size, num_classes]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool.java index 52286432580..21b206fdbf3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPool.OP_NAME, + inputsClass = MaxPool.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPool extends RawOp implements Operand output; - private MaxPool(Operation operation) { - super(operation); + public MaxPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = MaxPool.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D input to pool over. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3d.java index 5d9bfdffa7d..b5b81c23295 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPool3d.OP_NAME, + inputsClass = MaxPool3d.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPool3d extends RawOp implements Operand private Output output; - private MaxPool3d(Operation operation) { - super(operation); + public MaxPool3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -149,6 +155,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = MaxPool3d.class + ) public static class Inputs extends RawOpInputs> { /** * Shape {@code [batch, depth, rows, cols, channels]} tensor to pool over. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGrad.java index 2bfe76491b4..63d7ff3aad9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGrad.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPool3dGrad.OP_NAME, + inputsClass = MaxPool3dGrad.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPool3dGrad extends RawOp implements Ope private Output output; - private MaxPool3dGrad(Operation operation) { - super(operation); + public MaxPool3dGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -155,6 +161,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = MaxPool3dGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGradGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGradGrad.java index 0db33576167..81e4cb166d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGradGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPool3dGradGrad.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPool3dGradGrad.OP_NAME, + inputsClass = MaxPool3dGradGrad.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPool3dGradGrad extends RawOp implements private Output output; - private MaxPool3dGradGrad(Operation operation) { - super(operation); + public MaxPool3dGradGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = MaxPool3dGradGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGrad.java index a2e15973e80..7dfe407119e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPoolGrad.OP_NAME, + inputsClass = MaxPoolGrad.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPoolGrad extends RawOp implements Opera private Output output; - private MaxPoolGrad(Operation operation) { - super(operation); + public MaxPoolGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -145,6 +151,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = MaxPoolGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGrad.java index baeb0c83a39..6f4addd3a9b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPoolGradGrad.OP_NAME, + inputsClass = MaxPoolGradGrad.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPoolGradGrad extends RawOp implements O private Output output; - private MaxPoolGradGrad(Operation operation) { - super(operation); + public MaxPoolGradGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -145,6 +151,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = MaxPoolGradGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGradWithArgmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGradWithArgmax.java index e9acc1d5cc9..4f9fe9c1bae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGradWithArgmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradGradWithArgmax.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPoolGradGradWithArgmax.OP_NAME, + inputsClass = MaxPoolGradGradWithArgmax.Inputs.class +) @Operator( group = "nn" ) @@ -48,8 +54,8 @@ public final class MaxPoolGradGradWithArgmax extends RawOp im private Output output; - private MaxPoolGradGradWithArgmax(Operation operation) { - super(operation); + public MaxPoolGradGradWithArgmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options includeBatchInIndex(Boolean includeBatchInIndex) { } } + @OpInputsMetadata( + outputsClass = MaxPoolGradGradWithArgmax.class + ) public static class Inputs extends RawOpInputs> { /** * The original input. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java index b2fd0acb7e3..d1a7131e2ea 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = MaxPoolGradWithArgmax.OP_NAME, + inputsClass = MaxPoolGradWithArgmax.Inputs.class +) public final class MaxPoolGradWithArgmax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class MaxPoolGradWithArgmax extends RawOp implem private Output output; - private MaxPoolGradWithArgmax(Operation operation) { - super(operation); + public MaxPoolGradWithArgmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -142,6 +148,9 @@ public Options includeBatchInIndex(Boolean includeBatchInIndex) { } } + @OpInputsMetadata( + outputsClass = MaxPoolGradWithArgmax.class + ) public static class Inputs extends RawOpInputs> { /** * The original input. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolWithArgmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolWithArgmax.java index 42e6dfbefb5..73b54f6f150 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolWithArgmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolWithArgmax.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code argmax} output */ +@OpMetadata( + opType = MaxPoolWithArgmax.OP_NAME, + inputsClass = MaxPoolWithArgmax.Inputs.class +) @Operator( group = "nn" ) @@ -62,8 +68,8 @@ public final class MaxPoolWithArgmax exten private Output argmax; - private MaxPoolWithArgmax(Operation operation) { - super(operation); + public MaxPoolWithArgmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); argmax = operation.output(outputIdx++); @@ -184,6 +190,9 @@ public Options includeBatchInIndex(Boolean includeBatchInIndex) { } } + @OpInputsMetadata( + outputsClass = MaxPoolWithArgmax.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. Input to pool over. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/NthElement.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/NthElement.java index 14df47bbcc2..7f4635a86ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/NthElement.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/NthElement.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -44,6 +46,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = NthElement.OP_NAME, + inputsClass = NthElement.Inputs.class +) @Operator( group = "nn" ) @@ -55,8 +61,8 @@ public final class NthElement extends RawOp implements Operan private Output values; - private NthElement(Operation operation) { - super(operation); + public NthElement(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; values = operation.output(outputIdx++); } @@ -137,6 +143,9 @@ public Options reverse(Boolean reverse) { } } + @OpInputsMetadata( + outputsClass = NthElement.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D or higher with last dimension at least {@code n+1}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedAvgPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedAvgPool.java index bceac12fb5d..76313824c90 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedAvgPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedAvgPool.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedAvgPool.OP_NAME, + inputsClass = QuantizedAvgPool.Inputs.class +) @Operator( group = "nn" ) @@ -53,8 +59,8 @@ public final class QuantizedAvgPool extends RawOp { private Output maxOutput; - private QuantizedAvgPool(Operation operation) { - super(operation); + public QuantizedAvgPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -127,6 +133,9 @@ public Output maxOutput() { return maxOutput; } + @OpInputsMetadata( + outputsClass = QuantizedAvgPool.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, channels]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBatchNormWithGlobalNormalization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBatchNormWithGlobalNormalization.java index 7ff416e12ef..7def1715bdc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBatchNormWithGlobalNormalization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBatchNormWithGlobalNormalization.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -40,6 +42,10 @@ * * @param data type for {@code result} output */ +@OpMetadata( + opType = QuantizedBatchNormWithGlobalNormalization.OP_NAME, + inputsClass = QuantizedBatchNormWithGlobalNormalization.Inputs.class +) @Operator( group = "nn" ) @@ -55,8 +61,8 @@ public final class QuantizedBatchNormWithGlobalNormalization private Output resultMax; - private QuantizedBatchNormWithGlobalNormalization(Operation operation) { - super(operation); + public QuantizedBatchNormWithGlobalNormalization(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; result = operation.output(outputIdx++); resultMin = operation.output(outputIdx++); @@ -155,6 +161,9 @@ public Output resultMax() { return resultMax; } + @OpInputsMetadata( + outputsClass = QuantizedBatchNormWithGlobalNormalization.class + ) public static class Inputs extends RawOpInputs> { /** * A 4D input Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBiasAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBiasAdd.java index a646479464f..4d44b9fadfc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBiasAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedBiasAdd.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedBiasAdd.OP_NAME, + inputsClass = QuantizedBiasAdd.Inputs.class +) @Operator( group = "nn" ) @@ -54,8 +60,8 @@ public final class QuantizedBiasAdd extends RawOp { private Output maxOut; - private QuantizedBiasAdd(Operation operation) { - super(operation); + public QuantizedBiasAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOut = operation.output(outputIdx++); @@ -121,6 +127,9 @@ public Output maxOut() { return maxOut; } + @OpInputsMetadata( + outputsClass = QuantizedBiasAdd.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java index ec7cfb17939..bccfc189f2a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DAndRelu.OP_NAME, + inputsClass = QuantizedConv2DAndRelu.Inputs.class +) public final class QuantizedConv2DAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DAndRelu extends RawOp { private Output maxOutput; - private QuantizedConv2DAndRelu(Operation operation) { - super(operation); + public QuantizedConv2DAndRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -241,6 +247,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DAndRelu.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java index c50cda07ac7..64113bb8c21 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DAndReluAndRequantize.OP_NAME, + inputsClass = QuantizedConv2DAndReluAndRequantize.Inputs.class +) public final class QuantizedConv2DAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DAndReluAndRequantize extend private Output maxOutput; - private QuantizedConv2DAndReluAndRequantize(Operation operation) { - super(operation); + public QuantizedConv2DAndReluAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -246,6 +252,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DAndReluAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java index 8fcd6d5e6e6..c24962c7ca5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DAndRequantize.OP_NAME, + inputsClass = QuantizedConv2DAndRequantize.Inputs.class +) public final class QuantizedConv2DAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DAndRequantize extends RawOp private Output maxOutput; - private QuantizedConv2DAndRequantize(Operation operation) { - super(operation); + public QuantizedConv2DAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -246,6 +252,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java index cd45b84d21f..594e907dd36 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DPerChannel.OP_NAME, + inputsClass = QuantizedConv2DPerChannel.Inputs.class +) public final class QuantizedConv2DPerChannel extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DPerChannel extends RawOp { private Output maxOutput; - private QuantizedConv2DPerChannel(Operation operation) { - super(operation); + public QuantizedConv2DPerChannel(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -190,6 +196,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DPerChannel.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java index aae3f6b2ce1..9599efe4f8e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBias.OP_NAME, + inputsClass = QuantizedConv2DWithBias.Inputs.class +) public final class QuantizedConv2DWithBias extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBias extends RawOp { private Output maxOutput; - private QuantizedConv2DWithBias(Operation operation) { - super(operation); + public QuantizedConv2DWithBias(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -243,6 +249,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBias.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java index 060fb88ec1f..63ff2e04d43 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBiasAndRelu.OP_NAME, + inputsClass = QuantizedConv2DWithBiasAndRelu.Inputs.class +) public final class QuantizedConv2DWithBiasAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBiasAndRelu extends Raw private Output maxOutput; - private QuantizedConv2DWithBiasAndRelu(Operation operation) { - super(operation); + public QuantizedConv2DWithBiasAndRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -243,6 +249,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBiasAndRelu.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java index 9709b6afc6d..e47f7adc8ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBiasAndReluAndRequantize.OP_NAME, + inputsClass = QuantizedConv2DWithBiasAndReluAndRequantize.Inputs.class +) public final class QuantizedConv2DWithBiasAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBiasAndReluAndRequantize maxOutput; - private QuantizedConv2DWithBiasAndReluAndRequantize(Operation operation) { - super(operation); + public QuantizedConv2DWithBiasAndReluAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -248,6 +254,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBiasAndReluAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java index 478048a5504..7e9062e953f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBiasAndRequantize.OP_NAME, + inputsClass = QuantizedConv2DWithBiasAndRequantize.Inputs.class +) public final class QuantizedConv2DWithBiasAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBiasAndRequantize exten private Output maxOutput; - private QuantizedConv2DWithBiasAndRequantize(Operation operation) { - super(operation); + public QuantizedConv2DWithBiasAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -248,6 +254,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBiasAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java index e9e88ec8b2e..90627ac39fa 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.OP_NAME, + inputsClass = QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.Inputs.class +) public final class QuantizedConv2DWithBiasSignedSumAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBiasSignedSumAndReluAndRequantize maxOutput; - private QuantizedConv2DWithBiasSignedSumAndReluAndRequantize(Operation operation) { - super(operation); + public QuantizedConv2DWithBiasSignedSumAndReluAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -255,6 +261,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java index b3178a7b7c9..53dd4316d87 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBiasSumAndRelu.OP_NAME, + inputsClass = QuantizedConv2DWithBiasSumAndRelu.Inputs.class +) public final class QuantizedConv2DWithBiasSumAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBiasSumAndRelu extends private Output maxOutput; - private QuantizedConv2DWithBiasSumAndRelu(Operation operation) { - super(operation); + public QuantizedConv2DWithBiasSumAndRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -245,6 +251,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBiasSumAndRelu.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java index 0a173c9732d..c734b9c62a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2DWithBiasSumAndReluAndRequantize.OP_NAME, + inputsClass = QuantizedConv2DWithBiasSumAndReluAndRequantize.Inputs.class +) public final class QuantizedConv2DWithBiasSumAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedConv2DWithBiasSumAndReluAndRequantize maxOutput; - private QuantizedConv2DWithBiasSumAndReluAndRequantize(Operation operation) { - super(operation); + public QuantizedConv2DWithBiasSumAndReluAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -255,6 +261,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2DWithBiasSumAndReluAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2d.java index 6c6d3b373fd..ba70fb60b9f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2d.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConv2d.OP_NAME, + inputsClass = QuantizedConv2d.Inputs.class +) @Operator( group = "nn" ) @@ -58,8 +64,8 @@ public final class QuantizedConv2d extends RawOp { private Output maxOutput; - private QuantizedConv2d(Operation operation) { - super(operation); + public QuantizedConv2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -215,6 +221,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = QuantizedConv2d.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java index 2471c3ab08f..6cd5a628445 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedDepthwiseConv2D.OP_NAME, + inputsClass = QuantizedDepthwiseConv2D.Inputs.class +) public final class QuantizedDepthwiseConv2D extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedDepthwiseConv2D extends RawOp { private Output maxOutput; - private QuantizedDepthwiseConv2D(Operation operation) { - super(operation); + public QuantizedDepthwiseConv2D(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -190,6 +196,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = QuantizedDepthwiseConv2D.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java index 4240f955a8d..fcd9d55596f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedDepthwiseConv2DWithBias.OP_NAME, + inputsClass = QuantizedDepthwiseConv2DWithBias.Inputs.class +) public final class QuantizedDepthwiseConv2DWithBias extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedDepthwiseConv2DWithBias extends R private Output maxOutput; - private QuantizedDepthwiseConv2DWithBias(Operation operation) { - super(operation); + public QuantizedDepthwiseConv2DWithBias(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = QuantizedDepthwiseConv2DWithBias.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java index b15046296b5..fedb1118f0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedDepthwiseConv2DWithBiasAndRelu.OP_NAME, + inputsClass = QuantizedDepthwiseConv2DWithBiasAndRelu.Inputs.class +) public final class QuantizedDepthwiseConv2DWithBiasAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedDepthwiseConv2DWithBiasAndRelu ex private Output maxOutput; - private QuantizedDepthwiseConv2DWithBiasAndRelu(Operation operation) { - super(operation); + public QuantizedDepthwiseConv2DWithBiasAndRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -243,6 +249,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedDepthwiseConv2DWithBiasAndRelu.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java index 8b39158ccee..db07ec87ba5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.OP_NAME, + inputsClass = QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.Inputs.class +) public final class QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize maxOutput; - private QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize(Operation operation) { - super(operation); + public QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -248,6 +254,9 @@ public Options paddingList(Long... paddingList) { } } + @OpInputsMetadata( + outputsClass = QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The original input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedInstanceNorm.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedInstanceNorm.java index 3fcc9015f32..6d23d7f2317 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedInstanceNorm.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedInstanceNorm.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -37,6 +39,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = QuantizedInstanceNorm.OP_NAME, + inputsClass = QuantizedInstanceNorm.Inputs.class +) @Operator( group = "nn" ) @@ -52,8 +58,8 @@ public final class QuantizedInstanceNorm extends RawOp { private Output yMax; - private QuantizedInstanceNorm(Operation operation) { - super(operation); + public QuantizedInstanceNorm(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); yMin = operation.output(outputIdx++); @@ -256,6 +262,9 @@ public Options minSeparation(Float minSeparation) { } } + @OpInputsMetadata( + outputsClass = QuantizedInstanceNorm.class + ) public static class Inputs extends RawOpInputs> { /** * A 4D input Tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedMaxPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedMaxPool.java index 57d54424778..6d73566d040 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedMaxPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedMaxPool.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedMaxPool.OP_NAME, + inputsClass = QuantizedMaxPool.Inputs.class +) @Operator( group = "nn" ) @@ -53,8 +59,8 @@ public final class QuantizedMaxPool extends RawOp { private Output maxOutput; - private QuantizedMaxPool(Operation operation) { - super(operation); + public QuantizedMaxPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); minOutput = operation.output(outputIdx++); @@ -127,6 +133,9 @@ public Output maxOutput() { return maxOutput; } + @OpInputsMetadata( + outputsClass = QuantizedMaxPool.class + ) public static class Inputs extends RawOpInputs> { /** * The 4D (batch x rows x cols x depth) Tensor to MaxReduce over. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu.java index 0f84eb38f99..19f43691d40 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = QuantizedRelu.OP_NAME, + inputsClass = QuantizedRelu.Inputs.class +) @Operator( group = "nn" ) @@ -53,8 +59,8 @@ public final class QuantizedRelu extends RawOp { private Output maxActivations; - private QuantizedRelu(Operation operation) { - super(operation); + public QuantizedRelu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); minActivations = operation.output(outputIdx++); @@ -113,6 +119,9 @@ public Output maxActivations() { return maxActivations; } + @OpInputsMetadata( + outputsClass = QuantizedRelu.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu6.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu6.java index ff29633a598..1288ae41c1a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu6.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedRelu6.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = QuantizedRelu6.OP_NAME, + inputsClass = QuantizedRelu6.Inputs.class +) @Operator( group = "nn" ) @@ -53,8 +59,8 @@ public final class QuantizedRelu6 extends RawOp { private Output maxActivations; - private QuantizedRelu6(Operation operation) { - super(operation); + public QuantizedRelu6(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); minActivations = operation.output(outputIdx++); @@ -113,6 +119,9 @@ public Output maxActivations() { return maxActivations; } + @OpInputsMetadata( + outputsClass = QuantizedRelu6.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedReluX.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedReluX.java index 0e479563b8b..c6fe479482f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedReluX.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedReluX.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -38,6 +40,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = QuantizedReluX.OP_NAME, + inputsClass = QuantizedReluX.Inputs.class +) @Operator( group = "nn" ) @@ -53,8 +59,8 @@ public final class QuantizedReluX extends RawOp { private Output maxActivations; - private QuantizedReluX(Operation operation) { - super(operation); + public QuantizedReluX(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); minActivations = operation.output(outputIdx++); @@ -115,6 +121,9 @@ public Output maxActivations() { return maxActivations; } + @OpInputsMetadata( + outputsClass = QuantizedReluX.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu.java index cb2a3ac068e..a8b4e9217c3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -46,6 +48,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = Relu.OP_NAME, + inputsClass = Relu.Inputs.class +) @Operator( group = "nn" ) @@ -57,8 +63,8 @@ public final class Relu extends RawOp implements Operand { private Output activations; - private Relu(Operation operation) { - super(operation); + public Relu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return activations; } + @OpInputsMetadata( + outputsClass = Relu.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6.java index 6a076656bd5..9df15d9675c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = Relu6.OP_NAME, + inputsClass = Relu6.Inputs.class +) @Operator( group = "nn" ) @@ -47,8 +53,8 @@ public final class Relu6 extends RawOp implements Operand private Output activations; - private Relu6(Operation operation) { - super(operation); + public Relu6(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return activations; } + @OpInputsMetadata( + outputsClass = Relu6.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java index a661ce63ea0..e6b607bb202 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = Relu6Grad.OP_NAME, + inputsClass = Relu6Grad.Inputs.class +) public final class Relu6Grad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class Relu6Grad extends RawOp implements Operand private Output backprops; - private Relu6Grad(Operation operation) { - super(operation); + public Relu6Grad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return backprops; } + @OpInputsMetadata( + outputsClass = Relu6Grad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding Relu6 operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java index 88ac8d1f067..3c7550021fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = ReluGrad.OP_NAME, + inputsClass = ReluGrad.Inputs.class +) public final class ReluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class ReluGrad extends RawOp implements Operand< private Output backprops; - private ReluGrad(Operation operation) { - super(operation); + public ReluGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return backprops; } + @OpInputsMetadata( + outputsClass = ReluGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding Relu operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Selu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Selu.java index f4bde617501..3262f1086e8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Selu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Selu.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = Selu.OP_NAME, + inputsClass = Selu.Inputs.class +) @Operator( group = "nn" ) @@ -52,8 +58,8 @@ public final class Selu extends RawOp implements Operand { private Output activations; - private Selu(Operation operation) { - super(operation); + public Selu(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return activations; } + @OpInputsMetadata( + outputsClass = Selu.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java index 4beb213c767..b7766d8292a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = SeluGrad.OP_NAME, + inputsClass = SeluGrad.Inputs.class +) public final class SeluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class SeluGrad extends RawOp implements Operand< private Output backprops; - private SeluGrad(Operation operation) { - super(operation); + public SeluGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return backprops; } + @OpInputsMetadata( + outputsClass = SeluGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding Selu operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softmax.java index 5bd51a5ac65..de8d65a202f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softmax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code softmax} output */ +@OpMetadata( + opType = Softmax.OP_NAME, + inputsClass = Softmax.Inputs.class +) @Operator( group = "nn" ) @@ -51,8 +57,8 @@ public final class Softmax extends RawOp implements Operand softmax; - private Softmax(Operation operation) { - super(operation); + public Softmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; softmax = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return softmax; } + @OpInputsMetadata( + outputsClass = Softmax.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D with shape {@code [batch_size, num_classes]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftmaxCrossEntropyWithLogits.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftmaxCrossEntropyWithLogits.java index 3d3cb903166..4c57032e423 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftmaxCrossEntropyWithLogits.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftmaxCrossEntropyWithLogits.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code loss} output */ +@OpMetadata( + opType = SoftmaxCrossEntropyWithLogits.OP_NAME, + inputsClass = SoftmaxCrossEntropyWithLogits.Inputs.class +) @Operator( group = "nn" ) @@ -50,8 +56,8 @@ public final class SoftmaxCrossEntropyWithLogits extends RawO private Output backprop; - private SoftmaxCrossEntropyWithLogits(Operation operation) { - super(operation); + public SoftmaxCrossEntropyWithLogits(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; loss = operation.output(outputIdx++); backprop = operation.output(outputIdx++); @@ -97,6 +103,9 @@ public Output backprop() { return backprop; } + @OpInputsMetadata( + outputsClass = SoftmaxCrossEntropyWithLogits.class + ) public static class Inputs extends RawOpInputs> { /** * batch_size x num_classes matrix diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softsign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softsign.java index 9d17f898281..913c13b1975 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softsign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Softsign.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code activations} output */ +@OpMetadata( + opType = Softsign.OP_NAME, + inputsClass = Softsign.Inputs.class +) @Operator( group = "nn" ) @@ -47,8 +53,8 @@ public final class Softsign extends RawOp implements Operand< private Output activations; - private Softsign(Operation operation) { - super(operation); + public Softsign(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; activations = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return activations; } + @OpInputsMetadata( + outputsClass = Softsign.class + ) public static class Inputs extends RawOpInputs> { /** * The features input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java index ba0d7d3c380..5d57b66c50c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code backprops} output */ +@OpMetadata( + opType = SoftsignGrad.OP_NAME, + inputsClass = SoftsignGrad.Inputs.class +) public final class SoftsignGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class SoftsignGrad extends RawOp implements Oper private Output backprops; - private SoftsignGrad(Operation operation) { - super(operation); + public SoftsignGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return backprops; } + @OpInputsMetadata( + outputsClass = SoftsignGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The backpropagated gradients to the corresponding softsign operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToBatch.java index 31b8b5d34ab..9969a3a1921 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToBatch.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -101,6 +103,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SpaceToBatch.OP_NAME, + inputsClass = SpaceToBatch.Inputs.class +) @Operator( group = "nn" ) @@ -112,8 +118,8 @@ public final class SpaceToBatch extends RawOp implements Operan private Output output; - private SpaceToBatch(Operation operation) { - super(operation); + public SpaceToBatch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -163,6 +169,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SpaceToBatch.class + ) public static class Inputs extends RawOpInputs> { /** * 4-D with shape {@code [batch, height, width, depth]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToDepth.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToDepth.java index 6fe32cec477..de02df239ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToDepth.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SpaceToDepth.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -104,6 +106,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SpaceToDepth.OP_NAME, + inputsClass = SpaceToDepth.Inputs.class +) @Operator( group = "nn" ) @@ -115,8 +121,8 @@ public final class SpaceToDepth extends RawOp implements Operan private Output output; - private SpaceToDepth(Operation operation) { - super(operation); + public SpaceToDepth(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -194,6 +200,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = SpaceToDepth.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SparseSoftmaxCrossEntropyWithLogits.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SparseSoftmaxCrossEntropyWithLogits.java index 5935d7eeb8f..2e08e77aa55 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SparseSoftmaxCrossEntropyWithLogits.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SparseSoftmaxCrossEntropyWithLogits.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code loss} output */ +@OpMetadata( + opType = SparseSoftmaxCrossEntropyWithLogits.OP_NAME, + inputsClass = SparseSoftmaxCrossEntropyWithLogits.Inputs.class +) @Operator( group = "nn" ) @@ -54,8 +60,8 @@ public final class SparseSoftmaxCrossEntropyWithLogits extend private Output backprop; - private SparseSoftmaxCrossEntropyWithLogits(Operation operation) { - super(operation); + public SparseSoftmaxCrossEntropyWithLogits(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; loss = operation.output(outputIdx++); backprop = operation.output(outputIdx++); @@ -100,6 +106,9 @@ public Output backprop() { return backprop; } + @OpInputsMetadata( + outputsClass = SparseSoftmaxCrossEntropyWithLogits.class + ) public static class Inputs extends RawOpInputs> { /** * batch_size x num_classes matrix diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/TopK.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/TopK.java index d549f556f17..36ccbf81754 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/TopK.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/TopK.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = TopK.OP_NAME, + inputsClass = TopK.Inputs.class +) @Operator( group = "nn" ) @@ -59,8 +65,8 @@ public final class TopK extends RawOp { private Output indices; - private TopK(Operation operation) { - super(operation); + public TopK(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; values = operation.output(outputIdx++); indices = operation.output(outputIdx++); @@ -146,6 +152,9 @@ public Options sorted(Boolean sorted) { } } + @OpInputsMetadata( + outputsClass = TopK.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D or higher with last dimension at least {@code k}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Dequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Dequantize.java index 053689d7023..22f0ac7402c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Dequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Dequantize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -81,6 +83,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Dequantize.OP_NAME, + inputsClass = Dequantize.Inputs.class +) @Operator( group = "quantization" ) @@ -92,8 +98,8 @@ public final class Dequantize extends RawOp implements Operan private Output output; - private Dequantize(Operation operation) { - super(operation); + public Dequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -247,6 +253,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = Dequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgs.java index 932f2420685..d48ce80ac57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgs.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; @@ -53,6 +55,10 @@ * *

    Quantization is called fake since the output is still in floating point. */ +@OpMetadata( + opType = FakeQuantWithMinMaxArgs.OP_NAME, + inputsClass = FakeQuantWithMinMaxArgs.Inputs.class +) @Operator( group = "quantization" ) @@ -64,8 +70,8 @@ public final class FakeQuantWithMinMaxArgs extends RawOp implements Operand outputs; - private FakeQuantWithMinMaxArgs(Operation operation) { - super(operation); + public FakeQuantWithMinMaxArgs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputs = operation.output(outputIdx++); } @@ -218,6 +224,9 @@ public Options narrowRange(Boolean narrowRange) { } } + @OpInputsMetadata( + outputsClass = FakeQuantWithMinMaxArgs.class + ) public static class Inputs extends RawOpInputs { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgsGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgsGradient.java index a87262d29e5..4d2dcf75930 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgsGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxArgsGradient.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** * Compute gradients for a FakeQuantWithMinMaxArgs operation. */ +@OpMetadata( + opType = FakeQuantWithMinMaxArgsGradient.OP_NAME, + inputsClass = FakeQuantWithMinMaxArgsGradient.Inputs.class +) @Operator( group = "quantization" ) @@ -44,8 +50,8 @@ public final class FakeQuantWithMinMaxArgsGradient extends RawOp implements Oper private Output backprops; - private FakeQuantWithMinMaxArgsGradient(Operation operation) { - super(operation); + public FakeQuantWithMinMaxArgsGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backprops = operation.output(outputIdx++); } @@ -201,6 +207,9 @@ public Options narrowRange(Boolean narrowRange) { } } + @OpInputsMetadata( + outputsClass = FakeQuantWithMinMaxArgsGradient.class + ) public static class Inputs extends RawOpInputs { /** * Backpropagated gradients above the FakeQuantWithMinMaxArgs operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVars.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVars.java index d94a0b1442a..3e27cbdf9a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVars.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVars.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; @@ -56,6 +58,10 @@ *

    This operation has a gradient and thus allows for training {@code min} and {@code max} * values. */ +@OpMetadata( + opType = FakeQuantWithMinMaxVars.OP_NAME, + inputsClass = FakeQuantWithMinMaxVars.Inputs.class +) @Operator( group = "quantization" ) @@ -67,8 +73,8 @@ public final class FakeQuantWithMinMaxVars extends RawOp implements Operand outputs; - private FakeQuantWithMinMaxVars(Operation operation) { - super(operation); + public FakeQuantWithMinMaxVars(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputs = operation.output(outputIdx++); } @@ -173,6 +179,9 @@ public Options narrowRange(Boolean narrowRange) { } } + @OpInputsMetadata( + outputsClass = FakeQuantWithMinMaxVars.class + ) public static class Inputs extends RawOpInputs { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsGradient.java index 38090987274..aa6f265322c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsGradient.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** * Compute gradients for a FakeQuantWithMinMaxVars operation. */ +@OpMetadata( + opType = FakeQuantWithMinMaxVarsGradient.OP_NAME, + inputsClass = FakeQuantWithMinMaxVarsGradient.Inputs.class +) @Operator( group = "quantization" ) @@ -48,8 +54,8 @@ public final class FakeQuantWithMinMaxVarsGradient extends RawOp { private Output backpropWrtMax; - private FakeQuantWithMinMaxVarsGradient(Operation operation) { - super(operation); + public FakeQuantWithMinMaxVarsGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backpropsWrtInput = operation.output(outputIdx++); backpropWrtMin = operation.output(outputIdx++); @@ -175,6 +181,9 @@ public Options narrowRange(Boolean narrowRange) { } } + @OpInputsMetadata( + outputsClass = FakeQuantWithMinMaxVarsGradient.class + ) public static class Inputs extends RawOpInputs { /** * Backpropagated gradients above the FakeQuantWithMinMaxVars operation. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannel.java index 8fba3a49f46..cafc1deed7e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannel.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; @@ -57,6 +59,10 @@ *

    This operation has a gradient and thus allows for training {@code min} and {@code max} * values. */ +@OpMetadata( + opType = FakeQuantWithMinMaxVarsPerChannel.OP_NAME, + inputsClass = FakeQuantWithMinMaxVarsPerChannel.Inputs.class +) @Operator( group = "quantization" ) @@ -68,8 +74,8 @@ public final class FakeQuantWithMinMaxVarsPerChannel extends RawOp implements Op private Output outputs; - private FakeQuantWithMinMaxVarsPerChannel(Operation operation) { - super(operation); + public FakeQuantWithMinMaxVarsPerChannel(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputs = operation.output(outputIdx++); } @@ -174,6 +180,9 @@ public Options narrowRange(Boolean narrowRange) { } } + @OpInputsMetadata( + outputsClass = FakeQuantWithMinMaxVarsPerChannel.class + ) public static class Inputs extends RawOpInputs { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannelGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannelGradient.java index 5ee0cbf2450..7ac14adde54 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannelGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/FakeQuantWithMinMaxVarsPerChannelGradient.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** * Compute gradients for a FakeQuantWithMinMaxVarsPerChannel operation. */ +@OpMetadata( + opType = FakeQuantWithMinMaxVarsPerChannelGradient.OP_NAME, + inputsClass = FakeQuantWithMinMaxVarsPerChannelGradient.Inputs.class +) @Operator( group = "quantization" ) @@ -48,8 +54,8 @@ public final class FakeQuantWithMinMaxVarsPerChannelGradient extends RawOp { private Output backpropWrtMax; - private FakeQuantWithMinMaxVarsPerChannelGradient(Operation operation) { - super(operation); + public FakeQuantWithMinMaxVarsPerChannelGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; backpropsWrtInput = operation.output(outputIdx++); backpropWrtMin = operation.output(outputIdx++); @@ -179,6 +185,9 @@ public Options narrowRange(Boolean narrowRange) { } } + @OpInputsMetadata( + outputsClass = FakeQuantWithMinMaxVarsPerChannelGradient.class + ) public static class Inputs extends RawOpInputs { /** * Backpropagated gradients above the FakeQuantWithMinMaxVars operation, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Quantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Quantize.java index d3223aa4e9d..cb1262782bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Quantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Quantize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -129,6 +131,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Quantize.OP_NAME, + inputsClass = Quantize.Inputs.class +) @Operator( group = "quantization" ) @@ -144,8 +150,8 @@ public final class Quantize extends RawOp { private Output outputMax; - private Quantize(Operation operation) { - super(operation); + public Quantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputMin = operation.output(outputIdx++); @@ -358,6 +364,9 @@ public Options ensureMinimumRange(Float ensureMinimumRange) { } } + @OpInputsMetadata( + outputsClass = Quantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantize.java index 4209187b9ef..aa037696632 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizeAndDequantize.OP_NAME, + inputsClass = QuantizeAndDequantize.Inputs.class +) @Operator( group = "quantization" ) @@ -50,8 +56,8 @@ public final class QuantizeAndDequantize extends RawOp implem private Output output; - private QuantizeAndDequantize(Operation operation) { - super(operation); + public QuantizeAndDequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -211,6 +217,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = QuantizeAndDequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV3.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV3.java index 4e2eff27b37..00948dce45b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV3.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV3.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizeAndDequantizeV3.OP_NAME, + inputsClass = QuantizeAndDequantizeV3.Inputs.class +) @Operator( group = "quantization" ) @@ -50,8 +56,8 @@ public final class QuantizeAndDequantizeV3 extends RawOp impl private Output output; - private QuantizeAndDequantizeV3(Operation operation) { - super(operation); + public QuantizeAndDequantizeV3(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -211,6 +217,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = QuantizeAndDequantizeV3.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4.java index 2b80e1cac70..64a32b38c9f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizeAndDequantizeV4.OP_NAME, + inputsClass = QuantizeAndDequantizeV4.Inputs.class +) @Operator( group = "quantization" ) @@ -49,8 +55,8 @@ public final class QuantizeAndDequantizeV4 extends RawOp impl private Output output; - private QuantizeAndDequantizeV4(Operation operation) { - super(operation); + public QuantizeAndDequantizeV4(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -286,6 +292,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = QuantizeAndDequantizeV4.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor to quantize and then dequantize. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4Grad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4Grad.java index 7a70dd57298..8b3d8614840 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4Grad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeAndDequantizeV4Grad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code input_backprop} output */ +@OpMetadata( + opType = QuantizeAndDequantizeV4Grad.OP_NAME, + inputsClass = QuantizeAndDequantizeV4Grad.Inputs.class +) @Operator( group = "quantization" ) @@ -53,8 +59,8 @@ public final class QuantizeAndDequantizeV4Grad extends RawOp private Output inputMaxBackprop; - private QuantizeAndDequantizeV4Grad(Operation operation) { - super(operation); + public QuantizeAndDequantizeV4Grad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; inputBackprop = operation.output(outputIdx++); inputMinBackprop = operation.output(outputIdx++); @@ -152,6 +158,9 @@ public Options axis(Long axis) { } } + @OpInputsMetadata( + outputsClass = QuantizeAndDequantizeV4Grad.class + ) public static class Inputs extends RawOpInputs> { /** * The gradients input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeDownAndShrinkRange.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeDownAndShrinkRange.java index 20302f5e4ad..f59057ba298 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeDownAndShrinkRange.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizeDownAndShrinkRange.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -57,6 +59,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizeDownAndShrinkRange.OP_NAME, + inputsClass = QuantizeDownAndShrinkRange.Inputs.class +) @Operator( group = "quantization" ) @@ -72,8 +78,8 @@ public final class QuantizeDownAndShrinkRange extends RawOp { private Output outputMax; - private QuantizeDownAndShrinkRange(Operation operation) { - super(operation); + public QuantizeDownAndShrinkRange(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputMin = operation.output(outputIdx++); @@ -132,6 +138,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = QuantizeDownAndShrinkRange.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedConcat.java index f95f901d0d2..fb59b8078ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedConcat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedConcat.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = QuantizedConcat.OP_NAME, + inputsClass = QuantizedConcat.Inputs.class +) @Operator( group = "quantization" ) @@ -54,8 +60,8 @@ public final class QuantizedConcat extends RawOp { private Output outputMax; - private QuantizedConcat(Operation operation) { - super(operation); + public QuantizedConcat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputMin = operation.output(outputIdx++); @@ -118,6 +124,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = QuantizedConcat.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D. The dimension along which to concatenate. Must be in the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java index b600d7992f9..8504967d26b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = QuantizedMatMulWithBiasAndDequantize.OP_NAME, + inputsClass = QuantizedMatMulWithBiasAndDequantize.Inputs.class +) public final class QuantizedMatMulWithBiasAndDequantize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class QuantizedMatMulWithBiasAndDequantize exten private Output out; - private QuantizedMatMulWithBiasAndDequantize(Operation operation) { - super(operation); + public QuantizedMatMulWithBiasAndDequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -195,6 +201,9 @@ public Options inputQuantMode(String inputQuantMode) { } } + @OpInputsMetadata( + outputsClass = QuantizedMatMulWithBiasAndDequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java index 2e8463dad54..3bd6f6b498c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = QuantizedMatMulWithBiasAndRequantize.OP_NAME, + inputsClass = QuantizedMatMulWithBiasAndRequantize.Inputs.class +) public final class QuantizedMatMulWithBiasAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class QuantizedMatMulWithBiasAndRequantize exten private Output maxOut; - private QuantizedMatMulWithBiasAndRequantize(Operation operation) { - super(operation); + public QuantizedMatMulWithBiasAndRequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); minOut = operation.output(outputIdx++); @@ -214,6 +220,9 @@ public Options inputQuantMode(String inputQuantMode) { } } + @OpInputsMetadata( + outputsClass = QuantizedMatMulWithBiasAndRequantize.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/RequantizationRange.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/RequantizationRange.java index ad1515a3e56..8be9d51e564 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/RequantizationRange.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/RequantizationRange.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -39,6 +41,10 @@ * used to produce the {@code requested_output_min} and {@code requested_output_max} for * {@code Requantize}. */ +@OpMetadata( + opType = RequantizationRange.OP_NAME, + inputsClass = RequantizationRange.Inputs.class +) @Operator( group = "quantization" ) @@ -52,8 +58,8 @@ public final class RequantizationRange extends RawOp { private Output outputMax; - private RequantizationRange(Operation operation) { - super(operation); + public RequantizationRange(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputMin = operation.output(outputIdx++); outputMax = operation.output(outputIdx++); @@ -98,6 +104,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = RequantizationRange.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Requantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Requantize.java index 0790db45399..ed07bdee180 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Requantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/Requantize.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Requantize.OP_NAME, + inputsClass = Requantize.Inputs.class +) @Operator( group = "quantization" ) @@ -59,8 +65,8 @@ public final class Requantize extends RawOp { private Output outputMax; - private Requantize(Operation operation) { - super(operation); + public Requantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); outputMin = operation.output(outputIdx++); @@ -124,6 +130,9 @@ public Output outputMax() { return outputMax; } + @OpInputsMetadata( + outputsClass = Requantize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedBincount.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedBincount.java index 0de4dea307e..26ae65c0bc0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedBincount.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedBincount.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RaggedBincount.OP_NAME, + inputsClass = RaggedBincount.Inputs.class +) @Operator( group = "ragged" ) @@ -54,8 +60,8 @@ public final class RaggedBincount extends RawOp implements Op private Output output; - private RaggedBincount(Operation operation) { - super(operation); + public RaggedBincount(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -142,6 +148,9 @@ public Options binaryOutput(Boolean binaryOutput) { } } + @OpInputsMetadata( + outputsClass = RaggedBincount.class + ) public static class Inputs extends RawOpInputs> { /** * 1D int64 {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java index 6eff8ab6395..4a7f779ae18 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = RaggedCountSparseOutput.OP_NAME, + inputsClass = RaggedCountSparseOutput.Inputs.class +) public final class RaggedCountSparseOutput extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RaggedCountSparseOutput extends RawOp { private Output outputDenseShape; - private RaggedCountSparseOutput(Operation operation) { - super(operation); + public RaggedCountSparseOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -181,6 +187,9 @@ public Options maxlength(Long maxlength) { } } + @OpInputsMetadata( + outputsClass = RaggedCountSparseOutput.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor containing the row splits of the ragged tensor to count. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java index 9336beb882a..554c33b24c8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output_row_splits} output */ +@OpMetadata( + opType = RaggedCross.OP_NAME, + inputsClass = RaggedCross.Inputs.class +) public final class RaggedCross extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -51,8 +57,8 @@ public final class RaggedCross extends RawOp private Output outputRowSplits; - private RaggedCross(Operation operation) { - super(operation); + public RaggedCross(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputValues = operation.output(outputIdx++); outputRowSplits = operation.output(outputIdx++); @@ -124,6 +130,9 @@ public Output outputRowSplits() { return outputRowSplits; } + @OpInputsMetadata( + outputsClass = RaggedCross.class + ) public static class Inputs extends RawOpInputs> { /** * The values tensor for each RaggedTensor input. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java index 05747069d47..0f747675f76 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -58,6 +60,10 @@ * * @param data type for {@code output_dense_values} output */ +@OpMetadata( + opType = RaggedGather.OP_NAME, + inputsClass = RaggedGather.Inputs.class +) public final class RaggedGather extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -69,8 +75,8 @@ public final class RaggedGather extends RawO private Output outputDenseValues; @SuppressWarnings("unchecked") - private RaggedGather(Operation operation) { - super(operation); + public RaggedGather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputNestedSplitsLength = operation.outputListLength("output_nested_splits"); outputNestedSplits = Arrays.asList((Output[]) operation.outputList(outputIdx, outputNestedSplitsLength)); @@ -129,6 +135,9 @@ public Output outputDenseValues() { return outputDenseValues; } + @OpInputsMetadata( + outputsClass = RaggedGather.class + ) public static class Inputs extends RawOpInputs> { /** * The {@code nested_row_splits} tensors that define the row-partitioning for the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java index 98dcc92980a..3111907e994 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -52,6 +54,10 @@ * * @param data type for {@code rt_dense_values} output */ +@OpMetadata( + opType = RaggedRange.OP_NAME, + inputsClass = RaggedRange.Inputs.class +) public final class RaggedRange extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -62,8 +68,8 @@ public final class RaggedRange extends Raw private Output rtDenseValues; - private RaggedRange(Operation operation) { - super(operation); + public RaggedRange(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; rtNestedSplits = operation.output(outputIdx++); rtDenseValues = operation.output(outputIdx++); @@ -130,6 +136,9 @@ public Output rtDenseValues() { return rtDenseValues; } + @OpInputsMetadata( + outputsClass = RaggedRange.class + ) public static class Inputs extends RawOpInputs> { /** * The starts of each range. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java index e5d5e2c0d79..600b19b219f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -52,6 +54,10 @@ * * @param data type for {@code output_dense_values} output */ +@OpMetadata( + opType = RaggedTensorFromVariant.OP_NAME, + inputsClass = RaggedTensorFromVariant.Inputs.class +) public final class RaggedTensorFromVariant extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -63,8 +69,8 @@ public final class RaggedTensorFromVariant e private Output outputDenseValues; @SuppressWarnings("unchecked") - private RaggedTensorFromVariant(Operation operation) { - super(operation); + public RaggedTensorFromVariant(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputNestedSplitsLength = operation.outputListLength("output_nested_splits"); outputNestedSplits = Arrays.asList((Output[]) operation.outputList(outputIdx, outputNestedSplitsLength)); @@ -143,6 +149,9 @@ public Output outputDenseValues() { return outputDenseValues; } + @OpInputsMetadata( + outputsClass = RaggedTensorFromVariant.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code variant} Tensor containing encoded {@code RaggedTensor}s. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java index 6c15ca7797c..aef6bb6c241 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code sparse_values} output */ +@OpMetadata( + opType = RaggedTensorToSparse.OP_NAME, + inputsClass = RaggedTensorToSparse.Inputs.class +) public final class RaggedTensorToSparse extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -53,8 +59,8 @@ public final class RaggedTensorToSparse extends RawOp { private Output sparseDenseShape; - private RaggedTensorToSparse(Operation operation) { - super(operation); + public RaggedTensorToSparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseIndices = operation.output(outputIdx++); sparseValues = operation.output(outputIdx++); @@ -108,6 +114,9 @@ public Output sparseDenseShape() { return sparseDenseShape; } + @OpInputsMetadata( + outputsClass = RaggedTensorToSparse.class + ) public static class Inputs extends RawOpInputs> { /** * The {@code row_splits} for the {@code RaggedTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java index 3ad46ad6aee..4dc3e6c890f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -54,6 +56,10 @@ * * @param data type for {@code result} output */ +@OpMetadata( + opType = RaggedTensorToTensor.OP_NAME, + inputsClass = RaggedTensorToTensor.Inputs.class +) public final class RaggedTensorToTensor extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -62,8 +68,8 @@ public final class RaggedTensorToTensor extends RawOp implement private Output result; - private RaggedTensorToTensor(Operation operation) { - super(operation); + public RaggedTensorToTensor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; result = operation.output(outputIdx++); } @@ -139,6 +145,9 @@ public Output asOutput() { return result; } + @OpInputsMetadata( + outputsClass = RaggedTensorToTensor.class + ) public static class Inputs extends RawOpInputs> { /** * The desired shape of the output tensor. If left unspecified (empty), diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java index 7d02bcaf565..42a12ffded1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -45,6 +47,10 @@ * is wrapped in a scalar {@code variant} Tensor. See {@code RaggedTensorFromVariant} for the * corresponding decoding logic. */ +@OpMetadata( + opType = RaggedTensorToVariant.OP_NAME, + inputsClass = RaggedTensorToVariant.Inputs.class +) public final class RaggedTensorToVariant extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -54,8 +60,8 @@ public final class RaggedTensorToVariant extends RawOp implements Operand private Output encodedRagged; @SuppressWarnings("unchecked") - private RaggedTensorToVariant(Operation operation) { - super(operation); + public RaggedTensorToVariant(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; encodedRagged = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return (Output) encodedRagged; } + @OpInputsMetadata( + outputsClass = RaggedTensorToVariant.class + ) public static class Inputs extends RawOpInputs { /** * A list of one or more Tensors representing the splits of the input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java index 92dd3f247d5..f381cb631a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code dense_values_grad} output */ +@OpMetadata( + opType = RaggedTensorToVariantGradient.OP_NAME, + inputsClass = RaggedTensorToVariantGradient.Inputs.class +) public final class RaggedTensorToVariantGradient extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class RaggedTensorToVariantGradient extends RawOp private Output denseValuesGrad; - private RaggedTensorToVariantGradient(Operation operation) { - super(operation); + public RaggedTensorToVariantGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; denseValuesGrad = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return denseValuesGrad; } + @OpInputsMetadata( + outputsClass = RaggedTensorToVariantGradient.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code variant} Tensor containing encoded {@code RaggedTensor} gradients. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AllCandidateSampler.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AllCandidateSampler.java index c22544c3206..9f762d54632 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AllCandidateSampler.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AllCandidateSampler.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * the sampled candidates must be chosen independently of the context and of the * true labels. */ +@OpMetadata( + opType = AllCandidateSampler.OP_NAME, + inputsClass = AllCandidateSampler.Inputs.class +) @Operator( group = "random" ) @@ -56,8 +62,8 @@ public final class AllCandidateSampler extends RawOp { private Output sampledExpectedCount; - private AllCandidateSampler(Operation operation) { - super(operation); + public AllCandidateSampler(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sampledCandidates = operation.output(outputIdx++); trueExpectedCount = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = AllCandidateSampler.class + ) public static class Inputs extends RawOpInputs { /** * A batch_size * num_true matrix, in which each row contains the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java index 44c47dd9bdf..3120b1f7946 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; /** * The AnonymousRandomSeedGenerator operation */ +@OpMetadata( + opType = AnonymousRandomSeedGenerator.OP_NAME, + inputsClass = AnonymousRandomSeedGenerator.Inputs.class +) public final class AnonymousRandomSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class AnonymousRandomSeedGenerator extends RawOp { private Output deleter; @SuppressWarnings("unchecked") - private AnonymousRandomSeedGenerator(Operation operation) { - super(operation); + public AnonymousRandomSeedGenerator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); deleter = operation.output(outputIdx++); @@ -88,6 +94,9 @@ public Output deleter() { return deleter; } + @OpInputsMetadata( + outputsClass = AnonymousRandomSeedGenerator.class + ) public static class Inputs extends RawOpInputs { /** * The seed input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java index 09f83de9c76..e3e20b71bb4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,6 +36,10 @@ /** * The AnonymousSeedGenerator operation */ +@OpMetadata( + opType = AnonymousSeedGenerator.OP_NAME, + inputsClass = AnonymousSeedGenerator.Inputs.class +) public final class AnonymousSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class AnonymousSeedGenerator extends RawOp { private Output deleter; @SuppressWarnings("unchecked") - private AnonymousSeedGenerator(Operation operation) { - super(operation); + public AnonymousSeedGenerator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); deleter = operation.output(outputIdx++); @@ -91,6 +97,9 @@ public Output deleter() { return deleter; } + @OpInputsMetadata( + outputsClass = AnonymousSeedGenerator.class + ) public static class Inputs extends RawOpInputs { /** * The seed input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java index 6dad0b8957e..146310df6e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The DeleteRandomSeedGenerator operation */ +@OpMetadata( + opType = DeleteRandomSeedGenerator.OP_NAME, + inputsClass = DeleteRandomSeedGenerator.Inputs.class +) public final class DeleteRandomSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "DeleteRandomSeedGenerator"; - private DeleteRandomSeedGenerator(Operation operation) { - super(operation); + public DeleteRandomSeedGenerator(Operation operation) { + super(operation, OP_NAME); } /** @@ -60,6 +66,9 @@ public static DeleteRandomSeedGenerator create(Scope scope, Operand { /** * The handle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java index 33bcd5d5714..37cb0c3ca8e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The DeleteSeedGenerator operation */ +@OpMetadata( + opType = DeleteSeedGenerator.OP_NAME, + inputsClass = DeleteSeedGenerator.Inputs.class +) public final class DeleteSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "DeleteSeedGenerator"; - private DeleteSeedGenerator(Operation operation) { - super(operation); + public DeleteSeedGenerator(Operation operation) { + super(operation, OP_NAME); } /** @@ -60,6 +66,9 @@ public static DeleteSeedGenerator create(Scope scope, Operand h return new DeleteSeedGenerator(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = DeleteSeedGenerator.class + ) public static class Inputs extends RawOpInputs { /** * The handle input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java index 0eafa99d317..8af88774a35 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The DummySeedGenerator operation */ +@OpMetadata( + opType = DummySeedGenerator.OP_NAME, + inputsClass = DummySeedGenerator.Inputs.class +) public final class DummySeedGenerator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class DummySeedGenerator extends RawOp implements Operand { private Output handle; @SuppressWarnings("unchecked") - private DummySeedGenerator(Operation operation) { - super(operation); + public DummySeedGenerator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -76,6 +82,9 @@ public Output asOutput() { return (Output) handle; } + @OpInputsMetadata( + outputsClass = DummySeedGenerator.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new DummySeedGenerator(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/LogUniformCandidateSampler.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/LogUniformCandidateSampler.java index b49dd8fcd72..2c51755d5d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/LogUniformCandidateSampler.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/LogUniformCandidateSampler.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * the sampled candidates must be chosen independently of the context and of the * true labels. */ +@OpMetadata( + opType = LogUniformCandidateSampler.OP_NAME, + inputsClass = LogUniformCandidateSampler.Inputs.class +) @Operator( group = "random" ) @@ -56,8 +62,8 @@ public final class LogUniformCandidateSampler extends RawOp { private Output sampledExpectedCount; - private LogUniformCandidateSampler(Operation operation) { - super(operation); + public LogUniformCandidateSampler(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sampledCandidates = operation.output(outputIdx++); trueExpectedCount = operation.output(outputIdx++); @@ -194,6 +200,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = LogUniformCandidateSampler.class + ) public static class Inputs extends RawOpInputs { /** * A batch_size * num_true matrix, in which each row contains the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/Multinomial.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/Multinomial.java index 14af4090bd8..9edf3fbebf2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/Multinomial.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/Multinomial.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Multinomial.OP_NAME, + inputsClass = Multinomial.Inputs.class +) @Operator( group = "random" ) @@ -50,8 +56,8 @@ public final class Multinomial extends RawOp implements Opera private Output output; - private Multinomial(Operation operation) { - super(operation); + public Multinomial(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -180,6 +186,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = Multinomial.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D Tensor with shape {@code [batch_size, num_classes]}. Each slice {@code [i, :]} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java index bbf03174017..95a3f41e8bf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = NonDeterministicInts.OP_NAME, + inputsClass = NonDeterministicInts.Inputs.class +) public final class NonDeterministicInts extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class NonDeterministicInts extends RawOp implement private Output output; - private NonDeterministicInts(Operation operation) { - super(operation); + public NonDeterministicInts(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = NonDeterministicInts.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ParameterizedTruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ParameterizedTruncatedNormal.java index bce121c2d44..4d04fb97e18 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ParameterizedTruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ParameterizedTruncatedNormal.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ParameterizedTruncatedNormal.OP_NAME, + inputsClass = ParameterizedTruncatedNormal.Inputs.class +) @Operator( group = "random" ) @@ -49,8 +55,8 @@ public final class ParameterizedTruncatedNormal extends RawOp private Output output; - private ParameterizedTruncatedNormal(Operation operation) { - super(operation); + public ParameterizedTruncatedNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -167,6 +173,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = ParameterizedTruncatedNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. Batches are indexed by the 0th dimension. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGamma.java index eecd013f0af..bbef7f42515 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGamma.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomGamma.OP_NAME, + inputsClass = RandomGamma.Inputs.class +) @Operator( group = "random" ) @@ -50,8 +56,8 @@ public final class RandomGamma extends RawOp implements Opera private Output output; - private RandomGamma(Operation operation) { - super(operation); + public RandomGamma(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -163,6 +169,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomGamma.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D integer tensor. Shape of independent samples to draw from each diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java index 2355dde788e..220cfa27ee3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomGammaGrad.OP_NAME, + inputsClass = RandomGammaGrad.Inputs.class +) public final class RandomGammaGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RandomGammaGrad extends RawOp implements O private Output output; - private RandomGammaGrad(Operation operation) { - super(operation); + public RandomGammaGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RandomGammaGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The alpha input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomPoisson.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomPoisson.java index 2e98623df39..1f58f155643 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomPoisson.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomPoisson.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomPoisson.OP_NAME, + inputsClass = RandomPoisson.Inputs.class +) @Operator( group = "random" ) @@ -57,8 +63,8 @@ public final class RandomPoisson extends RawOp implements Ope private Output output; - private RandomPoisson(Operation operation) { - super(operation); + public RandomPoisson(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -192,6 +198,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomPoisson.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D integer tensor. Shape of independent samples to draw from each diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomShuffle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomShuffle.java index 35dad551e97..93f7482cbdc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomShuffle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomShuffle.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomShuffle.OP_NAME, + inputsClass = RandomShuffle.Inputs.class +) @Operator( group = "random" ) @@ -55,8 +61,8 @@ public final class RandomShuffle extends RawOp implements Opera private Output output; - private RandomShuffle(Operation operation) { - super(operation); + public RandomShuffle(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -163,6 +169,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomShuffle.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor to be shuffled. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomStandardNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomStandardNormal.java index fee428b29b7..91fc67dbcf3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomStandardNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomStandardNormal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomStandardNormal.OP_NAME, + inputsClass = RandomStandardNormal.Inputs.class +) @Operator( group = "random" ) @@ -49,8 +55,8 @@ public final class RandomStandardNormal extends RawOp impleme private Output output; - private RandomStandardNormal(Operation operation) { - super(operation); + public RandomStandardNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -158,6 +164,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomStandardNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniform.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniform.java index 25ded48d9bf..4cac0972cd5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniform.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniform.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomUniform.OP_NAME, + inputsClass = RandomUniform.Inputs.class +) @Operator( group = "random" ) @@ -50,8 +56,8 @@ public final class RandomUniform extends RawOp implements Ope private Output output; - private RandomUniform(Operation operation) { - super(operation); + public RandomUniform(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -159,6 +165,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomUniform.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniformInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniformInt.java index 417766f1e2a..3121cc28720 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniformInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomUniformInt.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RandomUniformInt.OP_NAME, + inputsClass = RandomUniformInt.Inputs.class +) @Operator( group = "random" ) @@ -53,8 +59,8 @@ public final class RandomUniformInt extends RawOp implements private Output output; - private RandomUniformInt(Operation operation) { - super(operation); + public RandomUniformInt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -164,6 +170,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = RandomUniformInt.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RecordInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RecordInput.java index 9b4cb861f58..978e2b9b1cb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RecordInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RecordInput.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** * Emits randomized records. */ +@OpMetadata( + opType = RecordInput.OP_NAME, + inputsClass = RecordInput.Inputs.class +) @Operator( group = "random" ) @@ -44,8 +50,8 @@ public final class RecordInput extends RawOp implements Operand { private Output records; - private RecordInput(Operation operation) { - super(operation); + public RecordInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; records = operation.output(outputIdx++); } @@ -253,6 +259,9 @@ public Options compressionType(String compressionType) { } } + @OpInputsMetadata( + outputsClass = RecordInput.class + ) public static class Inputs extends RawOpInputs { /** * Glob pattern for the data files. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java index d6ceea1475b..8ac6ce37392 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * (or any other distribution). The actual increment added to the * counter is an unspecified implementation choice. */ +@OpMetadata( + opType = RngReadAndSkip.OP_NAME, + inputsClass = RngReadAndSkip.Inputs.class +) public final class RngReadAndSkip extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class RngReadAndSkip extends RawOp implements Operand { private Output value; - private RngReadAndSkip(Operation operation) { - super(operation); + public RngReadAndSkip(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; value = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return value; } + @OpInputsMetadata( + outputsClass = RngReadAndSkip.class + ) public static class Inputs extends RawOpInputs { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java index e7eaaa81283..8e53d849cb7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -36,14 +38,18 @@ * (or any other distribution). The actual increment added to the * counter is an unspecified implementation detail. */ +@OpMetadata( + opType = RngSkip.OP_NAME, + inputsClass = RngSkip.Inputs.class +) public final class RngSkip extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "RngSkip"; - private RngSkip(Operation operation) { - super(operation); + public RngSkip(Operation operation) { + super(operation, OP_NAME); } /** @@ -67,6 +73,9 @@ public static RngSkip create(Scope scope, Operand resource, return new RngSkip(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = RngSkip.class + ) public static class Inputs extends RawOpInputs { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulRandomBinomial.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulRandomBinomial.java index 8f1467afe73..08ae95ce49d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulRandomBinomial.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulRandomBinomial.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatefulRandomBinomial.OP_NAME, + inputsClass = StatefulRandomBinomial.Inputs.class +) @Operator( group = "random" ) @@ -50,8 +56,8 @@ public final class StatefulRandomBinomial extends RawOp imple private Output output; - private StatefulRandomBinomial(Operation operation) { - super(operation); + public StatefulRandomBinomial(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -121,6 +127,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatefulRandomBinomial.class + ) public static class Inputs extends RawOpInputs> { /** * The resource input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulStandardNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulStandardNormal.java index f85677851fe..57c4ae60efc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulStandardNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulStandardNormal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatefulStandardNormal.OP_NAME, + inputsClass = StatefulStandardNormal.Inputs.class +) @Operator( group = "random" ) @@ -51,8 +57,8 @@ public final class StatefulStandardNormal extends RawOp impleme private Output output; - private StatefulStandardNormal(Operation operation) { - super(operation); + public StatefulStandardNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -114,6 +120,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatefulStandardNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java index ded58540672..962a6cfc095 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatefulTruncatedNormal.OP_NAME, + inputsClass = StatefulTruncatedNormal.Inputs.class +) public final class StatefulTruncatedNormal extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class StatefulTruncatedNormal extends RawOp implem private Output output; - private StatefulTruncatedNormal(Operation operation) { - super(operation); + public StatefulTruncatedNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -112,6 +118,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatefulTruncatedNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java index 4605231e1cb..fb3280dcc53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatefulUniform.OP_NAME, + inputsClass = StatefulUniform.Inputs.class +) public final class StatefulUniform extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class StatefulUniform extends RawOp implements Ope private Output output; - private StatefulUniform(Operation operation) { - super(operation); + public StatefulUniform(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -110,6 +116,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatefulUniform.class + ) public static class Inputs extends RawOpInputs> { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java index 527c1649f40..8dd4a433e45 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatefulUniformFullInt.OP_NAME, + inputsClass = StatefulUniformFullInt.Inputs.class +) public final class StatefulUniformFullInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class StatefulUniformFullInt extends RawOp impleme private Output output; - private StatefulUniformFullInt(Operation operation) { - super(operation); + public StatefulUniformFullInt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatefulUniformFullInt.class + ) public static class Inputs extends RawOpInputs> { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java index e9d475b2d72..2b723d4c6f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatefulUniformInt.OP_NAME, + inputsClass = StatefulUniformInt.Inputs.class +) public final class StatefulUniformInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class StatefulUniformInt extends RawOp implements private Output output; - private StatefulUniformInt(Operation operation) { - super(operation); + public StatefulUniformInt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatefulUniformInt.class + ) public static class Inputs extends RawOpInputs> { /** * The handle of the resource variable that stores the state of the RNG. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessMultinomial.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessMultinomial.java index 6d09e4dd755..adec914d632 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessMultinomial.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessMultinomial.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessMultinomial.OP_NAME, + inputsClass = StatelessMultinomial.Inputs.class +) @Operator( group = "random" ) @@ -50,8 +56,8 @@ public final class StatelessMultinomial extends RawOp impleme private Output output; - private StatelessMultinomial(Operation operation) { - super(operation); + public StatelessMultinomial(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -115,6 +121,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessMultinomial.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D Tensor with shape {@code [batch_size, num_classes]}. Each slice {@code [i, :]} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java index 70f982ac127..712a9fd0c57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessParameterizedTruncatedNormal.OP_NAME, + inputsClass = StatelessParameterizedTruncatedNormal.Inputs.class +) public final class StatelessParameterizedTruncatedNormal extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class StatelessParameterizedTruncatedNormal exte private Output output; - private StatelessParameterizedTruncatedNormal(Operation operation) { - super(operation); + public StatelessParameterizedTruncatedNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessParameterizedTruncatedNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java index 555f7113290..be2f0f66a1f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomBinomial.OP_NAME, + inputsClass = StatelessRandomBinomial.Inputs.class +) public final class StatelessRandomBinomial extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class StatelessRandomBinomial extends RawOp impl private Output output; - private StatelessRandomBinomial(Operation operation) { - super(operation); + public StatelessRandomBinomial(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -119,6 +125,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomBinomial.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java index b5383fa2d8a..2b53b2b1ab0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomGamma.OP_NAME, + inputsClass = StatelessRandomGamma.Inputs.class +) public final class StatelessRandomGamma extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class StatelessRandomGamma extends RawOp impleme private Output output; - private StatelessRandomGamma(Operation operation) { - super(operation); + public StatelessRandomGamma(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomGamma.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java index 762d5f95512..db2e3a5581b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * Picks the best algorithm based on device, and scrambles seed into key and counter. * This op picks the best counter-based RNG algorithm based on device, and scrambles a shape-[2] seed into a key and a counter, both needed by the counter-based algorithm. The scrambling is opaque but approximately satisfies the property that different seed results in different key/counter pair (which will in turn result in different random numbers). */ +@OpMetadata( + opType = StatelessRandomGetKeyCounterAlg.OP_NAME, + inputsClass = StatelessRandomGetKeyCounterAlg.Inputs.class +) public final class StatelessRandomGetKeyCounterAlg extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class StatelessRandomGetKeyCounterAlg extends RawOp { private Output alg; @SuppressWarnings("unchecked") - private StatelessRandomGetKeyCounterAlg(Operation operation) { - super(operation); + public StatelessRandomGetKeyCounterAlg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; key = operation.output(outputIdx++); counter = operation.output(outputIdx++); @@ -101,6 +107,9 @@ public Output alg() { return alg; } + @OpInputsMetadata( + outputsClass = StatelessRandomGetKeyCounterAlg.class + ) public static class Inputs extends RawOpInputs { /** * 2 seeds (shape [2]). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormal.java index dc42caec6c2..11cc0898f5d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomNormal.OP_NAME, + inputsClass = StatelessRandomNormal.Inputs.class +) @Operator( group = "random" ) @@ -51,8 +57,8 @@ public final class StatelessRandomNormal extends RawOp implem private Output output; - private StatelessRandomNormal(Operation operation) { - super(operation); + public StatelessRandomNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -109,6 +115,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java index 09e30c2e27c..ec9804de75e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomNormalV2.OP_NAME, + inputsClass = StatelessRandomNormalV2.Inputs.class +) public final class StatelessRandomNormalV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class StatelessRandomNormalV2 extends RawOp impl private Output output; - private StatelessRandomNormalV2(Operation operation) { - super(operation); + public StatelessRandomNormalV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -115,6 +121,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomNormalV2.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java index 4986962addd..6386c6a86de 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomPoisson.OP_NAME, + inputsClass = StatelessRandomPoisson.Inputs.class +) public final class StatelessRandomPoisson extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class StatelessRandomPoisson extends RawOp imple private Output output; - private StatelessRandomPoisson(Operation operation) { - super(operation); + public StatelessRandomPoisson(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomPoisson.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniform.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniform.java index 6ea9c4ff4b9..afab5de72ac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniform.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniform.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomUniform.OP_NAME, + inputsClass = StatelessRandomUniform.Inputs.class +) @Operator( group = "random" ) @@ -52,8 +58,8 @@ public final class StatelessRandomUniform extends RawOp imple private Output output; - private StatelessRandomUniform(Operation operation) { - super(operation); + public StatelessRandomUniform(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -110,6 +116,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomUniform.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java index 4857e11b0b0..bbe9d36c533 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomUniformFullInt.OP_NAME, + inputsClass = StatelessRandomUniformFullInt.Inputs.class +) public final class StatelessRandomUniformFullInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class StatelessRandomUniformFullInt extends RawO private Output output; - private StatelessRandomUniformFullInt(Operation operation) { - super(operation); + public StatelessRandomUniformFullInt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomUniformFullInt.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java index d31fd80c989..8c61332aacb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomUniformFullIntV2.OP_NAME, + inputsClass = StatelessRandomUniformFullIntV2.Inputs.class +) public final class StatelessRandomUniformFullIntV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class StatelessRandomUniformFullIntV2 extends Ra private Output output; - private StatelessRandomUniformFullIntV2(Operation operation) { - super(operation); + public StatelessRandomUniformFullIntV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomUniformFullIntV2.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java index 413724622a7..02676c26abd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomUniformInt.OP_NAME, + inputsClass = StatelessRandomUniformInt.Inputs.class +) public final class StatelessRandomUniformInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class StatelessRandomUniformInt extends RawOp im private Output output; - private StatelessRandomUniformInt(Operation operation) { - super(operation); + public StatelessRandomUniformInt(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomUniformInt.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java index 133643bc187..7160aa2f156 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomUniformIntV2.OP_NAME, + inputsClass = StatelessRandomUniformIntV2.Inputs.class +) public final class StatelessRandomUniformIntV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class StatelessRandomUniformIntV2 extends RawOp private Output output; - private StatelessRandomUniformIntV2(Operation operation) { - super(operation); + public StatelessRandomUniformIntV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomUniformIntV2.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java index e6414b181d2..8a7e5aa6b57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessRandomUniformV2.OP_NAME, + inputsClass = StatelessRandomUniformV2.Inputs.class +) public final class StatelessRandomUniformV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class StatelessRandomUniformV2 extends RawOp imp private Output output; - private StatelessRandomUniformV2(Operation operation) { - super(operation); + public StatelessRandomUniformV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -116,6 +122,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessRandomUniformV2.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormal.java index bc78fb53e6f..92be5db97b4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessTruncatedNormal.OP_NAME, + inputsClass = StatelessTruncatedNormal.Inputs.class +) @Operator( group = "random" ) @@ -53,8 +59,8 @@ public final class StatelessTruncatedNormal extends RawOp imp private Output output; - private StatelessTruncatedNormal(Operation operation) { - super(operation); + public StatelessTruncatedNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -111,6 +117,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessTruncatedNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java index da649d92ac0..318c1ccd592 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = StatelessTruncatedNormalV2.OP_NAME, + inputsClass = StatelessTruncatedNormalV2.Inputs.class +) public final class StatelessTruncatedNormalV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -51,8 +57,8 @@ public final class StatelessTruncatedNormalV2 extends RawOp i private Output output; - private StatelessTruncatedNormalV2(Operation operation) { - super(operation); + public StatelessTruncatedNormalV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -117,6 +123,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StatelessTruncatedNormalV2.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/TruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/TruncatedNormal.java index 2f0d60114e5..f03d07368dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/TruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/TruncatedNormal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TruncatedNormal.OP_NAME, + inputsClass = TruncatedNormal.Inputs.class +) @Operator( group = "random" ) @@ -51,8 +57,8 @@ public final class TruncatedNormal extends RawOp implements O private Output output; - private TruncatedNormal(Operation operation) { - super(operation); + public TruncatedNormal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -161,6 +167,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = TruncatedNormal.class + ) public static class Inputs extends RawOpInputs> { /** * The shape of the output tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/UniformCandidateSampler.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/UniformCandidateSampler.java index 5d95242967d..71d248ebf36 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/UniformCandidateSampler.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/UniformCandidateSampler.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * the sampled candidates must be chosen independently of the context and of the * true labels. */ +@OpMetadata( + opType = UniformCandidateSampler.OP_NAME, + inputsClass = UniformCandidateSampler.Inputs.class +) @Operator( group = "random" ) @@ -56,8 +62,8 @@ public final class UniformCandidateSampler extends RawOp { private Output sampledExpectedCount; - private UniformCandidateSampler(Operation operation) { - super(operation); + public UniformCandidateSampler(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sampledCandidates = operation.output(outputIdx++); trueExpectedCount = operation.output(outputIdx++); @@ -194,6 +200,9 @@ public Options seed2(Long seed2) { } } + @OpInputsMetadata( + outputsClass = UniformCandidateSampler.class + ) public static class Inputs extends RawOpInputs { /** * A batch_size * num_true matrix, in which each row contains the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastRecvV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastRecvV2.java index 167ed5dd8c8..e189877adb8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastRecvV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastRecvV2.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = CollectiveBcastRecvV2.OP_NAME, + inputsClass = CollectiveBcastRecvV2.Inputs.class +) public final class CollectiveBcastRecvV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class CollectiveBcastRecvV2 extends RawOp implemen private Output data; - private CollectiveBcastRecvV2(Operation operation) { - super(operation); + public CollectiveBcastRecvV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -158,6 +164,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = CollectiveBcastRecvV2.class + ) public static class Inputs extends RawOpInputs> { /** * The groupSize input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastSendV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastSendV2.java index afce1f5da7f..68a56685261 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastSendV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/CollectiveBcastSendV2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code data} output */ +@OpMetadata( + opType = CollectiveBcastSendV2.OP_NAME, + inputsClass = CollectiveBcastSendV2.Inputs.class +) public final class CollectiveBcastSendV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class CollectiveBcastSendV2 extends RawOp implemen private Output data; - private CollectiveBcastSendV2(Operation operation) { - super(operation); + public CollectiveBcastSendV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; data = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options timeoutSeconds(Float timeoutSeconds) { } } + @OpInputsMetadata( + outputsClass = CollectiveBcastSendV2.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/GetOptions.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/GetOptions.java index 472e350db9e..60f0af859fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/GetOptions.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/GetOptions.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Returns the {@code tf.data.Options} attached to {@code input_dataset}. */ +@OpMetadata( + opType = GetOptions.OP_NAME, + inputsClass = GetOptions.Inputs.class +) public final class GetOptions extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class GetOptions extends RawOp implements Operand { private Output serializedOptions; - private GetOptions(Operation operation) { - super(operation); + public GetOptions(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; serializedOptions = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return serializedOptions; } + @OpInputsMetadata( + outputsClass = GetOptions.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing the input dataset. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParameters.java index 10afcd94130..e3a751e5fb0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingFrequencyEstimatorParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingFrequencyEstimatorParameters.Inputs.class +) public final class LoadTPUEmbeddingFrequencyEstimatorParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingFrequencyEstimatorParameters"; - private LoadTPUEmbeddingFrequencyEstimatorParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingFrequencyEstimatorParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -161,6 +167,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingFrequencyEstimatorParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the frequency estimator optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java index 0ccf337801b..20dff3115d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"; - private LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -164,6 +170,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the frequency estimator optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java index 37ad5cf9e4b..e50431dc817 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingFrequencyEstimatorParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingFrequencyEstimatorParameters.Inputs.class +) public final class RetrieveTPUEmbeddingFrequencyEstimatorParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RetrieveTPUEmbeddingFrequencyEstimatorParameters extends RawO private Output lastHitStep; - private RetrieveTPUEmbeddingFrequencyEstimatorParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingFrequencyEstimatorParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); lastHitStep = operation.output(outputIdx++); @@ -181,6 +187,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingFrequencyEstimatorParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java index 4a9031c45e1..15fb4ba7a5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebu private Output gradientAccumulators; - private RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); lastHitStep = operation.output(outputIdx++); @@ -194,6 +200,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetAlg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetAlg.java index e5daa87691e..eef87bc6798 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetAlg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetAlg.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; /** * Picks the best counter-based RNG algorithm based on device. * This op picks the best counter-based RNG algorithm based on device. */ +@OpMetadata( + opType = StatelessRandomGetAlg.OP_NAME, + inputsClass = StatelessRandomGetAlg.Inputs.class +) public final class StatelessRandomGetAlg extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class StatelessRandomGetAlg extends RawOp implements Operand alg; - private StatelessRandomGetAlg(Operation operation) { - super(operation); + public StatelessRandomGetAlg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; alg = operation.output(outputIdx++); } @@ -75,6 +81,9 @@ public Output asOutput() { return alg; } + @OpInputsMetadata( + outputsClass = StatelessRandomGetAlg.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new StatelessRandomGetAlg(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetKeyCounter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetKeyCounter.java index 16c50e5c0ea..184f9765210 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetKeyCounter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/rawops/StatelessRandomGetKeyCounter.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Scrambles seed into key and counter, using the best algorithm based on device. * This op scrambles a shape-[2] seed into a key and a counter, both needed by counter-based RNG algorithms. The scrambing uses the best algorithm based on device. The scrambling is opaque but approximately satisfies the property that different seed results in different key/counter pair (which will in turn result in different random numbers). */ +@OpMetadata( + opType = StatelessRandomGetKeyCounter.OP_NAME, + inputsClass = StatelessRandomGetKeyCounter.Inputs.class +) public final class StatelessRandomGetKeyCounter extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class StatelessRandomGetKeyCounter extends RawOp { private Output counter; @SuppressWarnings("unchecked") - private StatelessRandomGetKeyCounter(Operation operation) { - super(operation); + public StatelessRandomGetKeyCounter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; key = operation.output(outputIdx++); counter = operation.output(outputIdx++); @@ -87,6 +93,9 @@ public Output counter() { return counter; } + @OpInputsMetadata( + outputsClass = StatelessRandomGetKeyCounter.class + ) public static class Inputs extends RawOpInputs { /** * 2 seeds (shape [2]). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java index a1bfbcd2bc3..48bb399cb03 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscAbs.OP_NAME, + inputsClass = RiscAbs.Inputs.class +) public final class RiscAbs extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscAbs extends RawOp implements Operand y; - private RiscAbs(Operation operation) { - super(operation); + public RiscAbs(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscAbs.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java index cbaeb97e7db..4bf45e6ce3e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscAdd.OP_NAME, + inputsClass = RiscAdd.Inputs.class +) public final class RiscAdd extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class RiscAdd extends RawOp implements Operand z; - private RiscAdd(Operation operation) { - super(operation); + public RiscAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscAdd.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java index ffe646419f5..f5f7cb5bbe6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscBinaryArithmetic.OP_NAME, + inputsClass = RiscBinaryArithmetic.Inputs.class +) public final class RiscBinaryArithmetic extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscBinaryArithmetic extends RawOp impleme private Output z; - private RiscBinaryArithmetic(Operation operation) { - super(operation); + public RiscBinaryArithmetic(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscBinaryArithmetic.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java index d0221df4424..fe475060238 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * The RiscBinaryComparison operation */ +@OpMetadata( + opType = RiscBinaryComparison.OP_NAME, + inputsClass = RiscBinaryComparison.Inputs.class +) public final class RiscBinaryComparison extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class RiscBinaryComparison extends RawOp implements Operand private Output z; - private RiscBinaryComparison(Operation operation) { - super(operation); + public RiscBinaryComparison(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscBinaryComparison.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java index 9e13bc53c9a..6d2c64e587b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscBitcast.OP_NAME, + inputsClass = RiscBitcast.Inputs.class +) public final class RiscBitcast extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscBitcast extends RawOp implements Operand private Output y; - private RiscBitcast(Operation operation) { - super(operation); + public RiscBitcast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscBitcast.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java index f23d678ee3c..25885b631b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscBroadcast.OP_NAME, + inputsClass = RiscBroadcast.Inputs.class +) public final class RiscBroadcast extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscBroadcast extends RawOp implements Opera private Output output; - private RiscBroadcast(Operation operation) { - super(operation); + public RiscBroadcast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscBroadcast.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java index 19713cb9325..4b14d20072a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscCast.OP_NAME, + inputsClass = RiscCast.Inputs.class +) public final class RiscCast extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscCast extends RawOp implements Operand private Output y; - private RiscCast(Operation operation) { - super(operation); + public RiscCast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscCast.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java index 993e6fe0c4f..cf6173756f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscCeil.OP_NAME, + inputsClass = RiscCeil.Inputs.class +) public final class RiscCeil extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscCeil extends RawOp implements Operand< private Output y; - private RiscCeil(Operation operation) { - super(operation); + public RiscCeil(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscCeil.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java index 214e0f0729c..603c50dfbc4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscCholesky.OP_NAME, + inputsClass = RiscCholesky.Inputs.class +) public final class RiscCholesky extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscCholesky extends RawOp implements Oper private Output output; - private RiscCholesky(Operation operation) { - super(operation); + public RiscCholesky(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscCholesky.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java index b09c64975f3..8097ccaf66e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscConcat.OP_NAME, + inputsClass = RiscConcat.Inputs.class +) public final class RiscConcat extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RiscConcat extends RawOp implements Operand< private Output output; - private RiscConcat(Operation operation) { - super(operation); + public RiscConcat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscConcat.class + ) public static class Inputs extends RawOpInputs> { /** * The values input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java index 4799eeaea37..b899ee37385 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscCondition.OP_NAME, + inputsClass = RiscCondition.Inputs.class +) public final class RiscCondition extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class RiscCondition extends RawOp implements Ope private Output output; - private RiscCondition(Operation operation) { - super(operation); + public RiscCondition(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscCondition.class + ) public static class Inputs extends RawOpInputs> { /** * The pred input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java index eef8246faa8..5315c043ea6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscConv.OP_NAME, + inputsClass = RiscConv.Inputs.class +) public final class RiscConv extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscConv extends RawOp implements Operand< private Output output; - private RiscConv(Operation operation) { - super(operation); + public RiscConv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -180,6 +186,9 @@ public Options dilations(Long... dilations) { } } + @OpInputsMetadata( + outputsClass = RiscConv.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java index db02acbfcdf..95be103407b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscCos.OP_NAME, + inputsClass = RiscCos.Inputs.class +) public final class RiscCos extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscCos extends RawOp implements Operand y; - private RiscCos(Operation operation) { - super(operation); + public RiscCos(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscCos.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java index 6dcdf4e796b..4feb615f370 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscDiv.OP_NAME, + inputsClass = RiscDiv.Inputs.class +) public final class RiscDiv extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscDiv extends RawOp implements Operand z; - private RiscDiv(Operation operation) { - super(operation); + public RiscDiv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscDiv.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java index a3db102582d..a64374075ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code product} output */ +@OpMetadata( + opType = RiscDot.OP_NAME, + inputsClass = RiscDot.Inputs.class +) public final class RiscDot extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscDot extends RawOp implements Operand product; - private RiscDot(Operation operation) { - super(operation); + public RiscDot(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; product = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options transposeB(Boolean transposeB) { } } + @OpInputsMetadata( + outputsClass = RiscDot.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java index 6868da2c4bb..5f8a0095d84 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscExp.OP_NAME, + inputsClass = RiscExp.Inputs.class +) public final class RiscExp extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscExp extends RawOp implements Operand y; - private RiscExp(Operation operation) { - super(operation); + public RiscExp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscExp.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java index 31b27b88913..f418d9cd514 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscFft.OP_NAME, + inputsClass = RiscFft.Inputs.class +) public final class RiscFft extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscFft extends RawOp implements Operand private Output output; - private RiscFft(Operation operation) { - super(operation); + public RiscFft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscFft.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java index aabcfb22dc7..4091e029022 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscFloor.OP_NAME, + inputsClass = RiscFloor.Inputs.class +) public final class RiscFloor extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscFloor extends RawOp implements Operand private Output y; - private RiscFloor(Operation operation) { - super(operation); + public RiscFloor(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscFloor.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java index f4128861b39..bbf099905fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscGather.OP_NAME, + inputsClass = RiscGather.Inputs.class +) public final class RiscGather extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscGather extends RawOp implements Operand< private Output output; - private RiscGather(Operation operation) { - super(operation); + public RiscGather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -125,6 +131,9 @@ public Options batchDims(Long batchDims) { } } + @OpInputsMetadata( + outputsClass = RiscGather.class + ) public static class Inputs extends RawOpInputs> { /** * The params input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java index 120bb80b8ea..bef9b1f68c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscImag.OP_NAME, + inputsClass = RiscImag.Inputs.class +) public final class RiscImag extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class RiscImag extends RawOp implements Operand< private Output output; - private RiscImag(Operation operation) { - super(operation); + public RiscImag(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscImag.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java index b87b3932c05..de9e34af151 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * The RiscIsFinite operation */ +@OpMetadata( + opType = RiscIsFinite.OP_NAME, + inputsClass = RiscIsFinite.Inputs.class +) public final class RiscIsFinite extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class RiscIsFinite extends RawOp implements Operand { private Output y; - private RiscIsFinite(Operation operation) { - super(operation); + public RiscIsFinite(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscIsFinite.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java index b4428844f90..083a4ef1f51 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscLog.OP_NAME, + inputsClass = RiscLog.Inputs.class +) public final class RiscLog extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscLog extends RawOp implements Operand y; - private RiscLog(Operation operation) { - super(operation); + public RiscLog(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscLog.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java index a4ef38dd73d..d0cde8399e0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; /** * The RiscLogicalAnd operation */ +@OpMetadata( + opType = RiscLogicalAnd.OP_NAME, + inputsClass = RiscLogicalAnd.Inputs.class +) public final class RiscLogicalAnd extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -40,8 +46,8 @@ public final class RiscLogicalAnd extends RawOp implements Operand { private Output z; - private RiscLogicalAnd(Operation operation) { - super(operation); + public RiscLogicalAnd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscLogicalAnd.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java index c67de8825d5..f9a7084dd1b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; /** * The RiscLogicalNot operation */ +@OpMetadata( + opType = RiscLogicalNot.OP_NAME, + inputsClass = RiscLogicalNot.Inputs.class +) public final class RiscLogicalNot extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -40,8 +46,8 @@ public final class RiscLogicalNot extends RawOp implements Operand { private Output z; - private RiscLogicalNot(Operation operation) { - super(operation); + public RiscLogicalNot(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -76,6 +82,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscLogicalNot.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java index 59273fb3f82..85d2fa52e18 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; /** * The RiscLogicalOr operation */ +@OpMetadata( + opType = RiscLogicalOr.OP_NAME, + inputsClass = RiscLogicalOr.Inputs.class +) public final class RiscLogicalOr extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -40,8 +46,8 @@ public final class RiscLogicalOr extends RawOp implements Operand { private Output z; - private RiscLogicalOr(Operation operation) { - super(operation); + public RiscLogicalOr(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscLogicalOr.class + ) public static class Inputs extends RawOpInputs { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java index c5a73463a66..b8192bb2767 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code max} output */ +@OpMetadata( + opType = RiscMax.OP_NAME, + inputsClass = RiscMax.Inputs.class +) public final class RiscMax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RiscMax extends RawOp implements Operand max; - private RiscMax(Operation operation) { - super(operation); + public RiscMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; max = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return max; } + @OpInputsMetadata( + outputsClass = RiscMax.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java index 5d565b09d10..868ea20be57 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscMin.OP_NAME, + inputsClass = RiscMin.Inputs.class +) public final class RiscMin extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscMin extends RawOp implements Operand z; - private RiscMin(Operation operation) { - super(operation); + public RiscMin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscMin.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java index 64285c41c37..3ed0db57db9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscMul.OP_NAME, + inputsClass = RiscMul.Inputs.class +) public final class RiscMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscMul extends RawOp implements Operand z; - private RiscMul(Operation operation) { - super(operation); + public RiscMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscMul.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java index 9a759a04a6f..dca8cb7339b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscNeg.OP_NAME, + inputsClass = RiscNeg.Inputs.class +) public final class RiscNeg extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscNeg extends RawOp implements Operand y; - private RiscNeg(Operation operation) { - super(operation); + public RiscNeg(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscNeg.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java index 96d50ba86d5..9d3be8d9f94 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscPad.OP_NAME, + inputsClass = RiscPad.Inputs.class +) public final class RiscPad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscPad extends RawOp implements Operand output; - private RiscPad(Operation operation) { - super(operation); + public RiscPad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscPad.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java index 0ee16aa7da8..9e905d099a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscPool.OP_NAME, + inputsClass = RiscPool.Inputs.class +) public final class RiscPool extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscPool extends RawOp implements Operand< private Output output; - private RiscPool(Operation operation) { - super(operation); + public RiscPool(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -135,6 +141,9 @@ public Options dataFormat(String dataFormat) { } } + @OpInputsMetadata( + outputsClass = RiscPool.class + ) public static class Inputs extends RawOpInputs> { /** * The value input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java index 531ea2570e0..b92f2416109 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscPow.OP_NAME, + inputsClass = RiscPow.Inputs.class +) public final class RiscPow extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscPow extends RawOp implements Operand z; - private RiscPow(Operation operation) { - super(operation); + public RiscPow(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscPow.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java index b5b1c995a4e..314086dc12c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * The RiscRandomUniform operation */ +@OpMetadata( + opType = RiscRandomUniform.OP_NAME, + inputsClass = RiscRandomUniform.Inputs.class +) public final class RiscRandomUniform extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class RiscRandomUniform extends RawOp implements Operand private Output output; - private RiscRandomUniform(Operation operation) { - super(operation); + public RiscRandomUniform(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -118,6 +124,9 @@ public Options seed(Long seed) { } } + @OpInputsMetadata( + outputsClass = RiscRandomUniform.class + ) public static class Inputs extends RawOpInputs { /** * The shape input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java index 4c0e1b00708..a39747c841d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscReal.OP_NAME, + inputsClass = RiscReal.Inputs.class +) public final class RiscReal extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class RiscReal extends RawOp implements Operand< private Output output; - private RiscReal(Operation operation) { - super(operation); + public RiscReal(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscReal.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java index 68a4fb6e6cb..de63c91a2ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscReduce.OP_NAME, + inputsClass = RiscReduce.Inputs.class +) public final class RiscReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscReduce extends RawOp implements Operan private Output output; - private RiscReduce(Operation operation) { - super(operation); + public RiscReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscReduce.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java index a05a200a662..7e23a40f6ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscRem.OP_NAME, + inputsClass = RiscRem.Inputs.class +) public final class RiscRem extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscRem extends RawOp implements Operand z; - private RiscRem(Operation operation) { - super(operation); + public RiscRem(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscRem.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java index 1d31b746200..c4644eaba0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscReshape.OP_NAME, + inputsClass = RiscReshape.Inputs.class +) public final class RiscReshape extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscReshape extends RawOp implements Opera private Output output; - private RiscReshape(Operation operation) { - super(operation); + public RiscReshape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscReshape.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java index 6c63576cece..a0d5274b4c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscReverse.OP_NAME, + inputsClass = RiscReverse.Inputs.class +) public final class RiscReverse extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscReverse extends RawOp implements Opera private Output output; - private RiscReverse(Operation operation) { - super(operation); + public RiscReverse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -83,6 +89,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscReverse.class + ) public static class Inputs extends RawOpInputs> { /** * The tensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java index 9d4b7e75a5f..d3cff331777 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscScatter.OP_NAME, + inputsClass = RiscScatter.Inputs.class +) public final class RiscScatter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscScatter extends RawOp implements Opera private Output output; - private RiscScatter(Operation operation) { - super(operation); + public RiscScatter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscScatter.class + ) public static class Inputs extends RawOpInputs> { /** * The indices input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java index 48b15342aa8..885afc9637e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscShape.OP_NAME, + inputsClass = RiscShape.Inputs.class +) public final class RiscShape extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RiscShape extends RawOp implements Operand private Output output; - private RiscShape(Operation operation) { - super(operation); + public RiscShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscShape.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java index d9dc1645279..5fad88fcaee 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscSign.OP_NAME, + inputsClass = RiscSign.Inputs.class +) public final class RiscSign extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscSign extends RawOp implements Operand< private Output y; - private RiscSign(Operation operation) { - super(operation); + public RiscSign(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscSign.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java index 17c1ae867f8..4c1e1b32459 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscSlice.OP_NAME, + inputsClass = RiscSlice.Inputs.class +) public final class RiscSlice extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscSlice extends RawOp implements Operand private Output output; - private RiscSlice(Operation operation) { - super(operation); + public RiscSlice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -86,6 +92,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscSlice.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java index 05f439e84d7..5a47733f8e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscSort.OP_NAME, + inputsClass = RiscSort.Inputs.class +) public final class RiscSort extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscSort extends RawOp implements Operand< private Output output; - private RiscSort(Operation operation) { - super(operation); + public RiscSort(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RiscSort.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java index c10e67550b5..c88d61ada53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscSqueeze.OP_NAME, + inputsClass = RiscSqueeze.Inputs.class +) public final class RiscSqueeze extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscSqueeze extends RawOp implements Operand private Output output; - private RiscSqueeze(Operation operation) { - super(operation); + public RiscSqueeze(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options squeezeDims(Long... squeezeDims) { } } + @OpInputsMetadata( + outputsClass = RiscSqueeze.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java index 2da64a6b5fa..0377fff1210 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code z} output */ +@OpMetadata( + opType = RiscSub.OP_NAME, + inputsClass = RiscSub.Inputs.class +) public final class RiscSub extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscSub extends RawOp implements Operand z; - private RiscSub(Operation operation) { - super(operation); + public RiscSub(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; z = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return z; } + @OpInputsMetadata( + outputsClass = RiscSub.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java index 58d0eb662d5..bfa7c1feadc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscTranspose.OP_NAME, + inputsClass = RiscTranspose.Inputs.class +) public final class RiscTranspose extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RiscTranspose extends RawOp implements Opera private Output y; - private RiscTranspose(Operation operation) { - super(operation); + public RiscTranspose(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscTranspose.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java index cfd5976bb4c..a156d728712 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RiscTriangularSolve.OP_NAME, + inputsClass = RiscTriangularSolve.Inputs.class +) public final class RiscTriangularSolve extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscTriangularSolve extends RawOp implemen private Output output; - private RiscTriangularSolve(Operation operation) { - super(operation); + public RiscTriangularSolve(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -148,6 +154,9 @@ public Options adjoint(Boolean adjoint) { } } + @OpInputsMetadata( + outputsClass = RiscTriangularSolve.class + ) public static class Inputs extends RawOpInputs> { /** * The matrix input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java index 59b52dc8a17..46922a99b60 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -35,6 +37,10 @@ * * @param data type for {@code y} output */ +@OpMetadata( + opType = RiscUnary.OP_NAME, + inputsClass = RiscUnary.Inputs.class +) public final class RiscUnary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RiscUnary extends RawOp implements Operand private Output y; - private RiscUnary(Operation operation) { - super(operation); + public RiscUnary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; y = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return y; } + @OpInputsMetadata( + outputsClass = RiscUnary.class + ) public static class Inputs extends RawOpInputs> { /** * The x input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java index 010f239e8db..2e65c88f97f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java @@ -32,12 +32,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * The RiscWhile operation */ +@OpMetadata( + opType = RiscWhile.OP_NAME, + inputsClass = RiscWhile.Inputs.class +) public final class RiscWhile extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RiscWhile extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private RiscWhile(Operation operation) { - super(operation); + public RiscWhile(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -181,6 +187,9 @@ public Options parallelIterations(Long parallelIterations) { } } + @OpInputsMetadata( + outputsClass = RiscWhile.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft.java index 31fdd2bb931..c9948f82b85 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The BatchFFT operation */ +@OpMetadata( + opType = BatchFft.OP_NAME, + inputsClass = BatchFft.Inputs.class +) @Operator( group = "signal" ) @@ -45,8 +51,8 @@ public final class BatchFft extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private BatchFft(Operation operation) { - super(operation); + public BatchFft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = BatchFft.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft2d.java index 1b7fb596496..5416598cdbe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft2d.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The BatchFFT2D operation */ +@OpMetadata( + opType = BatchFft2d.OP_NAME, + inputsClass = BatchFft2d.Inputs.class +) @Operator( group = "signal" ) @@ -45,8 +51,8 @@ public final class BatchFft2d extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private BatchFft2d(Operation operation) { - super(operation); + public BatchFft2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = BatchFft2d.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft3d.java index 35c940d8abf..36969696370 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchFft3d.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The BatchFFT3D operation */ +@OpMetadata( + opType = BatchFft3d.OP_NAME, + inputsClass = BatchFft3d.Inputs.class +) @Operator( group = "signal" ) @@ -45,8 +51,8 @@ public final class BatchFft3d extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private BatchFft3d(Operation operation) { - super(operation); + public BatchFft3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = BatchFft3d.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft.java index 9fedc13692e..184b536fa50 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The BatchIFFT operation */ +@OpMetadata( + opType = BatchIfft.OP_NAME, + inputsClass = BatchIfft.Inputs.class +) @Operator( group = "signal" ) @@ -45,8 +51,8 @@ public final class BatchIfft extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private BatchIfft(Operation operation) { - super(operation); + public BatchIfft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = BatchIfft.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft2d.java index c7cf6cb1e96..b57ee7979bf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft2d.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The BatchIFFT2D operation */ +@OpMetadata( + opType = BatchIfft2d.OP_NAME, + inputsClass = BatchIfft2d.Inputs.class +) @Operator( group = "signal" ) @@ -45,8 +51,8 @@ public final class BatchIfft2d extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private BatchIfft2d(Operation operation) { - super(operation); + public BatchIfft2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = BatchIfft2d.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft3d.java index c23e16b79a3..ccc3327a78c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/BatchIfft3d.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** * The BatchIFFT3D operation */ +@OpMetadata( + opType = BatchIfft3d.OP_NAME, + inputsClass = BatchIfft3d.Inputs.class +) @Operator( group = "signal" ) @@ -45,8 +51,8 @@ public final class BatchIfft3d extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private BatchIfft3d(Operation operation) { - super(operation); + public BatchIfft3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return (Output) output; } + @OpInputsMetadata( + outputsClass = BatchIfft3d.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft.java index bf800846caa..571b3424a92 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Fft.OP_NAME, + inputsClass = Fft.Inputs.class +) @Operator( group = "signal" ) @@ -49,8 +55,8 @@ public final class Fft extends RawOp implements Operand { private Output output; - private Fft(Operation operation) { - super(operation); + public Fft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Fft.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft2d.java index 097ea3cf43d..a7edbbd2580 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft2d.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Fft2d.OP_NAME, + inputsClass = Fft2d.Inputs.class +) @Operator( group = "signal" ) @@ -49,8 +55,8 @@ public final class Fft2d extends RawOp implements Operand { private Output output; - private Fft2d(Operation operation) { - super(operation); + public Fft2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Fft2d.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft3d.java index 7fcf6354abe..a42664be517 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Fft3d.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Fft3d.OP_NAME, + inputsClass = Fft3d.Inputs.class +) @Operator( group = "signal" ) @@ -49,8 +55,8 @@ public final class Fft3d extends RawOp implements Operand { private Output output; - private Fft3d(Operation operation) { - super(operation); + public Fft3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Fft3d.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft.java index 787c57575df..2e9718d758e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Ifft.OP_NAME, + inputsClass = Ifft.Inputs.class +) @Operator( group = "signal" ) @@ -49,8 +55,8 @@ public final class Ifft extends RawOp implements Operand { private Output output; - private Ifft(Operation operation) { - super(operation); + public Ifft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Ifft.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft2d.java index fc3e42c7d18..99db6dbd9e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft2d.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Ifft2d.OP_NAME, + inputsClass = Ifft2d.Inputs.class +) @Operator( group = "signal" ) @@ -49,8 +55,8 @@ public final class Ifft2d extends RawOp implements Operand { private Output output; - private Ifft2d(Operation operation) { - super(operation); + public Ifft2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Ifft2d.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft3d.java index 9d9910f20b6..edb8810e3c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Ifft3d.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Ifft3d.OP_NAME, + inputsClass = Ifft3d.Inputs.class +) @Operator( group = "signal" ) @@ -49,8 +55,8 @@ public final class Ifft3d extends RawOp implements Operand { private Output output; - private Ifft3d(Operation operation) { - super(operation); + public Ifft3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Ifft3d.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft.java index bf619275e89..718337402bf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -51,6 +53,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Irfft.OP_NAME, + inputsClass = Irfft.Inputs.class +) @Operator( group = "signal" ) @@ -62,8 +68,8 @@ public final class Irfft extends RawOp implements Operand private Output output; - private Irfft(Operation operation) { - super(operation); + public Irfft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -125,6 +131,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Irfft.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft2d.java index 76ef5401d06..1b7e31b2ab5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft2d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -52,6 +54,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Irfft2d.OP_NAME, + inputsClass = Irfft2d.Inputs.class +) @Operator( group = "signal" ) @@ -63,8 +69,8 @@ public final class Irfft2d extends RawOp implements Operand output; - private Irfft2d(Operation operation) { - super(operation); + public Irfft2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -126,6 +132,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Irfft2d.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft3d.java index 951d90c24f5..d989c8f610a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Irfft3d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -52,6 +54,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Irfft3d.OP_NAME, + inputsClass = Irfft3d.Inputs.class +) @Operator( group = "signal" ) @@ -63,8 +69,8 @@ public final class Irfft3d extends RawOp implements Operand output; - private Irfft3d(Operation operation) { - super(operation); + public Irfft3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -126,6 +132,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Irfft3d.class + ) public static class Inputs extends RawOpInputs> { /** * A complex tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft.java index aaecb285947..3f47556ad2c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -47,6 +49,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Rfft.OP_NAME, + inputsClass = Rfft.Inputs.class +) @Operator( group = "signal" ) @@ -58,8 +64,8 @@ public final class Rfft extends RawOp implements Operand { private Output output; - private Rfft(Operation operation) { - super(operation); + public Rfft(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Rfft.class + ) public static class Inputs extends RawOpInputs> { /** * A float32 tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft2d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft2d.java index 1fc337c853f..959564338a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft2d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft2d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Rfft2d.OP_NAME, + inputsClass = Rfft2d.Inputs.class +) @Operator( group = "signal" ) @@ -59,8 +65,8 @@ public final class Rfft2d extends RawOp implements Operand { private Output output; - private Rfft2d(Operation operation) { - super(operation); + public Rfft2d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Rfft2d.class + ) public static class Inputs extends RawOpInputs> { /** * A float32 tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft3d.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft3d.java index e10cc8b9879..9203d09409e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft3d.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/signal/Rfft3d.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Rfft3d.OP_NAME, + inputsClass = Rfft3d.Inputs.class +) @Operator( group = "signal" ) @@ -59,8 +65,8 @@ public final class Rfft3d extends RawOp implements Operand { private Output output; - private Rfft3d(Operation operation) { - super(operation); + public Rfft3d(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -107,6 +113,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Rfft3d.class + ) public static class Inputs extends RawOpInputs> { /** * A float32 tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddManySparseToTensorsMap.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddManySparseToTensorsMap.java index 7757774ace2..5aa97de3122 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddManySparseToTensorsMap.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddManySparseToTensorsMap.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -53,6 +55,10 @@ * {@code sparse.AddManySparseToTensorsMap} as the {@code shared_name} passed to * {@code TakeManySparseFromTensorsMap}. Ensure the Operations are colocated. */ +@OpMetadata( + opType = AddManySparseToTensorsMap.OP_NAME, + inputsClass = AddManySparseToTensorsMap.Inputs.class +) @Operator( group = "sparse" ) @@ -64,8 +70,8 @@ public final class AddManySparseToTensorsMap extends RawOp implements Operand sparseHandles; - private AddManySparseToTensorsMap(Operation operation) { - super(operation); + public AddManySparseToTensorsMap(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseHandles = operation.output(outputIdx++); } @@ -175,6 +181,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = AddManySparseToTensorsMap.class + ) public static class Inputs extends RawOpInputs { /** * 2-D. The {@code indices} of the minibatch {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddSparseToTensorsMap.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddSparseToTensorsMap.java index 8a68e68e4ca..0737e08cc4a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddSparseToTensorsMap.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/AddSparseToTensorsMap.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -47,6 +49,10 @@ * {@code sparse.AddSparseToTensorsMap} as the {@code shared_name} passed to * {@code TakeManySparseFromTensorsMap}. Ensure the Operations are colocated. */ +@OpMetadata( + opType = AddSparseToTensorsMap.OP_NAME, + inputsClass = AddSparseToTensorsMap.Inputs.class +) @Operator( group = "sparse" ) @@ -58,8 +64,8 @@ public final class AddSparseToTensorsMap extends RawOp implements Operand sparseHandle; - private AddSparseToTensorsMap(Operation operation) { - super(operation); + public AddSparseToTensorsMap(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseHandle = operation.output(outputIdx++); } @@ -167,6 +173,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = AddSparseToTensorsMap.class + ) public static class Inputs extends RawOpInputs { /** * 2-D. The {@code indices} of the {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java index a83f853daac..281998ef065 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = DenseCountSparseOutput.OP_NAME, + inputsClass = DenseCountSparseOutput.Inputs.class +) public final class DenseCountSparseOutput extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class DenseCountSparseOutput extends RawOp { private Output outputDenseShape; - private DenseCountSparseOutput(Operation operation) { - super(operation); + public DenseCountSparseOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -173,6 +179,9 @@ public Options maxlength(Long maxlength) { } } + @OpInputsMetadata( + outputsClass = DenseCountSparseOutput.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor containing data to count. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToDenseSetOperation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToDenseSetOperation.java index ea896eac86f..f2b77c66329 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToDenseSetOperation.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToDenseSetOperation.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -43,6 +45,10 @@ * * @param data type for {@code result_values} output */ +@OpMetadata( + opType = DenseToDenseSetOperation.OP_NAME, + inputsClass = DenseToDenseSetOperation.Inputs.class +) @Operator( group = "sparse" ) @@ -58,8 +64,8 @@ public final class DenseToDenseSetOperation extends RawOp { private Output resultShape; - private DenseToDenseSetOperation(Operation operation) { - super(operation); + public DenseToDenseSetOperation(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resultIndices = operation.output(outputIdx++); resultValues = operation.output(outputIdx++); @@ -158,6 +164,9 @@ public Options validateIndices(Boolean validateIndices) { } } + @OpInputsMetadata( + outputsClass = DenseToDenseSetOperation.class + ) public static class Inputs extends RawOpInputs> { /** * {@code Tensor} with rank {@code n}. 1st {@code n-1} dimensions must be the same as {@code set2}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToSparseSetOperation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToSparseSetOperation.java index 6cf1c374394..7552f756671 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToSparseSetOperation.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseToSparseSetOperation.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code result_values} output */ +@OpMetadata( + opType = DenseToSparseSetOperation.OP_NAME, + inputsClass = DenseToSparseSetOperation.Inputs.class +) @Operator( group = "sparse" ) @@ -64,8 +70,8 @@ public final class DenseToSparseSetOperation extends RawOp { private Output resultShape; - private DenseToSparseSetOperation(Operation operation) { - super(operation); + public DenseToSparseSetOperation(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resultIndices = operation.output(outputIdx++); resultValues = operation.output(outputIdx++); @@ -172,6 +178,9 @@ public Options validateIndices(Boolean validateIndices) { } } + @OpInputsMetadata( + outputsClass = DenseToSparseSetOperation.class + ) public static class Inputs extends RawOpInputs> { /** * {@code Tensor} with rank {@code n}. 1st {@code n-1} dimensions must be the same as {@code set2}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DeserializeSparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DeserializeSparse.java index 837e8db112b..86daa749a4e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DeserializeSparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DeserializeSparse.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -77,6 +79,10 @@ * * @param data type for {@code sparse_values} output */ +@OpMetadata( + opType = DeserializeSparse.OP_NAME, + inputsClass = DeserializeSparse.Inputs.class +) @Operator( group = "sparse" ) @@ -92,8 +98,8 @@ public final class DeserializeSparse extends RawOp { private Output sparseShape; - private DeserializeSparse(Operation operation) { - super(operation); + public DeserializeSparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseIndices = operation.output(outputIdx++); sparseValues = operation.output(outputIdx++); @@ -148,6 +154,9 @@ public Output sparseShape() { return sparseShape; } + @OpInputsMetadata( + outputsClass = DeserializeSparse.class + ) public static class Inputs extends RawOpInputs> { /** * The serialized {@code SparseTensor} objects. The last dimension diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorApplyGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorApplyGradient.java index 8496ad6a614..ce71474efcf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorApplyGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorApplyGradient.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -37,6 +39,10 @@ * Does not add if local_step is smaller than the accumulator's * global_step. */ +@OpMetadata( + opType = SparseAccumulatorApplyGradient.OP_NAME, + inputsClass = SparseAccumulatorApplyGradient.Inputs.class +) @Operator( group = "sparse" ) @@ -46,8 +52,8 @@ public final class SparseAccumulatorApplyGradient extends RawOp { */ public static final String OP_NAME = "SparseAccumulatorApplyGradient"; - private SparseAccumulatorApplyGradient(Operation operation) { - super(operation); + public SparseAccumulatorApplyGradient(Operation operation) { + super(operation, OP_NAME); } /** @@ -83,6 +89,9 @@ public static SparseAccumulatorApplyGradient create(Scope scope, Operand { /** * The handle to a accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorTakeGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorTakeGradient.java index f1558dd83d4..0ed16f6b4bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorTakeGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAccumulatorTakeGradient.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -46,6 +48,10 @@ * * @param data type for {@code values} output */ +@OpMetadata( + opType = SparseAccumulatorTakeGradient.OP_NAME, + inputsClass = SparseAccumulatorTakeGradient.Inputs.class +) @Operator( group = "sparse" ) @@ -61,8 +67,8 @@ public final class SparseAccumulatorTakeGradient extends RawOp private Output shape; - private SparseAccumulatorTakeGradient(Operation operation) { - super(operation); + public SparseAccumulatorTakeGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; indices = operation.output(outputIdx++); values = operation.output(outputIdx++); @@ -119,6 +125,9 @@ public Output shape() { return shape; } + @OpInputsMetadata( + outputsClass = SparseAccumulatorTakeGradient.class + ) public static class Inputs extends RawOpInputs> { /** * The handle to a SparseConditionalAccumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAdd.java index 8e372cacf9b..ceb5a3160f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code sum_values} output */ +@OpMetadata( + opType = SparseAdd.OP_NAME, + inputsClass = SparseAdd.Inputs.class +) @Operator( group = "sparse" ) @@ -64,8 +70,8 @@ public final class SparseAdd extends RawOp { private Output sumShape; - private SparseAdd(Operation operation) { - super(operation); + public SparseAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sumIndices = operation.output(outputIdx++); sumValues = operation.output(outputIdx++); @@ -131,6 +137,9 @@ public Output sumShape() { return sumShape; } + @OpInputsMetadata( + outputsClass = SparseAdd.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. The {@code indices} of the first {@code SparseTensor}, size {@code [nnz, ndims]} Matrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAddGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAddGrad.java index c99dd5a7a08..6bffeaf384b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAddGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseAddGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -41,6 +43,10 @@ * * @param data type for {@code a_val_grad} output */ +@OpMetadata( + opType = SparseAddGrad.OP_NAME, + inputsClass = SparseAddGrad.Inputs.class +) @Operator( group = "sparse" ) @@ -54,8 +60,8 @@ public final class SparseAddGrad extends RawOp { private Output bValGrad; - private SparseAddGrad(Operation operation) { - super(operation); + public SparseAddGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; aValGrad = operation.output(outputIdx++); bValGrad = operation.output(outputIdx++); @@ -107,6 +113,9 @@ public Output bValGrad() { return bValGrad; } + @OpInputsMetadata( + outputsClass = SparseAddGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D with shape {@code [nnz(sum)]}. The gradient with respect to diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseBincount.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseBincount.java index 6064e201455..d978960f4e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseBincount.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseBincount.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -43,6 +45,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseBincount.OP_NAME, + inputsClass = SparseBincount.Inputs.class +) @Operator( group = "sparse" ) @@ -54,8 +60,8 @@ public final class SparseBincount extends RawOp implements Op private Output output; - private SparseBincount(Operation operation) { - super(operation); + public SparseBincount(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Options binaryOutput(Boolean binaryOutput) { } } + @OpInputsMetadata( + outputsClass = SparseBincount.class + ) public static class Inputs extends RawOpInputs> { /** * 2D int64 {@code Tensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConcat.java index d35c0837991..81bd3c2b84c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConcat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConcat.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -75,6 +77,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseConcat.OP_NAME, + inputsClass = SparseConcat.Inputs.class +) @Operator( group = "sparse" ) @@ -90,8 +96,8 @@ public final class SparseConcat extends RawOp { private Output outputShape; - private SparseConcat(Operation operation) { - super(operation); + public SparseConcat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -151,6 +157,9 @@ public Output outputShape() { return outputShape; } + @OpInputsMetadata( + outputsClass = SparseConcat.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. Indices of each input {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConditionalAccumulator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConditionalAccumulator.java index 7ddc90749d9..8c36581baff 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConditionalAccumulator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseConditionalAccumulator.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -43,6 +45,10 @@ * resets the aggregate to 0, and increments the global_step recorded by * the accumulator. */ +@OpMetadata( + opType = SparseConditionalAccumulator.OP_NAME, + inputsClass = SparseConditionalAccumulator.Inputs.class +) @Operator( group = "sparse" ) @@ -54,8 +60,8 @@ public final class SparseConditionalAccumulator extends RawOp implements Operand private Output handle; - private SparseConditionalAccumulator(Operation operation) { - super(operation); + public SparseConditionalAccumulator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -189,6 +195,9 @@ public Options reductionType(String reductionType) { } } + @OpInputsMetadata( + outputsClass = SparseConditionalAccumulator.class + ) public static class Inputs extends RawOpInputs { /** * The type of the value being accumulated. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java index bee78d51b23..6c2bfdac1dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseCountSparseOutput.OP_NAME, + inputsClass = SparseCountSparseOutput.Inputs.class +) public final class SparseCountSparseOutput extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class SparseCountSparseOutput extends RawOp { private Output outputDenseShape; - private SparseCountSparseOutput(Operation operation) { - super(operation); + public SparseCountSparseOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -177,6 +183,9 @@ public Options maxlength(Long maxlength) { } } + @OpInputsMetadata( + outputsClass = SparseCountSparseOutput.class + ) public static class Inputs extends RawOpInputs> { /** * Tensor containing the indices of the sparse tensor to count. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCross.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCross.java index 0208d01c554..de00b98597b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCross.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCross.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -72,6 +74,10 @@ * Fingerprint64("e"), Fingerprint64("c"))) * */ +@OpMetadata( + opType = SparseCross.OP_NAME, + inputsClass = SparseCross.Inputs.class +) @Operator( group = "sparse" ) @@ -87,8 +93,8 @@ public final class SparseCross extends RawOp { private Output outputShape; - private SparseCross(Operation operation) { - super(operation); + public SparseCross(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -149,6 +155,9 @@ public Output outputShape() { return outputShape; } + @OpInputsMetadata( + outputsClass = SparseCross.class + ) public static class Inputs extends RawOpInputs { /** * 2-D. Indices of each input {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCrossHashed.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCrossHashed.java index af73abe7e8d..e2af02a02a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCrossHashed.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCrossHashed.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -72,6 +74,10 @@ * Fingerprint64("e"), Fingerprint64("c"))) * */ +@OpMetadata( + opType = SparseCrossHashed.OP_NAME, + inputsClass = SparseCrossHashed.Inputs.class +) @Operator( group = "sparse" ) @@ -87,8 +93,8 @@ public final class SparseCrossHashed extends RawOp { private Output outputShape; - private SparseCrossHashed(Operation operation) { - super(operation); + public SparseCrossHashed(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -155,6 +161,9 @@ public Output outputShape() { return outputShape; } + @OpInputsMetadata( + outputsClass = SparseCrossHashed.class + ) public static class Inputs extends RawOpInputs { /** * 2-D. Indices of each input {@code SparseTensor}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseAdd.java index 9f2e841c0a1..1e50002d9b0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseDenseCwiseAdd.OP_NAME, + inputsClass = SparseDenseCwiseAdd.Inputs.class +) @Operator( group = "sparse" ) @@ -55,8 +61,8 @@ public final class SparseDenseCwiseAdd extends RawOp implements private Output output; - private SparseDenseCwiseAdd(Operation operation) { - super(operation); + public SparseDenseCwiseAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseDenseCwiseAdd.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseDiv.java index 9b519b392ca..42b1734610f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseDiv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseDiv.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseDenseCwiseDiv.OP_NAME, + inputsClass = SparseDenseCwiseDiv.Inputs.class +) @Operator( group = "sparse" ) @@ -50,8 +56,8 @@ public final class SparseDenseCwiseDiv extends RawOp implements private Output output; - private SparseDenseCwiseDiv(Operation operation) { - super(operation); + public SparseDenseCwiseDiv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseDenseCwiseDiv.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseMul.java index f316405eb7f..2a0e0b3507b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseDenseCwiseMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseDenseCwiseMul.OP_NAME, + inputsClass = SparseDenseCwiseMul.Inputs.class +) @Operator( group = "sparse" ) @@ -53,8 +59,8 @@ public final class SparseDenseCwiseMul extends RawOp implements private Output output; - private SparseDenseCwiseMul(Operation operation) { - super(operation); + public SparseDenseCwiseMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseDenseCwiseMul.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRows.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRows.java index 6473358aad6..206c829d54a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRows.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRows.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -72,6 +74,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseFillEmptyRows.OP_NAME, + inputsClass = SparseFillEmptyRows.Inputs.class +) @Operator( group = "sparse" ) @@ -89,8 +95,8 @@ public final class SparseFillEmptyRows extends RawOp { private Output reverseIndexMap; - private SparseFillEmptyRows(Operation operation) { - super(operation); + public SparseFillEmptyRows(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -162,6 +168,9 @@ public Output reverseIndexMap() { return reverseIndexMap; } + @OpInputsMetadata( + outputsClass = SparseFillEmptyRows.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. the indices of the sparse tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRowsGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRowsGrad.java index 94f5033d574..8ee85db13ef 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRowsGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseFillEmptyRowsGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -44,6 +46,10 @@ * * @param data type for {@code d_values} output */ +@OpMetadata( + opType = SparseFillEmptyRowsGrad.OP_NAME, + inputsClass = SparseFillEmptyRowsGrad.Inputs.class +) @Operator( group = "sparse" ) @@ -57,8 +63,8 @@ public final class SparseFillEmptyRowsGrad extends RawOp { private Output dDefaultValue; - private SparseFillEmptyRowsGrad(Operation operation) { - super(operation); + public SparseFillEmptyRowsGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; dValues = operation.output(outputIdx++); dDefaultValue = operation.output(outputIdx++); @@ -102,6 +108,9 @@ public Output dDefaultValue() { return dDefaultValue; } + @OpInputsMetadata( + outputsClass = SparseFillEmptyRowsGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. The reverse index map from SparseFillEmptyRows. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseMatMul.java index 8c9388a0aaa..0c4f8a13b52 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseMatMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -43,6 +45,10 @@ *

    The gradient computation of this operation will only take advantage of sparsity * in the input gradient when that gradient comes from a Relu. */ +@OpMetadata( + opType = SparseMatMul.OP_NAME, + inputsClass = SparseMatMul.Inputs.class +) @Operator( group = "sparse" ) @@ -54,8 +60,8 @@ public final class SparseMatMul extends RawOp implements Operand { private Output product; - private SparseMatMul(Operation operation) { - super(operation); + public SparseMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; product = operation.output(outputIdx++); } @@ -210,6 +216,9 @@ public Options bIsSparse(Boolean bIsSparse) { } } + @OpInputsMetadata( + outputsClass = SparseMatMul.class + ) public static class Inputs extends RawOpInputs { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMax.java index 989e3f5c48a..ce1ee1affd3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseReduceMax.OP_NAME, + inputsClass = SparseReduceMax.Inputs.class +) @Operator( group = "sparse" ) @@ -59,8 +65,8 @@ public final class SparseReduceMax extends RawOp implements O private Output output; - private SparseReduceMax(Operation operation) { - super(operation); + public SparseReduceMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = SparseReduceMax.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMaxSparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMaxSparse.java index 5ec12e79fa4..6728db570bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMaxSparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceMaxSparse.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseReduceMaxSparse.OP_NAME, + inputsClass = SparseReduceMaxSparse.Inputs.class +) @Operator( group = "sparse" ) @@ -63,8 +69,8 @@ public final class SparseReduceMaxSparse extends RawOp { private Output outputShape; - private SparseReduceMaxSparse(Operation operation) { - super(operation); + public SparseReduceMaxSparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -163,6 +169,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = SparseReduceMaxSparse.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSum.java index f81a5eff0cf..fca1832c967 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseReduceSum.OP_NAME, + inputsClass = SparseReduceSum.Inputs.class +) @Operator( group = "sparse" ) @@ -59,8 +65,8 @@ public final class SparseReduceSum extends RawOp implements Ope private Output output; - private SparseReduceSum(Operation operation) { - super(operation); + public SparseReduceSum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = SparseReduceSum.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSumSparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSumSparse.java index e97e512582a..01cee87946a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSumSparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReduceSumSparse.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -48,6 +50,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseReduceSumSparse.OP_NAME, + inputsClass = SparseReduceSumSparse.Inputs.class +) @Operator( group = "sparse" ) @@ -63,8 +69,8 @@ public final class SparseReduceSumSparse extends RawOp { private Output outputShape; - private SparseReduceSumSparse(Operation operation) { - super(operation); + public SparseReduceSumSparse(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -163,6 +169,9 @@ public Options keepDims(Boolean keepDims) { } } + @OpInputsMetadata( + outputsClass = SparseReduceSumSparse.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReorder.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReorder.java index 4f3ef7a762a..99e095e2123 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReorder.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReorder.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -43,6 +45,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseReorder.OP_NAME, + inputsClass = SparseReorder.Inputs.class +) @Operator( group = "sparse" ) @@ -56,8 +62,8 @@ public final class SparseReorder extends RawOp { private Output outputValues; - private SparseReorder(Operation operation) { - super(operation); + public SparseReorder(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -105,6 +111,9 @@ public Output outputValues() { return outputValues; } + @OpInputsMetadata( + outputsClass = SparseReorder.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReshape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReshape.java index 5b569b8ea6c..671461d6c96 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReshape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseReshape.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; @@ -45,6 +47,10 @@ * {@code input_shape} has length {@code R_in}, {@code output_indices} has shape {@code [N, R_out]}, and * {@code output_shape} has length {@code R_out}. */ +@OpMetadata( + opType = SparseReshape.OP_NAME, + inputsClass = SparseReshape.Inputs.class +) @Operator( group = "sparse" ) @@ -58,8 +64,8 @@ public final class SparseReshape extends RawOp { private Output outputShape; - private SparseReshape(Operation operation) { - super(operation); + public SparseReshape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputShape = operation.output(outputIdx++); @@ -108,6 +114,9 @@ public Output outputShape() { return outputShape; } + @OpInputsMetadata( + outputsClass = SparseReshape.class + ) public static class Inputs extends RawOpInputs { /** * 2-D. {@code N x R_in} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMean.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMean.java index af9f65b4551..301de791181 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMean.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMean.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentMean.OP_NAME, + inputsClass = SparseSegmentMean.Inputs.class +) @Operator( group = "sparse" ) @@ -50,8 +56,8 @@ public final class SparseSegmentMean extends RawOp implements private Output output; - private SparseSegmentMean(Operation operation) { - super(operation); + public SparseSegmentMean(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentMean.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanGrad.java index 80c3acc5da1..e0ec592149a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentMeanGrad.OP_NAME, + inputsClass = SparseSegmentMeanGrad.Inputs.class +) @Operator( group = "sparse" ) @@ -50,8 +56,8 @@ public final class SparseSegmentMeanGrad extends RawOp implem private Output output; - private SparseSegmentMeanGrad(Operation operation) { - super(operation); + public SparseSegmentMeanGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentMeanGrad.class + ) public static class Inputs extends RawOpInputs> { /** * gradient propagated to the SparseSegmentMean op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanWithNumSegments.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanWithNumSegments.java index 8fe4f039b62..4e254db04bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanWithNumSegments.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentMeanWithNumSegments.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentMeanWithNumSegments.OP_NAME, + inputsClass = SparseSegmentMeanWithNumSegments.Inputs.class +) @Operator( group = "sparse" ) @@ -52,8 +58,8 @@ public final class SparseSegmentMeanWithNumSegments extends R private Output output; - private SparseSegmentMeanWithNumSegments(Operation operation) { - super(operation); + public SparseSegmentMeanWithNumSegments(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentMeanWithNumSegments.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtN.java index f5b0a967137..4632ad9c8ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtN.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentSqrtN.OP_NAME, + inputsClass = SparseSegmentSqrtN.Inputs.class +) @Operator( group = "sparse" ) @@ -49,8 +55,8 @@ public final class SparseSegmentSqrtN extends RawOp implement private Output output; - private SparseSegmentSqrtN(Operation operation) { - super(operation); + public SparseSegmentSqrtN(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentSqrtN.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNGrad.java index 2b5d8e00ae1..244e659d8f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentSqrtNGrad.OP_NAME, + inputsClass = SparseSegmentSqrtNGrad.Inputs.class +) @Operator( group = "sparse" ) @@ -50,8 +56,8 @@ public final class SparseSegmentSqrtNGrad extends RawOp imple private Output output; - private SparseSegmentSqrtNGrad(Operation operation) { - super(operation); + public SparseSegmentSqrtNGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentSqrtNGrad.class + ) public static class Inputs extends RawOpInputs> { /** * gradient propagated to the SparseSegmentSqrtN op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNWithNumSegments.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNWithNumSegments.java index 090732ab6e2..eaa499e8611 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNWithNumSegments.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSqrtNWithNumSegments.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentSqrtNWithNumSegments.OP_NAME, + inputsClass = SparseSegmentSqrtNWithNumSegments.Inputs.class +) @Operator( group = "sparse" ) @@ -53,8 +59,8 @@ public final class SparseSegmentSqrtNWithNumSegments extends private Output output; - private SparseSegmentSqrtNWithNumSegments(Operation operation) { - super(operation); + public SparseSegmentSqrtNWithNumSegments(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentSqrtNWithNumSegments.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSum.java index 7c78dfd8392..cd66f0acde3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -62,6 +64,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentSum.OP_NAME, + inputsClass = SparseSegmentSum.Inputs.class +) @Operator( group = "sparse" ) @@ -73,8 +79,8 @@ public final class SparseSegmentSum extends RawOp implements private Output output; - private SparseSegmentSum(Operation operation) { - super(operation); + public SparseSegmentSum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -116,6 +122,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentSum.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumGrad.java index 3689868f892..2ce26681cc3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentSumGrad.OP_NAME, + inputsClass = SparseSegmentSumGrad.Inputs.class +) @Operator( group = "sparse" ) @@ -50,8 +56,8 @@ public final class SparseSegmentSumGrad extends RawOp impleme private Output output; - private SparseSegmentSumGrad(Operation operation) { - super(operation); + public SparseSegmentSumGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentSumGrad.class + ) public static class Inputs extends RawOpInputs> { /** * gradient propagated to the SparseSegmentSum op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumWithNumSegments.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumWithNumSegments.java index 45c84027eec..8f9aa9d583c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumWithNumSegments.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSegmentSumWithNumSegments.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -60,6 +62,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSegmentSumWithNumSegments.OP_NAME, + inputsClass = SparseSegmentSumWithNumSegments.Inputs.class +) @Operator( group = "sparse" ) @@ -71,8 +77,8 @@ public final class SparseSegmentSumWithNumSegments extends Ra private Output output; - private SparseSegmentSumWithNumSegments(Operation operation) { - super(operation); + public SparseSegmentSumWithNumSegments(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -117,6 +123,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSegmentSumWithNumSegments.class + ) public static class Inputs extends RawOpInputs> { /** * The data input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSlice.java index 881d26ef94c..c03d46a6507 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSlice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSlice.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -53,6 +55,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseSlice.OP_NAME, + inputsClass = SparseSlice.Inputs.class +) @Operator( group = "sparse" ) @@ -68,8 +74,8 @@ public final class SparseSlice extends RawOp { private Output outputShape; - private SparseSlice(Operation operation) { - super(operation); + public SparseSlice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -133,6 +139,9 @@ public Output outputShape() { return outputShape; } + @OpInputsMetadata( + outputsClass = SparseSlice.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D tensor represents the indices of the sparse tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSliceGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSliceGrad.java index 0899e873513..d3686ba4ce9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSliceGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSliceGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -40,6 +42,10 @@ * * @param data type for {@code val_grad} output */ +@OpMetadata( + opType = SparseSliceGrad.OP_NAME, + inputsClass = SparseSliceGrad.Inputs.class +) @Operator( group = "sparse" ) @@ -51,8 +57,8 @@ public final class SparseSliceGrad extends RawOp implements Ope private Output valGrad; - private SparseSliceGrad(Operation operation) { - super(operation); + public SparseSliceGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; valGrad = operation.output(outputIdx++); } @@ -96,6 +102,9 @@ public Output asOutput() { return valGrad; } + @OpInputsMetadata( + outputsClass = SparseSliceGrad.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D. The gradient with respect to diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSoftmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSoftmax.java index 0cc01eeee58..d0318d8905b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSoftmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSoftmax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -49,6 +51,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseSoftmax.OP_NAME, + inputsClass = SparseSoftmax.Inputs.class +) @Operator( group = "sparse" ) @@ -60,8 +66,8 @@ public final class SparseSoftmax extends RawOp implements Ope private Output output; - private SparseSoftmax(Operation operation) { - super(operation); + public SparseSoftmax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -103,6 +109,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseSoftmax.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code NNZ x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMaximum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMaximum.java index 6a36379bd8d..12f6247d4f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMaximum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMaximum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseSparseMaximum.OP_NAME, + inputsClass = SparseSparseMaximum.Inputs.class +) @Operator( group = "sparse" ) @@ -51,8 +57,8 @@ public final class SparseSparseMaximum extends RawOp { private Output outputValues; - private SparseSparseMaximum(Operation operation) { - super(operation); + public SparseSparseMaximum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -106,6 +112,9 @@ public Output outputValues() { return outputValues; } + @OpInputsMetadata( + outputsClass = SparseSparseMaximum.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMinimum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMinimum.java index 6a32e085cee..25a0e3e77f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMinimum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSparseMinimum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseSparseMinimum.OP_NAME, + inputsClass = SparseSparseMinimum.Inputs.class +) @Operator( group = "sparse" ) @@ -51,8 +57,8 @@ public final class SparseSparseMinimum extends RawOp { private Output outputValues; - private SparseSparseMinimum(Operation operation) { - super(operation); + public SparseSparseMinimum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputIndices = operation.output(outputIdx++); outputValues = operation.output(outputIdx++); @@ -106,6 +112,9 @@ public Output outputValues() { return outputValues; } + @OpInputsMetadata( + outputsClass = SparseSparseMinimum.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. {@code N x R} matrix with the indices of non-empty values in a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSplit.java index 0b830e4441f..2b5a5944fff 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSplit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseSplit.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -56,6 +58,10 @@ * * @param data type for {@code output_values} output */ +@OpMetadata( + opType = SparseSplit.OP_NAME, + inputsClass = SparseSplit.Inputs.class +) @Operator( group = "sparse" ) @@ -72,8 +78,8 @@ public final class SparseSplit extends RawOp { private List> outputShape; @SuppressWarnings("unchecked") - private SparseSplit(Operation operation) { - super(operation); + public SparseSplit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputIndicesLength = operation.outputListLength("output_indices"); outputIndices = Arrays.asList((Output[]) operation.outputList(outputIdx, outputIndicesLength)); @@ -144,6 +150,9 @@ public List> outputShape() { return outputShape; } + @OpInputsMetadata( + outputsClass = SparseSplit.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D. The dimension along which to split. Must be in the range diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseAdd.java index e828e6476fb..08a1b4ca726 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseAdd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SparseTensorDenseAdd.OP_NAME, + inputsClass = SparseTensorDenseAdd.Inputs.class +) @Operator( group = "sparse" ) @@ -49,8 +55,8 @@ public final class SparseTensorDenseAdd extends RawOp implement private Output output; - private SparseTensorDenseAdd(Operation operation) { - super(operation); + public SparseTensorDenseAdd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SparseTensorDenseAdd.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. The {@code indices} of the {@code SparseTensor}, with shape {@code [nnz, ndims]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseMatMul.java index 559b82d90c5..552aabfa24e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseTensorDenseMatMul.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -46,6 +48,10 @@ * * @param data type for {@code product} output */ +@OpMetadata( + opType = SparseTensorDenseMatMul.OP_NAME, + inputsClass = SparseTensorDenseMatMul.Inputs.class +) @Operator( group = "sparse" ) @@ -57,8 +63,8 @@ public final class SparseTensorDenseMatMul extends RawOp implem private Output product; - private SparseTensorDenseMatMul(Operation operation) { - super(operation); + public SparseTensorDenseMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; product = operation.output(outputIdx++); } @@ -171,6 +177,9 @@ public Options adjointB(Boolean adjointB) { } } + @OpInputsMetadata( + outputsClass = SparseTensorDenseMatMul.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D. The {@code indices} of the {@code SparseTensor}, size {@code [nnz, 2]} Matrix. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToDense.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToDense.java index dac1f7569b9..bb2c0f1c955 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToDense.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToDense.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -53,6 +55,10 @@ * * @param data type for {@code dense} output */ +@OpMetadata( + opType = SparseToDense.OP_NAME, + inputsClass = SparseToDense.Inputs.class +) @Operator( group = "sparse" ) @@ -64,8 +70,8 @@ public final class SparseToDense extends RawOp implements Opera private Output dense; - private SparseToDense(Operation operation) { - super(operation); + public SparseToDense(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; dense = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options validateIndices(Boolean validateIndices) { } } + @OpInputsMetadata( + outputsClass = SparseToDense.class + ) public static class Inputs extends RawOpInputs> { /** * 0-D, 1-D, or 2-D. {@code sparse_indices[i]} contains the complete diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToSparseSetOperation.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToSparseSetOperation.java index 1d4ea083ad6..179cd8d70ef 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToSparseSetOperation.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseToSparseSetOperation.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -55,6 +57,10 @@ * * @param data type for {@code result_values} output */ +@OpMetadata( + opType = SparseToSparseSetOperation.OP_NAME, + inputsClass = SparseToSparseSetOperation.Inputs.class +) @Operator( group = "sparse" ) @@ -70,8 +76,8 @@ public final class SparseToSparseSetOperation extends RawOp { private Output resultShape; - private SparseToSparseSetOperation(Operation operation) { - super(operation); + public SparseToSparseSetOperation(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; resultIndices = operation.output(outputIdx++); resultValues = operation.output(outputIdx++); @@ -186,6 +192,9 @@ public Options validateIndices(Boolean validateIndices) { } } + @OpInputsMetadata( + outputsClass = SparseToSparseSetOperation.class + ) public static class Inputs extends RawOpInputs> { /** * 2D {@code Tensor}, indices of a {@code SparseTensor}. Must be in row-major diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/TakeManySparseFromTensorsMap.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/TakeManySparseFromTensorsMap.java index eaec58d9901..2481608003d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/TakeManySparseFromTensorsMap.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/TakeManySparseFromTensorsMap.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -78,6 +80,10 @@ * * @param data type for {@code sparse_values} output */ +@OpMetadata( + opType = TakeManySparseFromTensorsMap.OP_NAME, + inputsClass = TakeManySparseFromTensorsMap.Inputs.class +) @Operator( group = "sparse" ) @@ -93,8 +99,8 @@ public final class TakeManySparseFromTensorsMap extends RawOp { private Output sparseShape; - private TakeManySparseFromTensorsMap(Operation operation) { - super(operation); + public TakeManySparseFromTensorsMap(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sparseIndices = operation.output(outputIdx++); sparseValues = operation.output(outputIdx++); @@ -219,6 +225,9 @@ public Options sharedName(String sharedName) { } } + @OpInputsMetadata( + outputsClass = TakeManySparseFromTensorsMap.class + ) public static class Inputs extends RawOpInputs> { /** * 1-D, The {@code N} serialized {@code SparseTensor} objects. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Join.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Join.java index 6985e0283c7..6705ae83853 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Join.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Join.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -45,6 +47,10 @@ * * */ +@OpMetadata( + opType = Join.OP_NAME, + inputsClass = Join.Inputs.class +) @Operator( group = "strings" ) @@ -56,8 +62,8 @@ public final class Join extends RawOp implements Operand { private Output output; - private Join(Operation operation) { - super(operation); + public Join(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -133,6 +139,9 @@ public Options separator(String separator) { } } + @OpInputsMetadata( + outputsClass = Join.class + ) public static class Inputs extends RawOpInputs { /** * A list of string tensors. The tensors must all have the same shape, diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Lower.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Lower.java index 83f55b3ef0f..f14268de14a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Lower.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Lower.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -42,6 +44,10 @@ * * */ +@OpMetadata( + opType = Lower.OP_NAME, + inputsClass = Lower.Inputs.class +) @Operator( group = "strings" ) @@ -53,8 +59,8 @@ public final class Lower extends RawOp implements Operand { private Output output; - private Lower(Operation operation) { - super(operation); + public Lower(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options encoding(String encoding) { } } + @OpInputsMetadata( + outputsClass = Lower.class + ) public static class Inputs extends RawOpInputs { /** * The input to be lower-cased. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ReduceJoin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ReduceJoin.java index 2e9e709d68c..b33f61736ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ReduceJoin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ReduceJoin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -55,6 +57,10 @@ * tf.reduce_join(a) = tf.reduce_join(a, [1, 0]) ==> "abcd" * */ +@OpMetadata( + opType = ReduceJoin.OP_NAME, + inputsClass = ReduceJoin.Inputs.class +) @Operator( group = "strings" ) @@ -66,8 +72,8 @@ public final class ReduceJoin extends RawOp implements Operand { private Output output; - private ReduceJoin(Operation operation) { - super(operation); + public ReduceJoin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -173,6 +179,9 @@ public Options separator(String separator) { } } + @OpInputsMetadata( + outputsClass = ReduceJoin.class + ) public static class Inputs extends RawOpInputs { /** * The input to be joined. All reduced indices must have non-zero size. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexFullMatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexFullMatch.java index 31157d96864..6d6333859c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexFullMatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexFullMatch.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; import org.tensorflow.types.TString; @@ -50,6 +52,10 @@ * * */ +@OpMetadata( + opType = RegexFullMatch.OP_NAME, + inputsClass = RegexFullMatch.Inputs.class +) @Operator( group = "strings" ) @@ -61,8 +67,8 @@ public final class RegexFullMatch extends RawOp implements Operand { private Output output; - private RegexFullMatch(Operation operation) { - super(operation); + public RegexFullMatch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -100,6 +106,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RegexFullMatch.class + ) public static class Inputs extends RawOpInputs { /** * A string tensor of the text to be processed. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexReplace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexReplace.java index ca767d27f1e..1e26c1d06db 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexReplace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/RegexReplace.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ * replacement string provided in {@code rewrite}. * It follows the re2 syntax (https://github.com/google/re2/wiki/Syntax) */ +@OpMetadata( + opType = RegexReplace.OP_NAME, + inputsClass = RegexReplace.Inputs.class +) @Operator( group = "strings" ) @@ -46,8 +52,8 @@ public final class RegexReplace extends RawOp implements Operand { private Output output; - private RegexReplace(Operation operation) { - super(operation); + public RegexReplace(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -131,6 +137,9 @@ public Options replaceGlobal(Boolean replaceGlobal) { } } + @OpInputsMetadata( + outputsClass = RegexReplace.class + ) public static class Inputs extends RawOpInputs { /** * The text to be processed. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java index d101be0d46a..a45599ddc4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TBool; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * if the input matches the regex pattern provided. *

    The pattern follows the re2 syntax (https://github.com/google/re2/wiki/Syntax) */ +@OpMetadata( + opType = StaticRegexFullMatch.OP_NAME, + inputsClass = StaticRegexFullMatch.Inputs.class +) public final class StaticRegexFullMatch extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class StaticRegexFullMatch extends RawOp implements Operand private Output output; - private StaticRegexFullMatch(Operation operation) { - super(operation); + public StaticRegexFullMatch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = StaticRegexFullMatch.class + ) public static class Inputs extends RawOpInputs { /** * A string tensor of the text to be processed. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java index fdbb0c3e4da..a72312ffe18 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; /** * Replaces the match of pattern in input with rewrite. * It follows the re2 syntax (https://github.com/google/re2/wiki/Syntax) */ +@OpMetadata( + opType = StaticRegexReplace.OP_NAME, + inputsClass = StaticRegexReplace.Inputs.class +) public final class StaticRegexReplace extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class StaticRegexReplace extends RawOp implements Operand private Output output; - private StaticRegexReplace(Operation operation) { - super(operation); + public StaticRegexReplace(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -123,6 +129,9 @@ public Options replaceGlobal(Boolean replaceGlobal) { } } + @OpInputsMetadata( + outputsClass = StaticRegexReplace.class + ) public static class Inputs extends RawOpInputs { /** * The text to be processed. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringFormat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringFormat.java index a7becde73ec..806cdb57184 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringFormat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringFormat.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -36,6 +38,10 @@ * Formats a string template using a list of tensors. * Formats a string template using a list of tensors, pretty-printing tensor summaries. */ +@OpMetadata( + opType = StringFormat.OP_NAME, + inputsClass = StringFormat.Inputs.class +) @Operator( group = "strings" ) @@ -47,8 +53,8 @@ public final class StringFormat extends RawOp implements Operand { private Output output; - private StringFormat(Operation operation) { - super(operation); + public StringFormat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -174,6 +180,9 @@ public Options summarize(Long summarize) { } } + @OpInputsMetadata( + outputsClass = StringFormat.class + ) public static class Inputs extends RawOpInputs { /** * The list of tensors to format into the placeholder string. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringLength.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringLength.java index 98aa6dbec32..65f27f3f431 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringLength.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringLength.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -46,6 +48,10 @@ * * */ +@OpMetadata( + opType = StringLength.OP_NAME, + inputsClass = StringLength.Inputs.class +) @Operator( group = "strings" ) @@ -57,8 +63,8 @@ public final class StringLength extends RawOp implements Operand { private Output output; - private StringLength(Operation operation) { - super(operation); + public StringLength(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options unit(String unit) { } } + @OpInputsMetadata( + outputsClass = StringLength.class + ) public static class Inputs extends RawOpInputs { /** * The strings for which to compute the length for each element. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringNGrams.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringNGrams.java index 49d00162650..fe203cd8415 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringNGrams.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringNGrams.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -41,6 +43,10 @@ * * @param data type for {@code ngrams_splits} output */ +@OpMetadata( + opType = StringNGrams.OP_NAME, + inputsClass = StringNGrams.Inputs.class +) @Operator( group = "strings" ) @@ -54,8 +60,8 @@ public final class StringNGrams extends RawOp { private Output ngramsSplits; - private StringNGrams(Operation operation) { - super(operation); + public StringNGrams(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; ngrams = operation.output(outputIdx++); ngramsSplits = operation.output(outputIdx++); @@ -122,6 +128,9 @@ public Output ngramsSplits() { return ngramsSplits; } + @OpInputsMetadata( + outputsClass = StringNGrams.class + ) public static class Inputs extends RawOpInputs> { /** * The values tensor of the ragged string tensor to make ngrams out of. Must be a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringSplit.java index 499cc5269c0..41c6a344a41 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringSplit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StringSplit.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -55,6 +57,10 @@ * leading or trailing whitespace. *

    Note that the above mentioned behavior matches python's str.split. */ +@OpMetadata( + opType = StringSplit.OP_NAME, + inputsClass = StringSplit.Inputs.class +) @Operator( group = "strings" ) @@ -70,8 +76,8 @@ public final class StringSplit extends RawOp { private Output shape; - private StringSplit(Operation operation) { - super(operation); + public StringSplit(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; indices = operation.output(outputIdx++); values = operation.output(outputIdx++); @@ -163,6 +169,9 @@ public Options maxsplit(Long maxsplit) { } } + @OpInputsMetadata( + outputsClass = StringSplit.class + ) public static class Inputs extends RawOpInputs { /** * {@code 1-D} string {@code Tensor}, the strings to split. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Strip.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Strip.java index b5aefbb65ae..fcb125dbca9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Strip.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Strip.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -42,6 +44,10 @@ * * */ +@OpMetadata( + opType = Strip.OP_NAME, + inputsClass = Strip.Inputs.class +) @Operator( group = "strings" ) @@ -53,8 +59,8 @@ public final class Strip extends RawOp implements Operand { private Output output; - private Strip(Operation operation) { - super(operation); + public Strip(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Strip.class + ) public static class Inputs extends RawOpInputs { /** * A string {@code Tensor} of any shape. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Substr.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Substr.java index acfebc4fe95..a8f3f7cc178 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Substr.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Substr.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -102,6 +104,10 @@ *

  • {@code ValueError}: If {@code pos} and {@code len} are not the same shape.
  • * */ +@OpMetadata( + opType = Substr.OP_NAME, + inputsClass = Substr.Inputs.class +) @Operator( group = "strings" ) @@ -113,8 +119,8 @@ public final class Substr extends RawOp implements Operand { private Output output; - private Substr(Operation operation) { - super(operation); + public Substr(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -202,6 +208,9 @@ public Options unit(String unit) { } } + @OpInputsMetadata( + outputsClass = Substr.class + ) public static class Inputs extends RawOpInputs { /** * Tensor of strings diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucket.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucket.java index f801507c1ad..8dec93b94cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucket.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucket.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ * This functionality will be deprecated and it's recommended to use * {@code tf.string_to_hash_bucket_fast()} or {@code tf.string_to_hash_bucket_strong()}. */ +@OpMetadata( + opType = ToHashBucket.OP_NAME, + inputsClass = ToHashBucket.Inputs.class +) @Operator( group = "strings" ) @@ -50,8 +56,8 @@ public final class ToHashBucket extends RawOp implements Operand { private Output output; - private ToHashBucket(Operation operation) { - super(operation); + public ToHashBucket(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ToHashBucket.class + ) public static class Inputs extends RawOpInputs { /** * The stringTensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketFast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketFast.java index 39cdf8a5405..a35fe637b42 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketFast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketFast.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -49,6 +51,10 @@ * * */ +@OpMetadata( + opType = ToHashBucketFast.OP_NAME, + inputsClass = ToHashBucketFast.Inputs.class +) @Operator( group = "strings" ) @@ -60,8 +66,8 @@ public final class ToHashBucketFast extends RawOp implements Operand { private Output output; - private ToHashBucketFast(Operation operation) { - super(operation); + public ToHashBucketFast(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ToHashBucketFast.class + ) public static class Inputs extends RawOpInputs { /** * The strings to assign a hash bucket. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketStrong.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketStrong.java index f9ca9b8115f..641e5d41e87 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketStrong.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToHashBucketStrong.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -55,6 +57,10 @@ * * */ +@OpMetadata( + opType = ToHashBucketStrong.OP_NAME, + inputsClass = ToHashBucketStrong.Inputs.class +) @Operator( group = "strings" ) @@ -66,8 +72,8 @@ public final class ToHashBucketStrong extends RawOp implements Operand { private Output output; - private ToHashBucketStrong(Operation operation) { - super(operation); + public ToHashBucketStrong(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -112,6 +118,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ToHashBucketStrong.class + ) public static class Inputs extends RawOpInputs { /** * The strings to assign a hash bucket. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToNumber.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToNumber.java index f8f325282fc..3f58cbadd07 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToNumber.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/ToNumber.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TFloat32; @@ -51,6 +53,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ToNumber.OP_NAME, + inputsClass = ToNumber.Inputs.class +) @Operator( group = "strings" ) @@ -62,8 +68,8 @@ public final class ToNumber extends RawOp implements Operand< private Output output; - private ToNumber(Operation operation) { - super(operation); + public ToNumber(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -116,6 +122,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ToNumber.class + ) public static class Inputs extends RawOpInputs> { /** * The stringTensor input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java index a130514dc54..9929ce78a97 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -52,6 +54,10 @@ * * @param data type for {@code row_splits} output */ +@OpMetadata( + opType = UnicodeDecode.OP_NAME, + inputsClass = UnicodeDecode.Inputs.class +) public final class UnicodeDecode extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -62,8 +68,8 @@ public final class UnicodeDecode extends RawOp { private Output charValues; - private UnicodeDecode(Operation operation) { - super(operation); + public UnicodeDecode(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; rowSplits = operation.output(outputIdx++); charValues = operation.output(outputIdx++); @@ -241,6 +247,9 @@ public Options replaceControlCharacters(Boolean replaceControlCharacters) { } } + @OpInputsMetadata( + outputsClass = UnicodeDecode.class + ) public static class Inputs extends RawOpInputs> { /** * The text to be decoded. Can have any shape. Note that the output is flattened diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java index c3467e760b0..a186bea2a79 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -56,6 +58,10 @@ * * @param data type for {@code row_splits} output */ +@OpMetadata( + opType = UnicodeDecodeWithOffsets.OP_NAME, + inputsClass = UnicodeDecodeWithOffsets.Inputs.class +) public final class UnicodeDecodeWithOffsets extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -68,8 +74,8 @@ public final class UnicodeDecodeWithOffsets extends RawOp { private Output charToByteStarts; - private UnicodeDecodeWithOffsets(Operation operation) { - super(operation); + public UnicodeDecodeWithOffsets(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; rowSplits = operation.output(outputIdx++); charValues = operation.output(outputIdx++); @@ -258,6 +264,9 @@ public Options replaceControlCharacters(Boolean replaceControlCharacters) { } } + @OpInputsMetadata( + outputsClass = UnicodeDecodeWithOffsets.class + ) public static class Inputs extends RawOpInputs> { /** * The text to be decoded. Can have any shape. Note that the output is flattened diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java index ffa29a2ddb4..a14c69eea1b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -47,6 +49,10 @@ * output = ['Hello', 'World'] * */ +@OpMetadata( + opType = UnicodeEncode.OP_NAME, + inputsClass = UnicodeEncode.Inputs.class +) public final class UnicodeEncode extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -55,8 +61,8 @@ public final class UnicodeEncode extends RawOp implements Operand { private Output output; - private UnicodeEncode(Operation operation) { - super(operation); + public UnicodeEncode(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -181,6 +187,9 @@ public Options replacementChar(Long replacementChar) { } } + @OpInputsMetadata( + outputsClass = UnicodeEncode.class + ) public static class Inputs extends RawOpInputs { /** * A 1D tensor containing the unicode codepoints that should be encoded. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeScript.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeScript.java index ce046508f36..7d63d69a16a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeScript.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeScript.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; @@ -52,6 +54,10 @@ * * */ +@OpMetadata( + opType = UnicodeScript.OP_NAME, + inputsClass = UnicodeScript.Inputs.class +) @Operator( group = "strings" ) @@ -63,8 +69,8 @@ public final class UnicodeScript extends RawOp implements Operand { private Output output; - private UnicodeScript(Operation operation) { - super(operation); + public UnicodeScript(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -99,6 +105,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = UnicodeScript.class + ) public static class Inputs extends RawOpInputs { /** * A Tensor of int32 Unicode code points. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeTranscode.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeTranscode.java index 90194f8b23a..a433dddf82f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeTranscode.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeTranscode.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -68,6 +70,10 @@ * * */ +@OpMetadata( + opType = UnicodeTranscode.OP_NAME, + inputsClass = UnicodeTranscode.Inputs.class +) @Operator( group = "strings" ) @@ -79,8 +85,8 @@ public final class UnicodeTranscode extends RawOp implements Operand { private Output output; - private UnicodeTranscode(Operation operation) { - super(operation); + public UnicodeTranscode(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -241,6 +247,9 @@ public Options replaceControlCharacters(Boolean replaceControlCharacters) { } } + @OpInputsMetadata( + outputsClass = UnicodeTranscode.class + ) public static class Inputs extends RawOpInputs { /** * The text to be processed. Can have any shape. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java index 58b1eb8ac3b..9d3e0484d58 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -59,6 +61,10 @@ * # output_array ==> ['this:is:a:test'] * */ +@OpMetadata( + opType = UnsortedSegmentJoin.OP_NAME, + inputsClass = UnsortedSegmentJoin.Inputs.class +) @Operator( group = "strings" ) @@ -70,8 +76,8 @@ public final class UnsortedSegmentJoin extends RawOp implements Operand private Output output; - private UnsortedSegmentJoin(Operation operation) { - super(operation); + public UnsortedSegmentJoin(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -152,6 +158,9 @@ public Options separator(String separator) { } } + @OpInputsMetadata( + outputsClass = UnsortedSegmentJoin.class + ) public static class Inputs extends RawOpInputs { /** * The input to be joined. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Upper.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Upper.java index 7ba066da30c..84378ddac55 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Upper.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/Upper.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -42,6 +44,10 @@ * * */ +@OpMetadata( + opType = Upper.OP_NAME, + inputsClass = Upper.Inputs.class +) @Operator( group = "strings" ) @@ -53,8 +59,8 @@ public final class Upper extends RawOp implements Operand { private Output output; - private Upper(Operation operation) { - super(operation); + public Upper(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options encoding(String encoding) { } } + @OpInputsMetadata( + outputsClass = Upper.class + ) public static class Inputs extends RawOpInputs { /** * The input to be upper-cased. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/AudioSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/AudioSummary.java index 9a78e8b1bfc..a27777b8eab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/AudioSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/AudioSummary.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TString; @@ -44,6 +46,10 @@ * generated sequentially as 'tag/audio/0', 'tag/audio/1', etc. * */ +@OpMetadata( + opType = AudioSummary.OP_NAME, + inputsClass = AudioSummary.Inputs.class +) @Operator( group = "summary" ) @@ -55,8 +61,8 @@ public final class AudioSummary extends RawOp implements Operand { private Output summary; - private AudioSummary(Operation operation) { - super(operation); + public AudioSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -135,6 +141,9 @@ public Options maxOutputs(Long maxOutputs) { } } + @OpInputsMetadata( + outputsClass = AudioSummary.class + ) public static class Inputs extends RawOpInputs { /** * Scalar. Used to build the {@code tag} attribute of the summary values. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java index 58b8dcef495..3cd27230947 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The CloseSummaryWriter operation */ +@OpMetadata( + opType = CloseSummaryWriter.OP_NAME, + inputsClass = CloseSummaryWriter.Inputs.class +) public final class CloseSummaryWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "CloseSummaryWriter"; - private CloseSummaryWriter(Operation operation) { - super(operation); + public CloseSummaryWriter(Operation operation) { + super(operation, OP_NAME); } /** @@ -57,6 +63,9 @@ public static CloseSummaryWriter create(Scope scope, Operand wr return new CloseSummaryWriter(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = CloseSummaryWriter.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java index 8af194417b1..3195ee604cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java @@ -26,20 +26,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * The CreateSummaryDbWriter operation */ +@OpMetadata( + opType = CreateSummaryDbWriter.OP_NAME, + inputsClass = CreateSummaryDbWriter.Inputs.class +) public final class CreateSummaryDbWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "CreateSummaryDbWriter"; - private CreateSummaryDbWriter(Operation operation) { - super(operation); + public CreateSummaryDbWriter(Operation operation) { + super(operation, OP_NAME); } /** @@ -68,6 +74,9 @@ public static CreateSummaryDbWriter create(Scope scope, Operand return new CreateSummaryDbWriter(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = CreateSummaryDbWriter.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java index aecb55a68f7..af2bc01a83c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -33,14 +35,18 @@ /** * The CreateSummaryFileWriter operation */ +@OpMetadata( + opType = CreateSummaryFileWriter.OP_NAME, + inputsClass = CreateSummaryFileWriter.Inputs.class +) public final class CreateSummaryFileWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "CreateSummaryFileWriter"; - private CreateSummaryFileWriter(Operation operation) { - super(operation); + public CreateSummaryFileWriter(Operation operation) { + super(operation, OP_NAME); } /** @@ -69,6 +75,9 @@ public static CreateSummaryFileWriter create(Scope scope, Operand { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java index 44f22343c23..3da6797c325 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The FlushSummaryWriter operation */ +@OpMetadata( + opType = FlushSummaryWriter.OP_NAME, + inputsClass = FlushSummaryWriter.Inputs.class +) public final class FlushSummaryWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "FlushSummaryWriter"; - private FlushSummaryWriter(Operation operation) { - super(operation); + public FlushSummaryWriter(Operation operation) { + super(operation, OP_NAME); } /** @@ -57,6 +63,9 @@ public static FlushSummaryWriter create(Scope scope, Operand wr return new FlushSummaryWriter(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = FlushSummaryWriter.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/HistogramSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/HistogramSummary.java index 6e05d278830..7762807eb96 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/HistogramSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/HistogramSummary.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ * has one summary value containing a histogram for {@code values}. *

    This op reports an {@code InvalidArgument} error if any value is not finite. */ +@OpMetadata( + opType = HistogramSummary.OP_NAME, + inputsClass = HistogramSummary.Inputs.class +) @Operator( group = "summary" ) @@ -50,8 +56,8 @@ public final class HistogramSummary extends RawOp implements Operand { private Output summary; - private HistogramSummary(Operation operation) { - super(operation); + public HistogramSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return summary; } + @OpInputsMetadata( + outputsClass = HistogramSummary.class + ) public static class Inputs extends RawOpInputs { /** * Scalar. Tag to use for the {@code Summary.Value}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImageSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImageSummary.java index ff83df181ad..6bd4dcf1989 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImageSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImageSummary.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -71,6 +73,10 @@ * replaced by this tensor in the output image. The default value is the color * red. */ +@OpMetadata( + opType = ImageSummary.OP_NAME, + inputsClass = ImageSummary.Inputs.class +) @Operator( group = "summary" ) @@ -82,8 +88,8 @@ public final class ImageSummary extends RawOp implements Operand { private Output summary; - private ImageSummary(Operation operation) { - super(operation); + public ImageSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -187,6 +193,9 @@ public Options badColor(Tensor badColor) { } } + @OpInputsMetadata( + outputsClass = ImageSummary.class + ) public static class Inputs extends RawOpInputs { /** * Scalar. Used to build the {@code tag} attribute of the summary values. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java index 0150201909b..5953f0f0bbd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java @@ -26,20 +26,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * The ImportEvent operation */ +@OpMetadata( + opType = ImportEvent.OP_NAME, + inputsClass = ImportEvent.Inputs.class +) public final class ImportEvent extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ImportEvent"; - private ImportEvent(Operation operation) { - super(operation); + public ImportEvent(Operation operation) { + super(operation, OP_NAME); } /** @@ -61,6 +67,9 @@ public static ImportEvent create(Scope scope, Operand writer, return new ImportEvent(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ImportEvent.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/MergeSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/MergeSummary.java index 643d36d2894..5437e3399d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/MergeSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/MergeSummary.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -40,6 +42,10 @@ *

    When the Op is run, it reports an {@code InvalidArgument} error if multiple values * in the summaries to merge use the same tag. */ +@OpMetadata( + opType = MergeSummary.OP_NAME, + inputsClass = MergeSummary.Inputs.class +) @Operator( group = "summary" ) @@ -51,8 +57,8 @@ public final class MergeSummary extends RawOp implements Operand { private Output summary; - private MergeSummary(Operation operation) { - super(operation); + public MergeSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return summary; } + @OpInputsMetadata( + outputsClass = MergeSummary.class + ) public static class Inputs extends RawOpInputs { /** * Can be of any shape. Each must contain serialized {@code Summary} protocol diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ScalarSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ScalarSummary.java index c459ba8413a..259972a5e49 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ScalarSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ScalarSummary.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -37,6 +39,10 @@ * The input {@code tags} and {@code values} must have the same shape. The generated summary * has a summary value for each tag-value pair in {@code tags} and {@code values}. */ +@OpMetadata( + opType = ScalarSummary.OP_NAME, + inputsClass = ScalarSummary.Inputs.class +) @Operator( group = "summary" ) @@ -48,8 +54,8 @@ public final class ScalarSummary extends RawOp implements Operand { private Output summary; - private ScalarSummary(Operation operation) { - super(operation); + public ScalarSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return summary; } + @OpInputsMetadata( + outputsClass = ScalarSummary.class + ) public static class Inputs extends RawOpInputs { /** * Tags for the summary. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java index d1eeb17ac2f..a074ab65484 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; /** * Produces a summary of any statistics recorded by the given statistics manager. */ +@OpMetadata( + opType = StatsAggregatorSummary.OP_NAME, + inputsClass = StatsAggregatorSummary.Inputs.class +) public final class StatsAggregatorSummary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class StatsAggregatorSummary extends RawOp implements Operand summary; - private StatsAggregatorSummary(Operation operation) { - super(operation); + public StatsAggregatorSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return summary; } + @OpInputsMetadata( + outputsClass = StatsAggregatorSummary.class + ) public static class Inputs extends RawOpInputs { /** * The iterator input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java index c3a9d913d76..2e4af55ec82 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * The SummaryWriter operation */ +@OpMetadata( + opType = SummaryWriter.OP_NAME, + inputsClass = SummaryWriter.Inputs.class +) public final class SummaryWriter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class SummaryWriter extends RawOp implements Operand { private Output writer; @SuppressWarnings("unchecked") - private SummaryWriter(Operation operation) { - super(operation); + public SummaryWriter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; writer = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options container(String container) { } } + @OpInputsMetadata( + outputsClass = SummaryWriter.class + ) public static class Inputs extends RawOpInputs { /** * The sharedName attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/TensorSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/TensorSummary.java index ccd1dcb3c67..d828e584e31 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/TensorSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/TensorSummary.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ /** * Outputs a {@code Summary} protocol buffer with a tensor and per-plugin data. */ +@OpMetadata( + opType = TensorSummary.OP_NAME, + inputsClass = TensorSummary.Inputs.class +) @Operator( group = "summary" ) @@ -46,8 +52,8 @@ public final class TensorSummary extends RawOp implements Operand { private Output summary; - private TensorSummary(Operation operation) { - super(operation); + public TensorSummary(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; summary = operation.output(outputIdx++); } @@ -88,6 +94,9 @@ public Output asOutput() { return summary; } + @OpInputsMetadata( + outputsClass = TensorSummary.class + ) public static class Inputs extends RawOpInputs { /** * A string attached to this summary. Used for organization in TensorBoard. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java index 03c11c27c74..7b00a14229b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -36,14 +38,18 @@ * Writes encoded audio summary {@code tensor} at {@code step} with {@code tag} using summary {@code writer}. * {@code sample_rate} is the audio sample rate is Hz. */ +@OpMetadata( + opType = WriteAudioSummary.OP_NAME, + inputsClass = WriteAudioSummary.Inputs.class +) public final class WriteAudioSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteAudioSummary"; - private WriteAudioSummary(Operation operation) { - super(operation); + public WriteAudioSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -111,6 +117,9 @@ public Options maxOutputs(Long maxOutputs) { } } + @OpInputsMetadata( + outputsClass = WriteAudioSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java index 791f2e53025..811f79441b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * Writes a graph summary. * Writes TensorFlow graph {@code tensor} at {@code step} using summary {@code writer}. */ +@OpMetadata( + opType = WriteGraphSummary.OP_NAME, + inputsClass = WriteGraphSummary.Inputs.class +) public final class WriteGraphSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteGraphSummary"; - private WriteGraphSummary(Operation operation) { - super(operation); + public WriteGraphSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static WriteGraphSummary create(Scope scope, Operand wri return new WriteGraphSummary(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = WriteGraphSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java index 93f3c00c93e..24fd3c8eda9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,14 +37,18 @@ * Writes a histogram summary. * Writes histogram {@code values} at {@code step} with {@code tag} using summary {@code writer}. */ +@OpMetadata( + opType = WriteHistogramSummary.OP_NAME, + inputsClass = WriteHistogramSummary.Inputs.class +) public final class WriteHistogramSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteHistogramSummary"; - private WriteHistogramSummary(Operation operation) { - super(operation); + public WriteHistogramSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -68,6 +74,9 @@ public static WriteHistogramSummary create(Scope scope, Operand return new WriteHistogramSummary(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = WriteHistogramSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java index 9b1b64ae148..a2437bfed6e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -38,14 +40,18 @@ * Writes image {@code tensor} at {@code step} with {@code tag} using summary {@code writer}. * {@code tensor} is image with shape [height, width, channels]. */ +@OpMetadata( + opType = WriteImageSummary.OP_NAME, + inputsClass = WriteImageSummary.Inputs.class +) public final class WriteImageSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteImageSummary"; - private WriteImageSummary(Operation operation) { - super(operation); + public WriteImageSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -113,6 +119,9 @@ public Options maxImages(Long maxImages) { } } + @OpInputsMetadata( + outputsClass = WriteImageSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java index 66b92490f04..3633dd822ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * Writes a serialized proto summary. * Writes {@code tensor}, a serialized proto at {@code step} using summary {@code writer}. */ +@OpMetadata( + opType = WriteRawProtoSummary.OP_NAME, + inputsClass = WriteRawProtoSummary.Inputs.class +) public final class WriteRawProtoSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteRawProtoSummary"; - private WriteRawProtoSummary(Operation operation) { - super(operation); + public WriteRawProtoSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -65,6 +71,9 @@ public static WriteRawProtoSummary create(Scope scope, Operand return new WriteRawProtoSummary(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = WriteRawProtoSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java index 3edd2898fd6..7412f19c50c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -36,14 +38,18 @@ * Writes a scalar summary. * Writes scalar {@code value} at {@code step} with {@code tag} using summary {@code writer}. */ +@OpMetadata( + opType = WriteScalarSummary.OP_NAME, + inputsClass = WriteScalarSummary.Inputs.class +) public final class WriteScalarSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteScalarSummary"; - private WriteScalarSummary(Operation operation) { - super(operation); + public WriteScalarSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -69,6 +75,9 @@ public static WriteScalarSummary create(Scope scope, Operand wr return new WriteScalarSummary(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = WriteScalarSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java index 0c1f7822242..fd2b37c3b8b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,14 +37,18 @@ * Writes a tensor summary. * Writes {@code tensor} at {@code step} with {@code tag} using summary {@code writer}. */ +@OpMetadata( + opType = WriteSummary.OP_NAME, + inputsClass = WriteSummary.Inputs.class +) public final class WriteSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "WriteSummary"; - private WriteSummary(Operation operation) { - super(operation); + public WriteSummary(Operation operation) { + super(operation, OP_NAME); } /** @@ -71,6 +77,9 @@ public static WriteSummary create(Scope scope, Operand writer, return new WriteSummary(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = WriteSummary.class + ) public static class Inputs extends RawOpInputs { /** * The writer input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java index f5cf0420941..37a0862c547 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -49,6 +51,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = AllToAll.OP_NAME, + inputsClass = AllToAll.Inputs.class +) public final class AllToAll extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -57,8 +63,8 @@ public final class AllToAll extends RawOp implements Operand private Output output; - private AllToAll(Operation operation) { - super(operation); + public AllToAll(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -106,6 +112,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = AllToAll.class + ) public static class Inputs extends RawOpInputs> { /** * The local input to the sum. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CollectivePermute.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CollectivePermute.java index 0741956dcc7..c98049a5299 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CollectivePermute.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CollectivePermute.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CollectivePermute.OP_NAME, + inputsClass = CollectivePermute.Inputs.class +) public final class CollectivePermute extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class CollectivePermute extends RawOp implements O private Output output; - private CollectivePermute(Operation operation) { - super(operation); + public CollectivePermute(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -89,6 +95,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = CollectivePermute.class + ) public static class Inputs extends RawOpInputs> { /** * The local input to be permuted. Currently only supports float and diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java index 450c390d6e2..a18e76ed0a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; /** @@ -35,6 +37,10 @@ * CompilationResultProto, which holds a status and an error message if an error * occurred during compilation. */ +@OpMetadata( + opType = CompilationResult.OP_NAME, + inputsClass = CompilationResult.Inputs.class +) public final class CompilationResult extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class CompilationResult extends RawOp implements Operand { private Output output; - private CompilationResult(Operation operation) { - super(operation); + public CompilationResult(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = CompilationResult.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new CompilationResult(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Compile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Compile.java index ecf0ae4cc3f..f45cc4f701a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Compile.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Compile.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TBool; @@ -53,6 +55,10 @@ * used to look up the program in the compilation cache. * 'may_modify_variables' indicates whether variables may be modified. */ +@OpMetadata( + opType = Compile.OP_NAME, + inputsClass = Compile.Inputs.class +) @Operator( group = "tpu" ) @@ -69,8 +75,8 @@ public final class Compile extends RawOp { private List> mayModifyVariables; @SuppressWarnings("unchecked") - private Compile(Operation operation) { - super(operation); + public Compile(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; compilationStatus = operation.output(outputIdx++); int programLength = operation.outputListLength("program"); @@ -134,6 +140,9 @@ public List> mayModifyVariables() { return mayModifyVariables; } + @OpInputsMetadata( + outputsClass = Compile.class + ) public static class Inputs extends RawOpInputs { /** * The dynamicShapes input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompileSucceededAssert.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompileSucceededAssert.java index d4a2b599e91..53b64ef5482 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompileSucceededAssert.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompileSucceededAssert.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ * pending device interactions fail. *

    'compilation_status' is a serialized CompilationResultProto. */ +@OpMetadata( + opType = CompileSucceededAssert.OP_NAME, + inputsClass = CompileSucceededAssert.Inputs.class +) @Operator( group = "tpu" ) @@ -44,8 +50,8 @@ public final class CompileSucceededAssert extends RawOp { */ public static final String OP_NAME = "TPUCompileSucceededAssert"; - private CompileSucceededAssert(Operation operation) { - super(operation); + public CompileSucceededAssert(Operation operation) { + super(operation, OP_NAME); } /** @@ -64,6 +70,9 @@ public static CompileSucceededAssert create(Scope scope, Operand compil return new CompileSucceededAssert(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = CompileSucceededAssert.class + ) public static class Inputs extends RawOpInputs { /** * The compilationStatus input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java index 8c951410670..6cc6432a70d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java @@ -27,11 +27,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; /** * Sets up the centralized structures for a distributed TPU system. */ +@OpMetadata( + opType = ConfigureDistributedTPU.OP_NAME, + inputsClass = ConfigureDistributedTPU.Inputs.class +) public final class ConfigureDistributedTPU extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -40,8 +46,8 @@ public final class ConfigureDistributedTPU extends RawOp implements Operand topology; - private ConfigureDistributedTPU(Operation operation) { - super(operation); + public ConfigureDistributedTPU(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; topology = operation.output(outputIdx++); } @@ -220,6 +226,9 @@ public Options compilationFailureClosesChips(Boolean compilationFailureClosesChi } } + @OpInputsMetadata( + outputsClass = ConfigureDistributedTPU.class + ) public static class Inputs extends RawOpInputs { /** * Reserved. Do not use. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java index 029bcc9974a..fc8b8be7b96 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java @@ -25,18 +25,24 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; /** * Sets up TPUEmbedding in a distributed TPU system. */ +@OpMetadata( + opType = ConfigureTPUEmbedding.OP_NAME, + inputsClass = ConfigureTPUEmbedding.Inputs.class +) public final class ConfigureTPUEmbedding extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ConfigureTPUEmbedding"; - private ConfigureTPUEmbedding(Operation operation) { - super(operation); + public ConfigureTPUEmbedding(Operation operation) { + super(operation, OP_NAME); } /** @@ -56,6 +62,9 @@ public static ConfigureTPUEmbedding create(Scope scope, String config) { return new ConfigureTPUEmbedding(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ConfigureTPUEmbedding.class + ) public static class Inputs extends RawOpInputs { /** * Serialized tensorflow.tpu.TPUEmbeddingConfiguration that diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java index 52c9e9a4bf0..16cc920242e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = CrossReplicaSum.OP_NAME, + inputsClass = CrossReplicaSum.Inputs.class +) public final class CrossReplicaSum extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class CrossReplicaSum extends RawOp implements O private Output output; - private CrossReplicaSum(Operation operation) { - super(operation); + public CrossReplicaSum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = CrossReplicaSum.class + ) public static class Inputs extends RawOpInputs> { /** * The local input to the sum. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java index 16e93996681..0a12ae56535 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -37,6 +39,10 @@ * differentiation of graphs containing embeddings via the TPU Embedding Python * libraries. */ +@OpMetadata( + opType = EmbeddingActivations.OP_NAME, + inputsClass = EmbeddingActivations.Inputs.class +) public final class EmbeddingActivations extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class EmbeddingActivations extends RawOp implements Operand output; - private EmbeddingActivations(Operation operation) { - super(operation); + public EmbeddingActivations(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -90,6 +96,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = EmbeddingActivations.class + ) public static class Inputs extends RawOpInputs { /** * A trainable variable, enabling optimizers to find this op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java index dc6902258ab..51b84ed8f18 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java @@ -27,20 +27,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; /** * An op that enqueues a list of input batch tensors to TPUEmbedding. */ +@OpMetadata( + opType = EnqueueTPUEmbeddingIntegerBatch.OP_NAME, + inputsClass = EnqueueTPUEmbeddingIntegerBatch.Inputs.class +) public final class EnqueueTPUEmbeddingIntegerBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "EnqueueTPUEmbeddingIntegerBatch"; - private EnqueueTPUEmbeddingIntegerBatch(Operation operation) { - super(operation); + public EnqueueTPUEmbeddingIntegerBatch(Operation operation) { + super(operation, OP_NAME); } /** @@ -107,6 +113,9 @@ public Options deviceOrdinal(Long deviceOrdinal) { } } + @OpInputsMetadata( + outputsClass = EnqueueTPUEmbeddingIntegerBatch.class + ) public static class Inputs extends RawOpInputs { /** * A list of 1D tensors, one for each embedding table, containing the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java index 8128d89956f..6d63b32a5bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -42,14 +44,18 @@ * with dim_size() equal to the total number of lookups into the table described by * the corresponding feature. */ +@OpMetadata( + opType = EnqueueTPUEmbeddingRaggedTensorBatch.OP_NAME, + inputsClass = EnqueueTPUEmbeddingRaggedTensorBatch.Inputs.class +) public final class EnqueueTPUEmbeddingRaggedTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "EnqueueTPUEmbeddingRaggedTensorBatch"; - private EnqueueTPUEmbeddingRaggedTensorBatch(Operation operation) { - super(operation); + public EnqueueTPUEmbeddingRaggedTensorBatch(Operation operation) { + super(operation, OP_NAME); } /** @@ -312,6 +318,9 @@ public Options numFeatures(Long... numFeatures) { } } + @OpInputsMetadata( + outputsClass = EnqueueTPUEmbeddingRaggedTensorBatch.class + ) public static class Inputs extends RawOpInputs { /** * A list of rank 1 Tensors specifying the break points for splitting diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java index b94730af5f3..f33df0dcc5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -43,14 +45,18 @@ * must have the same shape, i.e. rank 1 with dim_size() equal to the total * number of lookups into the table described by the corresponding table_id. */ +@OpMetadata( + opType = EnqueueTPUEmbeddingSparseBatch.OP_NAME, + inputsClass = EnqueueTPUEmbeddingSparseBatch.Inputs.class +) public final class EnqueueTPUEmbeddingSparseBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "EnqueueTPUEmbeddingSparseBatch"; - private EnqueueTPUEmbeddingSparseBatch(Operation operation) { - super(operation); + public EnqueueTPUEmbeddingSparseBatch(Operation operation) { + super(operation, OP_NAME); } /** @@ -199,6 +205,9 @@ public Options combiners(String... combiners) { } } + @OpInputsMetadata( + outputsClass = EnqueueTPUEmbeddingSparseBatch.class + ) public static class Inputs extends RawOpInputs { /** * A list of rank 1 Tensors specifying the training example and diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java index d431aec3598..75f59eca096 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -42,14 +44,18 @@ * with dim_size() equal to the total number of lookups into the table described by * the corresponding feature. */ +@OpMetadata( + opType = EnqueueTPUEmbeddingSparseTensorBatch.OP_NAME, + inputsClass = EnqueueTPUEmbeddingSparseTensorBatch.Inputs.class +) public final class EnqueueTPUEmbeddingSparseTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "EnqueueTPUEmbeddingSparseTensorBatch"; - private EnqueueTPUEmbeddingSparseTensorBatch(Operation operation) { - super(operation); + public EnqueueTPUEmbeddingSparseTensorBatch(Operation operation) { + super(operation, OP_NAME); } /** @@ -310,6 +316,9 @@ public Options numFeatures(Long... numFeatures) { } } + @OpInputsMetadata( + outputsClass = EnqueueTPUEmbeddingSparseTensorBatch.class + ) public static class Inputs extends RawOpInputs { /** * A list of rank 1 Tensors specifying the training example to diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Execute.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Execute.java index 14c939ac6f9..68e5d3a3f43 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Execute.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Execute.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -39,6 +41,10 @@ * Op that loads and executes a TPU program on a TPU device. * For the internal use of the distributed TPU compiler. */ +@OpMetadata( + opType = Execute.OP_NAME, + inputsClass = Execute.Inputs.class +) @Operator( group = "tpu" ) @@ -51,8 +57,8 @@ public final class Execute extends RawOp implements Iterable> { private List> results; @SuppressWarnings("unchecked") - private Execute(Operation operation) { - super(operation); + public Execute(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int resultsLength = operation.outputListLength("results"); results = Arrays.asList(operation.outputList(outputIdx, resultsLength)); @@ -95,6 +101,9 @@ public Iterator> iterator() { return (Iterator) results.iterator(); } + @OpInputsMetadata( + outputsClass = Execute.class + ) public static class Inputs extends RawOpInputs { /** * The args input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ExecuteAndUpdateVariables.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ExecuteAndUpdateVariables.java index 260af4b758a..3506c5104b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ExecuteAndUpdateVariables.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ExecuteAndUpdateVariables.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -45,6 +47,10 @@ * program outputs are consumed by these variables will not appear in the op * output. For the internal use of the distributed TPU compiler. */ +@OpMetadata( + opType = ExecuteAndUpdateVariables.OP_NAME, + inputsClass = ExecuteAndUpdateVariables.Inputs.class +) @Operator( group = "tpu" ) @@ -57,8 +63,8 @@ public final class ExecuteAndUpdateVariables extends RawOp implements Iterable> results; @SuppressWarnings("unchecked") - private ExecuteAndUpdateVariables(Operation operation) { - super(operation); + public ExecuteAndUpdateVariables(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int resultsLength = operation.outputListLength("results"); results = Arrays.asList(operation.outputList(outputIdx, resultsLength)); @@ -114,6 +120,9 @@ public Iterator> iterator() { return (Iterator) results.iterator(); } + @OpInputsMetadata( + outputsClass = ExecuteAndUpdateVariables.class + ) public static class Inputs extends RawOpInputs { /** * The args input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java index 4f5e1a1dd73..a7b04bdfc2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = InfeedDequeue.OP_NAME, + inputsClass = InfeedDequeue.Inputs.class +) public final class InfeedDequeue extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class InfeedDequeue extends RawOp implements Opera private Output output; - private InfeedDequeue(Operation operation) { - super(operation); + public InfeedDequeue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = InfeedDequeue.class + ) public static class Inputs extends RawOpInputs> { /** * The type of elements in the tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java index 5dd69faeae1..92658e1ea61 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java @@ -31,12 +31,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Fetches multiple values from infeed as an XLA tuple. */ +@OpMetadata( + opType = InfeedDequeueTuple.OP_NAME, + inputsClass = InfeedDequeueTuple.Inputs.class +) public final class InfeedDequeueTuple extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class InfeedDequeueTuple extends RawOp implements Iterable> outputs; @SuppressWarnings("unchecked") - private InfeedDequeueTuple(Operation operation) { - super(operation); + public InfeedDequeueTuple(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList(operation.outputList(outputIdx, outputsLength)); @@ -92,6 +98,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = InfeedDequeueTuple.class + ) public static class Inputs extends RawOpInputs { /** * The element types of each element in `outputs`. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java index 07d0f827355..019f71df8dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java @@ -28,20 +28,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * An op which feeds a single Tensor value into the computation. */ +@OpMetadata( + opType = InfeedEnqueue.OP_NAME, + inputsClass = InfeedEnqueue.Inputs.class +) public final class InfeedEnqueue extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "InfeedEnqueue"; - private InfeedEnqueue(Operation operation) { - super(operation); + public InfeedEnqueue(Operation operation) { + super(operation, OP_NAME); } /** @@ -189,6 +195,9 @@ public Options deviceOrdinal(Long deviceOrdinal) { } } + @OpInputsMetadata( + outputsClass = InfeedEnqueue.class + ) public static class Inputs extends RawOpInputs { /** * A tensor that will be provided using the infeed mechanism. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java index 09c8647caf7..f46fec406cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.family.TType; /** * An op which enqueues prelinearized buffer into TPU infeed. */ +@OpMetadata( + opType = InfeedEnqueuePrelinearizedBuffer.OP_NAME, + inputsClass = InfeedEnqueuePrelinearizedBuffer.Inputs.class +) public final class InfeedEnqueuePrelinearizedBuffer extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "InfeedEnqueuePrelinearizedBuffer"; - private InfeedEnqueuePrelinearizedBuffer(Operation operation) { - super(operation); + public InfeedEnqueuePrelinearizedBuffer(Operation operation) { + super(operation, OP_NAME); } /** @@ -99,6 +105,9 @@ public Options deviceOrdinal(Long deviceOrdinal) { } } + @OpInputsMetadata( + outputsClass = InfeedEnqueuePrelinearizedBuffer.class + ) public static class Inputs extends RawOpInputs { /** * A variant tensor representing linearized output. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java index 1704d634091..807f7007e4b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java @@ -29,19 +29,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; /** * Feeds multiple Tensor values into the computation as an XLA tuple. */ +@OpMetadata( + opType = InfeedEnqueueTuple.OP_NAME, + inputsClass = InfeedEnqueueTuple.Inputs.class +) public final class InfeedEnqueueTuple extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "InfeedEnqueueTuple"; - private InfeedEnqueueTuple(Operation operation) { - super(operation); + public InfeedEnqueueTuple(Operation operation) { + super(operation, OP_NAME); } /** @@ -173,6 +179,9 @@ public Options deviceOrdinal(Long deviceOrdinal) { } } + @OpInputsMetadata( + outputsClass = InfeedEnqueueTuple.class + ) public static class Inputs extends RawOpInputs { /** * A list of tensors that will be provided using the infeed mechanism. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java index fa607e3d65f..1c0b6b3cc62 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingADAMParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingADAMParameters.Inputs.class +) public final class LoadTPUEmbeddingADAMParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingADAMParameters"; - private LoadTPUEmbeddingADAMParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingADAMParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingADAMParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the ADAM optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParametersGradAccumDebug.java index 3689a78b02f..954bd247d65 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingADAMParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingADAMParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingADAMParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingADAMParametersGradAccumDebug"; - private LoadTPUEmbeddingADAMParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingADAMParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -165,6 +171,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingADAMParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the ADAM optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java index bc573e662ae..62ea77c23fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingAdadeltaParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingAdadeltaParameters.Inputs.class +) public final class LoadTPUEmbeddingAdadeltaParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingAdadeltaParameters"; - private LoadTPUEmbeddingAdadeltaParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingAdadeltaParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingAdadeltaParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the Adadelta optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.java index 55c53423e92..0aaa8ccf6d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingAdadeltaParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingAdadeltaParametersGradAccumDebug"; - private LoadTPUEmbeddingAdadeltaParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingAdadeltaParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -165,6 +171,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the Adadelta optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java index 4932fcb0f30..14136f03a9d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingAdagradParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingAdagradParameters.Inputs.class +) public final class LoadTPUEmbeddingAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingAdagradParameters"; - private LoadTPUEmbeddingAdagradParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingAdagradParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -160,6 +166,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingAdagradParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the Adagrad optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParametersGradAccumDebug.java index 6c6982d38b9..e373ad25faa 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingAdagradParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingAdagradParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingAdagradParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingAdagradParametersGradAccumDebug"; - private LoadTPUEmbeddingAdagradParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingAdagradParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingAdagradParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the Adagrad optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java index 5aa18411600..ec974b52bf6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingCenteredRMSPropParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingCenteredRMSPropParameters.Inputs.class +) public final class LoadTPUEmbeddingCenteredRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingCenteredRMSPropParameters"; - private LoadTPUEmbeddingCenteredRMSPropParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingCenteredRMSPropParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -165,6 +171,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingCenteredRMSPropParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the centered RMSProp optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java index 77c88b54456..c4f8fa36ad9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingFTRLParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingFTRLParameters.Inputs.class +) public final class LoadTPUEmbeddingFTRLParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingFTRLParameters"; - private LoadTPUEmbeddingFTRLParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingFTRLParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingFTRLParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the FTRL optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParametersGradAccumDebug.java index 78438e9fec9..40369244b28 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingFTRLParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingFTRLParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingFTRLParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingFTRLParametersGradAccumDebug"; - private LoadTPUEmbeddingFTRLParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingFTRLParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -165,6 +171,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingFTRLParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the FTRL optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java index 111f9c8541b..0356769ccd8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingMDLAdagradLightParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingMDLAdagradLightParameters.Inputs.class +) public final class LoadTPUEmbeddingMDLAdagradLightParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingMDLAdagradLightParameters"; - private LoadTPUEmbeddingMDLAdagradLightParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingMDLAdagradLightParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -165,6 +171,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingMDLAdagradLightParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the MDL Adagrad Light optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java index 1abf0471457..33c1c6e162d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingMomentumParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingMomentumParameters.Inputs.class +) public final class LoadTPUEmbeddingMomentumParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingMomentumParameters"; - private LoadTPUEmbeddingMomentumParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingMomentumParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -160,6 +166,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingMomentumParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the Momentum optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParametersGradAccumDebug.java index ba22bbb8363..56967a766c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingMomentumParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingMomentumParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingMomentumParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingMomentumParametersGradAccumDebug"; - private LoadTPUEmbeddingMomentumParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingMomentumParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingMomentumParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the Momentum optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java index 9306ca0d29f..5f1fbb072f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingProximalAdagradParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingProximalAdagradParameters.Inputs.class +) public final class LoadTPUEmbeddingProximalAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingProximalAdagradParameters"; - private LoadTPUEmbeddingProximalAdagradParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingProximalAdagradParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -161,6 +167,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingProximalAdagradParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the proximal Adagrad optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.java index 8f1a7957da3..2096552062b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug"; - private LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the proximal Adagrad optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java index 424542eadb2..c108ce861fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** * The LoadTPUEmbeddingProximalYogiParameters operation */ +@OpMetadata( + opType = LoadTPUEmbeddingProximalYogiParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingProximalYogiParameters.Inputs.class +) public final class LoadTPUEmbeddingProximalYogiParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingProximalYogiParameters"; - private LoadTPUEmbeddingProximalYogiParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingProximalYogiParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -158,6 +164,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingProximalYogiParameters.class + ) public static class Inputs extends RawOpInputs { /** * The parameters input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.java index 4c3dfed12dc..94a1f0926ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** * The LoadTPUEmbeddingProximalYogiParametersGradAccumDebug operation */ +@OpMetadata( + opType = LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingProximalYogiParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingProximalYogiParametersGradAccumDebug"; - private LoadTPUEmbeddingProximalYogiParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingProximalYogiParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -160,6 +166,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The parameters input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java index 69e422ee802..30cdfa94252 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingRMSPropParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingRMSPropParameters.Inputs.class +) public final class LoadTPUEmbeddingRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingRMSPropParameters"; - private LoadTPUEmbeddingRMSPropParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingRMSPropParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -163,6 +169,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingRMSPropParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the RMSProp optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParametersGradAccumDebug.java index 3f987e4beac..c6d8c3a4800 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingRMSPropParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingRMSPropParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingRMSPropParametersGradAccumDebug"; - private LoadTPUEmbeddingRMSPropParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingRMSPropParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -165,6 +171,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingRMSPropParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the RMSProp optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java index 62e73a01c52..b43487c0c8b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingStochasticGradientDescentParameters.OP_NAME, + inputsClass = LoadTPUEmbeddingStochasticGradientDescentParameters.Inputs.class +) public final class LoadTPUEmbeddingStochasticGradientDescentParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingStochasticGradientDescentParameters"; - private LoadTPUEmbeddingStochasticGradientDescentParameters(Operation operation) { - super(operation); + public LoadTPUEmbeddingStochasticGradientDescentParameters(Operation operation) { + super(operation, OP_NAME); } /** @@ -158,6 +164,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingStochasticGradientDescentParameters.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the stochastic gradient descent optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java index dd5fd61bc44..01307c9da5e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,14 +38,18 @@ * parameters that are loaded from a checkpoint before a training loop is * executed. */ +@OpMetadata( + opType = LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.OP_NAME, + inputsClass = LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Inputs.class +) public final class LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug"; - private LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug(Operation operation) { - super(operation); + public LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); } /** @@ -161,6 +167,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * Value of parameters used in the stochastic gradient descent optimization algorithm. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java index c026d5f55a4..eb737021f93 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; /** @@ -35,6 +37,10 @@ * (for regular inference) to execute the TPU program on. The output is * consumed by TPUPartitionedCall. */ +@OpMetadata( + opType = OrdinalSelector.OP_NAME, + inputsClass = OrdinalSelector.Inputs.class +) public final class OrdinalSelector extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class OrdinalSelector extends RawOp implements Operand { private Output deviceOrdinals; - private OrdinalSelector(Operation operation) { - super(operation); + public OrdinalSelector(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; deviceOrdinals = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return deviceOrdinals; } + @OpInputsMetadata( + outputsClass = OrdinalSelector.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new OrdinalSelector(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java index b2cd4412cae..26d20976632 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = OutfeedDequeue.OP_NAME, + inputsClass = OutfeedDequeue.Inputs.class +) public final class OutfeedDequeue extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class OutfeedDequeue extends RawOp implements Oper private Output output; - private OutfeedDequeue(Operation operation) { - super(operation); + public OutfeedDequeue(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -129,6 +135,9 @@ public Options deviceOrdinal(Long deviceOrdinal) { } } + @OpInputsMetadata( + outputsClass = OutfeedDequeue.class + ) public static class Inputs extends RawOpInputs> { /** * The type of elements in the tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java index c667cf7a4ed..a14130ea768 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * This operation will block indefinitely until data is available. Output {@code i} * corresponds to XLA tuple element {@code i}. */ +@OpMetadata( + opType = OutfeedDequeueTuple.OP_NAME, + inputsClass = OutfeedDequeueTuple.Inputs.class +) public final class OutfeedDequeueTuple extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class OutfeedDequeueTuple extends RawOp implements Iterable> outputs; @SuppressWarnings("unchecked") - private OutfeedDequeueTuple(Operation operation) { - super(operation); + public OutfeedDequeueTuple(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList(operation.outputList(outputIdx, outputsLength)); @@ -137,6 +143,9 @@ public Options deviceOrdinal(Long deviceOrdinal) { } } + @OpInputsMetadata( + outputsClass = OutfeedDequeueTuple.class + ) public static class Inputs extends RawOpInputs { /** * The element types of each element in `outputs`. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java index 328e3a7da5f..84331496731 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * This operation will block indefinitely until data is available. Output {@code i} * corresponds to XLA tuple element {@code i}. */ +@OpMetadata( + opType = OutfeedDequeueTupleV2.OP_NAME, + inputsClass = OutfeedDequeueTupleV2.Inputs.class +) public final class OutfeedDequeueTupleV2 extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class OutfeedDequeueTupleV2 extends RawOp implements Iterable> outputs; @SuppressWarnings("unchecked") - private OutfeedDequeueTupleV2(Operation operation) { - super(operation); + public OutfeedDequeueTupleV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList(operation.outputList(outputIdx, outputsLength)); @@ -100,6 +106,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = OutfeedDequeueTupleV2.class + ) public static class Inputs extends RawOpInputs { /** * An int scalar tensor, representing the TPU device to use. This should be -1 when diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java index bdd0a7f4157..19bdb087c4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = OutfeedDequeueV2.OP_NAME, + inputsClass = OutfeedDequeueV2.Inputs.class +) public final class OutfeedDequeueV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -48,8 +54,8 @@ public final class OutfeedDequeueV2 extends RawOp implements Op private Output output; - private OutfeedDequeueV2(Operation operation) { - super(operation); + public OutfeedDequeueV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = OutfeedDequeueV2.class + ) public static class Inputs extends RawOpInputs> { /** * An int scalar tensor, representing the TPU device to use. This should be -1 when diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java index effdad173b0..a934cf681d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java @@ -26,20 +26,26 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * Enqueue a Tensor on the computation outfeed. */ +@OpMetadata( + opType = OutfeedEnqueue.OP_NAME, + inputsClass = OutfeedEnqueue.Inputs.class +) public final class OutfeedEnqueue extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "OutfeedEnqueue"; - private OutfeedEnqueue(Operation operation) { - super(operation); + public OutfeedEnqueue(Operation operation) { + super(operation, OP_NAME); } /** @@ -58,6 +64,9 @@ public static OutfeedEnqueue create(Scope scope, Operand input) return new OutfeedEnqueue(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = OutfeedEnqueue.class + ) public static class Inputs extends RawOpInputs { /** * A tensor that will be inserted into the outfeed queue. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java index 5e915899fd6..158315c18f7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java @@ -27,19 +27,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; /** * Enqueue multiple Tensor values on the computation outfeed. */ +@OpMetadata( + opType = OutfeedEnqueueTuple.OP_NAME, + inputsClass = OutfeedEnqueueTuple.Inputs.class +) public final class OutfeedEnqueueTuple extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "OutfeedEnqueueTuple"; - private OutfeedEnqueueTuple(Operation operation) { - super(operation); + public OutfeedEnqueueTuple(Operation operation) { + super(operation, OP_NAME); } /** @@ -59,6 +65,9 @@ public static OutfeedEnqueueTuple create(Scope scope, Iterable> input return new OutfeedEnqueueTuple(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = OutfeedEnqueueTuple.class + ) public static class Inputs extends RawOpInputs { /** * A list of tensors that will be inserted into the outfeed queue as an diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java index a85442af958..2e9e97399af 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Calls a function placed on a specified TPU device. */ +@OpMetadata( + opType = PartitionedCall.OP_NAME, + inputsClass = PartitionedCall.Inputs.class +) public final class PartitionedCall extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class PartitionedCall extends RawOp implements Iterable> output; @SuppressWarnings("unchecked") - private PartitionedCall(Operation operation) { - super(operation); + public PartitionedCall(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -133,6 +139,9 @@ public Options autotunerThresh(Long autotunerThresh) { } } + @OpInputsMetadata( + outputsClass = PartitionedCall.class + ) public static class Inputs extends RawOpInputs { /** * The arguments to the function. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java index 8e080b4fd05..3800a656fc4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = PartitionedInput.OP_NAME, + inputsClass = PartitionedInput.Inputs.class +) @Operator( group = "tpu" ) @@ -48,8 +54,8 @@ public final class PartitionedInput extends RawOp implements Op private Output output; - private PartitionedInput(Operation operation) { - super(operation); + public PartitionedInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -127,6 +133,9 @@ public Options partitionDim(Long partitionDim) { } } + @OpInputsMetadata( + outputsClass = PartitionedInput.class + ) public static class Inputs extends RawOpInputs> { /** * A list of partitioned inputs which must have the same shape. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedOutput.java index 626610bad5e..4bf51f2ba49 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedOutput.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = PartitionedOutput.OP_NAME, + inputsClass = PartitionedOutput.Inputs.class +) @Operator( group = "tpu" ) @@ -51,8 +57,8 @@ public final class PartitionedOutput extends RawOp implements I private List> output; @SuppressWarnings("unchecked") - private PartitionedOutput(Operation operation) { - super(operation); + public PartitionedOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList((Output[]) operation.outputList(outputIdx, outputLength)); @@ -133,6 +139,9 @@ public Options partitionDim(Long partitionDim) { } } + @OpInputsMetadata( + outputsClass = PartitionedOutput.class + ) public static class Inputs extends RawOpInputs> { /** * A tensor which represents the full shape of partitioned tensors. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java index ddb2f379660..df3eddd9523 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java @@ -29,12 +29,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * An op which linearizes one Tensor value to an opaque variant tensor. */ +@OpMetadata( + opType = Prelinearize.OP_NAME, + inputsClass = Prelinearize.Inputs.class +) public final class Prelinearize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class Prelinearize extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private Prelinearize(Operation operation) { - super(operation); + public Prelinearize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -180,6 +186,9 @@ public Options layout(Long... layout) { } } + @OpInputsMetadata( + outputsClass = Prelinearize.class + ) public static class Inputs extends RawOpInputs { /** * A tensor that will be linearized. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java index 344cab1a66a..bb9d386ff87 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java @@ -30,12 +30,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; /** * An op which linearizes multiple Tensor values to an opaque variant tensor. */ +@OpMetadata( + opType = PrelinearizeTuple.OP_NAME, + inputsClass = PrelinearizeTuple.Inputs.class +) public final class PrelinearizeTuple extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class PrelinearizeTuple extends RawOp implements Operand { private Output output; @SuppressWarnings("unchecked") - private PrelinearizeTuple(Operation operation) { - super(operation); + public PrelinearizeTuple(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -165,6 +171,9 @@ public Options layouts(Long... layouts) { } } + @OpInputsMetadata( + outputsClass = PrelinearizeTuple.class + ) public static class Inputs extends RawOpInputs { /** * A list of tensors that will be provided using the infeed mechanism. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java index 4dc37a810c5..b603c292192 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -40,6 +42,10 @@ * one Tensor of activations per table specified in the model. There can be at * most one RecvTPUEmbeddingActivations op in the TPU graph. */ +@OpMetadata( + opType = RecvTPUEmbeddingActivations.OP_NAME, + inputsClass = RecvTPUEmbeddingActivations.Inputs.class +) public final class RecvTPUEmbeddingActivations extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RecvTPUEmbeddingActivations extends RawOp implements Iterable private List> outputs; @SuppressWarnings("unchecked") - private RecvTPUEmbeddingActivations(Operation operation) { - super(operation); + public RecvTPUEmbeddingActivations(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList((Output[]) operation.outputList(outputIdx, outputsLength)); @@ -92,6 +98,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = RecvTPUEmbeddingActivations.class + ) public static class Inputs extends RawOpInputs { /** * Serialized TPUEmbeddingConfiguration proto. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java index 8e12a13b887..1696d439318 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java @@ -26,19 +26,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; /** * Metadata indicating how the TPU computation should be replicated. * This operation holds the metadata common to operations of a {@code tpu.replicate()} computation subgraph. */ +@OpMetadata( + opType = ReplicateMetadata.OP_NAME, + inputsClass = ReplicateMetadata.Inputs.class +) public final class ReplicateMetadata extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "TPUReplicateMetadata"; - private ReplicateMetadata(Operation operation) { - super(operation); + public ReplicateMetadata(Operation operation) { + super(operation, OP_NAME); } /** @@ -430,6 +436,9 @@ public Options useSpmdForXlaPartitioning(Boolean useSpmdForXlaPartitioning) { } } + @OpInputsMetadata( + outputsClass = ReplicateMetadata.class + ) public static class Inputs extends RawOpInputs { /** * Number of replicas of the computation diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java index 1bfa9e59e25..7aabbce711a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +48,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReplicatedInput.OP_NAME, + inputsClass = ReplicatedInput.Inputs.class +) public final class ReplicatedInput extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -54,8 +60,8 @@ public final class ReplicatedInput extends RawOp implements Ope private Output output; - private ReplicatedInput(Operation operation) { - super(operation); + public ReplicatedInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -183,6 +189,9 @@ public Options isPacked(Boolean isPacked) { } } + @OpInputsMetadata( + outputsClass = ReplicatedInput.class + ) public static class Inputs extends RawOpInputs> { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java index f644763c1e8..6d956b0ef19 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -45,6 +47,10 @@ * * @param data type for {@code outputs} output */ +@OpMetadata( + opType = ReplicatedOutput.OP_NAME, + inputsClass = ReplicatedOutput.Inputs.class +) public final class ReplicatedOutput extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine @@ -54,8 +60,8 @@ public final class ReplicatedOutput extends RawOp implements It private List> outputs; @SuppressWarnings("unchecked") - private ReplicatedOutput(Operation operation) { - super(operation); + public ReplicatedOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList((Output[]) operation.outputList(outputIdx, outputsLength)); @@ -97,6 +103,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = ReplicatedOutput.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java index d05f62e2f3e..122e62b05f0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingADAMParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingADAMParameters.Inputs.class +) public final class RetrieveTPUEmbeddingADAMParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingADAMParameters extends RawOp { private Output velocities; - private RetrieveTPUEmbeddingADAMParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingADAMParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); momenta = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingADAMParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParametersGradAccumDebug.java index c08259b0d89..e598b5ec412 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingADAMParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingADAMParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RetrieveTPUEmbeddingADAMParametersGradAccumDebug extends RawO private Output gradientAccumulators; - private RetrieveTPUEmbeddingADAMParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingADAMParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); momenta = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingADAMParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java index d7f5626f574..5b38fd50f80 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingAdadeltaParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingAdadeltaParameters.Inputs.class +) public final class RetrieveTPUEmbeddingAdadeltaParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingAdadeltaParameters extends RawOp { private Output updates; - private RetrieveTPUEmbeddingAdadeltaParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingAdadeltaParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingAdadeltaParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.java index 7740306b350..054fdbfc2cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug extends private Output gradientAccumulators; - private RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java index 64762a31449..4f77f80451a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingAdagradParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingAdagradParameters.Inputs.class +) public final class RetrieveTPUEmbeddingAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RetrieveTPUEmbeddingAdagradParameters extends RawOp { private Output accumulators; - private RetrieveTPUEmbeddingAdagradParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingAdagradParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -180,6 +186,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingAdagradParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.java index 78c29954df4..cf2c37361ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingAdagradParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingAdagradParametersGradAccumDebug extends R private Output gradientAccumulators; - private RetrieveTPUEmbeddingAdagradParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingAdagradParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java index a4dd7ba910e..9db53a133f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingCenteredRMSPropParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingCenteredRMSPropParameters.Inputs.class +) public final class RetrieveTPUEmbeddingCenteredRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RetrieveTPUEmbeddingCenteredRMSPropParameters extends RawOp { private Output mg; - private RetrieveTPUEmbeddingCenteredRMSPropParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingCenteredRMSPropParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); ms = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingCenteredRMSPropParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java index 6e1ab7a50c0..cd876266705 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingFTRLParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingFTRLParameters.Inputs.class +) public final class RetrieveTPUEmbeddingFTRLParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingFTRLParameters extends RawOp { private Output linears; - private RetrieveTPUEmbeddingFTRLParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingFTRLParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingFTRLParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.java index 5dc9db5c737..7dab6b465a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingFTRLParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RetrieveTPUEmbeddingFTRLParametersGradAccumDebug extends RawO private Output gradientAccumulators; - private RetrieveTPUEmbeddingFTRLParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingFTRLParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java index 0b341ec67e7..018babc869a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingMDLAdagradLightParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingMDLAdagradLightParameters.Inputs.class +) public final class RetrieveTPUEmbeddingMDLAdagradLightParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RetrieveTPUEmbeddingMDLAdagradLightParameters extends RawOp { private Output benefits; - private RetrieveTPUEmbeddingMDLAdagradLightParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingMDLAdagradLightParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingMDLAdagradLightParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java index 360d345257c..20f6ee9d54f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingMomentumParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingMomentumParameters.Inputs.class +) public final class RetrieveTPUEmbeddingMomentumParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RetrieveTPUEmbeddingMomentumParameters extends RawOp { private Output momenta; - private RetrieveTPUEmbeddingMomentumParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingMomentumParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); momenta = operation.output(outputIdx++); @@ -180,6 +186,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingMomentumParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.java index 5ba59def58d..aa51de4d404 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingMomentumParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingMomentumParametersGradAccumDebug extends private Output gradientAccumulators; - private RetrieveTPUEmbeddingMomentumParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingMomentumParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); momenta = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java index 5a351245ecb..3440a794cd4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingProximalAdagradParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingProximalAdagradParameters.Inputs.class +) public final class RetrieveTPUEmbeddingProximalAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RetrieveTPUEmbeddingProximalAdagradParameters extends RawOp { private Output accumulators; - private RetrieveTPUEmbeddingProximalAdagradParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingProximalAdagradParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -180,6 +186,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingProximalAdagradParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.java index 740410a1c49..3565cbce042 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug e private Output gradientAccumulators; - private RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); accumulators = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java index 3caff4629ad..083ef769cb9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java @@ -26,11 +26,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** * The RetrieveTPUEmbeddingProximalYogiParameters operation */ +@OpMetadata( + opType = RetrieveTPUEmbeddingProximalYogiParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingProximalYogiParameters.Inputs.class +) public final class RetrieveTPUEmbeddingProximalYogiParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -43,8 +49,8 @@ public final class RetrieveTPUEmbeddingProximalYogiParameters extends RawOp { private Output m; - private RetrieveTPUEmbeddingProximalYogiParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingProximalYogiParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); v = operation.output(outputIdx++); @@ -188,6 +194,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingProximalYogiParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.java index 50fe386f14f..61c55552cd7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.java @@ -26,11 +26,17 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** * The RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug operation */ +@OpMetadata( + opType = RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug exte private Output gradientAccumulators; - private RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); v = operation.output(outputIdx++); @@ -200,6 +206,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java index 0db31288db7..8826547e35e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingRMSPropParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingRMSPropParameters.Inputs.class +) public final class RetrieveTPUEmbeddingRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class RetrieveTPUEmbeddingRMSPropParameters extends RawOp { private Output mom; - private RetrieveTPUEmbeddingRMSPropParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingRMSPropParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); ms = operation.output(outputIdx++); @@ -192,6 +198,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingRMSPropParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.java index 3d32bd5d1f2..ee671e633d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -49,8 +55,8 @@ public final class RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug extends R private Output gradientAccumulators; - private RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); ms = operation.output(outputIdx++); @@ -204,6 +210,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java index 7d8ab5ae954..c789908ea4c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -36,6 +38,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingStochasticGradientDescentParameters.OP_NAME, + inputsClass = RetrieveTPUEmbeddingStochasticGradientDescentParameters.Inputs.class +) public final class RetrieveTPUEmbeddingStochasticGradientDescentParameters extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -44,8 +50,8 @@ public final class RetrieveTPUEmbeddingStochasticGradientDescentParameters exten private Output parameters; - private RetrieveTPUEmbeddingStochasticGradientDescentParameters(Operation operation) { - super(operation); + public RetrieveTPUEmbeddingStochasticGradientDescentParameters(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); } @@ -174,6 +180,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingStochasticGradientDescentParameters.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java index c199ce91f51..6941a062872 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -35,6 +37,10 @@ * the correct embedding table configuration. For example, this op is * used to retrieve updated parameters before saving a checkpoint. */ +@OpMetadata( + opType = RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.OP_NAME, + inputsClass = RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Inputs.class +) public final class RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -45,9 +51,9 @@ public final class RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAc private Output gradientAccumulators; - private RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug( + public RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug( Operation operation) { - super(operation); + super(operation, OP_NAME); int outputIdx = 0; parameters = operation.output(outputIdx++); gradientAccumulators = operation.output(outputIdx++); @@ -181,6 +187,9 @@ public Options config(String config) { } } + @OpInputsMetadata( + outputsClass = RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.class + ) public static class Inputs extends RawOpInputs { /** * The tableId attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java index 0556e51c322..45cf2fac6cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java @@ -27,19 +27,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** * Performs gradient updates of embedding tables. */ +@OpMetadata( + opType = SendTPUEmbeddingGradients.OP_NAME, + inputsClass = SendTPUEmbeddingGradients.Inputs.class +) public final class SendTPUEmbeddingGradients extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "SendTPUEmbeddingGradients"; - private SendTPUEmbeddingGradients(Operation operation) { - super(operation); + public SendTPUEmbeddingGradients(Operation operation) { + super(operation, OP_NAME); } /** @@ -112,6 +118,9 @@ public Options NN(Long NN) { } } + @OpInputsMetadata( + outputsClass = SendTPUEmbeddingGradients.class + ) public static class Inputs extends RawOpInputs { /** * A TensorList of gradients with which to update embedding tables. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java index bfd80d334da..b2a8949e876 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java @@ -25,19 +25,25 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; /** * Shuts down a running distributed TPU system. * The op returns an error if no system is running. */ +@OpMetadata( + opType = ShutdownDistributedTPU.OP_NAME, + inputsClass = ShutdownDistributedTPU.Inputs.class +) public final class ShutdownDistributedTPU extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ShutdownDistributedTPU"; - private ShutdownDistributedTPU(Operation operation) { - super(operation); + public ShutdownDistributedTPU(Operation operation) { + super(operation, OP_NAME); } /** @@ -54,6 +60,9 @@ public static ShutdownDistributedTPU create(Scope scope) { return new ShutdownDistributedTPU(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ShutdownDistributedTPU.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new ShutdownDistributedTPU(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java index 7bc258d87f6..1b8dffc0997 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; /** @@ -37,6 +39,10 @@ * * @deprecated use {@link org.tensorflow.op.tpu.CompilationResult} instead */ +@OpMetadata( + opType = TPUCompilationResult.OP_NAME, + inputsClass = TPUCompilationResult.Inputs.class +) @Deprecated public final class TPUCompilationResult extends RawOp implements Operand { /** @@ -46,8 +52,8 @@ public final class TPUCompilationResult extends RawOp implements Operand output; - private TPUCompilationResult(Operation operation) { - super(operation); + public TPUCompilationResult(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -80,6 +86,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TPUCompilationResult.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new TPUCompilationResult(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java index 360c11d3a5a..29ea1c90743 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; /** @@ -39,6 +41,10 @@ * * @deprecated use {@link org.tensorflow.op.tpu.EmbeddingActivations} instead */ +@OpMetadata( + opType = TPUEmbeddingActivations.OP_NAME, + inputsClass = TPUEmbeddingActivations.Inputs.class +) @Deprecated public final class TPUEmbeddingActivations extends RawOp implements Operand { /** @@ -48,8 +54,8 @@ public final class TPUEmbeddingActivations extends RawOp implements Operand output; - private TPUEmbeddingActivations(Operation operation) { - super(operation); + public TPUEmbeddingActivations(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TPUEmbeddingActivations.class + ) public static class Inputs extends RawOpInputs { /** * A trainable variable, enabling optimizers to find this op. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java index 124bccfda5c..31f02d03220 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; /** * Metadata indicating how the TPU computation should be replicated. @@ -33,6 +35,10 @@ * * @deprecated use {@link org.tensorflow.op.tpu.ReplicateMetadata} instead */ +@OpMetadata( + opType = TPUReplicateMetadata.OP_NAME, + inputsClass = TPUReplicateMetadata.Inputs.class +) @Deprecated public final class TPUReplicateMetadata extends RawOp { /** @@ -40,8 +46,8 @@ public final class TPUReplicateMetadata extends RawOp { */ public static final String OP_NAME = "TPUReplicateMetadata"; - private TPUReplicateMetadata(Operation operation) { - super(operation); + public TPUReplicateMetadata(Operation operation) { + super(operation, OP_NAME); } /** @@ -433,6 +439,9 @@ public Options useSpmdForXlaPartitioning(Boolean useSpmdForXlaPartitioning) { } } + @OpInputsMetadata( + outputsClass = TPUReplicateMetadata.class + ) public static class Inputs extends RawOpInputs { /** * Number of replicas of the computation diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java index bfb2ef52af0..b64bcef7902 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +50,10 @@ * * @deprecated use {@link org.tensorflow.op.tpu.ReplicatedInput} instead */ +@OpMetadata( + opType = TPUReplicatedInput.OP_NAME, + inputsClass = TPUReplicatedInput.Inputs.class +) @Deprecated public final class TPUReplicatedInput extends RawOp implements Operand { /** @@ -57,8 +63,8 @@ public final class TPUReplicatedInput extends RawOp implements private Output output; - private TPUReplicatedInput(Operation operation) { - super(operation); + public TPUReplicatedInput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -186,6 +192,9 @@ public Options isPacked(Boolean isPacked) { } } + @OpInputsMetadata( + outputsClass = TPUReplicatedInput.class + ) public static class Inputs extends RawOpInputs> { /** * The inputs input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java index e20f5d1b4c6..ac389716703 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -47,6 +49,10 @@ * * @deprecated use {@link org.tensorflow.op.tpu.ReplicatedOutput} instead */ +@OpMetadata( + opType = TPUReplicatedOutput.OP_NAME, + inputsClass = TPUReplicatedOutput.Inputs.class +) @Deprecated public final class TPUReplicatedOutput extends RawOp implements Iterable> { /** @@ -57,8 +63,8 @@ public final class TPUReplicatedOutput extends RawOp implements private List> outputs; @SuppressWarnings("unchecked") - private TPUReplicatedOutput(Operation operation) { - super(operation); + public TPUReplicatedOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList((Output[]) operation.outputList(outputIdx, outputsLength)); @@ -100,6 +106,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = TPUReplicatedOutput.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java index 5308bcc3b37..5f0a1ee179e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -38,14 +40,18 @@ * specifies the desired state, and format_state_var is the current state of the * variables. */ +@OpMetadata( + opType = TPUReshardVariables.OP_NAME, + inputsClass = TPUReshardVariables.Inputs.class +) public final class TPUReshardVariables extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "TPUReshardVariables"; - private TPUReshardVariables(Operation operation) { - super(operation); + public TPUReshardVariables(Operation operation) { + super(operation, OP_NAME); } /** @@ -69,6 +75,9 @@ public static TPUReshardVariables create(Scope scope, Iterable { /** * The vars input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java index cff0719e947..97e338c0046 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TString; /** @@ -34,6 +36,10 @@ * Heartbeats may be sent periodically to indicate the coordinator is still active, * to retrieve the current worker status and to expedite shutdown when necessary. */ +@OpMetadata( + opType = WorkerHeartbeat.OP_NAME, + inputsClass = WorkerHeartbeat.Inputs.class +) public final class WorkerHeartbeat extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -42,8 +48,8 @@ public final class WorkerHeartbeat extends RawOp implements Operand { private Output response; - private WorkerHeartbeat(Operation operation) { - super(operation); + public WorkerHeartbeat(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; response = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return response; } + @OpInputsMetadata( + outputsClass = WorkerHeartbeat.class + ) public static class Inputs extends RawOpInputs { /** * A string tensor containing a serialized WorkerHeartbeatRequest diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorApplyGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorApplyGradient.java index 64f16b73bab..32824da9c0b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorApplyGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorApplyGradient.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -36,6 +38,10 @@ * Applies a gradient to a given accumulator. * Does not add if local_step is lesser than the accumulator's global_step. */ +@OpMetadata( + opType = AccumulatorApplyGradient.OP_NAME, + inputsClass = AccumulatorApplyGradient.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class AccumulatorApplyGradient extends RawOp { */ public static final String OP_NAME = "AccumulatorApplyGradient"; - private AccumulatorApplyGradient(Operation operation) { - super(operation); + public AccumulatorApplyGradient(Operation operation) { + super(operation, OP_NAME); } /** @@ -70,6 +76,9 @@ public static AccumulatorApplyGradient create(Scope scope, Operand hand return new AccumulatorApplyGradient(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = AccumulatorApplyGradient.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorNumAccumulated.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorNumAccumulated.java index 73199c30e4b..f97d7db1848 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorNumAccumulated.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorNumAccumulated.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -34,6 +36,10 @@ /** * Returns the number of gradients aggregated in the given accumulators. */ +@OpMetadata( + opType = AccumulatorNumAccumulated.OP_NAME, + inputsClass = AccumulatorNumAccumulated.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class AccumulatorNumAccumulated extends RawOp implements Operand numAccumulated; - private AccumulatorNumAccumulated(Operation operation) { - super(operation); + public AccumulatorNumAccumulated(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; numAccumulated = operation.output(outputIdx++); } @@ -81,6 +87,9 @@ public Output asOutput() { return numAccumulated; } + @OpInputsMetadata( + outputsClass = AccumulatorNumAccumulated.class + ) public static class Inputs extends RawOpInputs { /** * The handle to an accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorSetGlobalStep.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorSetGlobalStep.java index 23092eb9759..12536b44d97 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorSetGlobalStep.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorSetGlobalStep.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -35,6 +37,10 @@ * Logs warning if the accumulator's value is already higher than * new_global_step. */ +@OpMetadata( + opType = AccumulatorSetGlobalStep.OP_NAME, + inputsClass = AccumulatorSetGlobalStep.Inputs.class +) @Operator( group = "train" ) @@ -44,8 +50,8 @@ public final class AccumulatorSetGlobalStep extends RawOp { */ public static final String OP_NAME = "AccumulatorSetGlobalStep"; - private AccumulatorSetGlobalStep(Operation operation) { - super(operation); + public AccumulatorSetGlobalStep(Operation operation) { + super(operation, OP_NAME); } /** @@ -67,6 +73,9 @@ public static AccumulatorSetGlobalStep create(Scope scope, Operand hand return new AccumulatorSetGlobalStep(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = AccumulatorSetGlobalStep.class + ) public static class Inputs extends RawOpInputs { /** * The handle to an accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorTakeGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorTakeGradient.java index 52d6bce2df3..2d7c955b968 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorTakeGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/AccumulatorTakeGradient.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -44,6 +46,10 @@ * * @param data type for {@code average} output */ +@OpMetadata( + opType = AccumulatorTakeGradient.OP_NAME, + inputsClass = AccumulatorTakeGradient.Inputs.class +) @Operator( group = "train" ) @@ -55,8 +61,8 @@ public final class AccumulatorTakeGradient extends RawOp implem private Output average; - private AccumulatorTakeGradient(Operation operation) { - super(operation); + public AccumulatorTakeGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; average = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return average; } + @OpInputsMetadata( + outputsClass = AccumulatorTakeGradient.class + ) public static class Inputs extends RawOpInputs> { /** * The handle to an accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java index 84b1ccd7a52..471b49c7995 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAdaMax.OP_NAME, + inputsClass = ApplyAdaMax.Inputs.class +) public final class ApplyAdaMax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -46,8 +52,8 @@ public final class ApplyAdaMax extends RawOp implements Operand private Output out; - private ApplyAdaMax(Operation operation) { - super(operation); + public ApplyAdaMax(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyAdaMax.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdadelta.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdadelta.java index 2d908547cfd..1ec7edff090 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdadelta.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdadelta.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAdadelta.OP_NAME, + inputsClass = ApplyAdadelta.Inputs.class +) @Operator( group = "train" ) @@ -51,8 +57,8 @@ public final class ApplyAdadelta extends RawOp implements Opera private Output out; - private ApplyAdadelta(Operation operation) { - super(operation); + public ApplyAdadelta(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -143,6 +149,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyAdadelta.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagrad.java index 1296c191656..25d2050ebe8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAdagrad.OP_NAME, + inputsClass = ApplyAdagrad.Inputs.class +) @Operator( group = "train" ) @@ -49,8 +55,8 @@ public final class ApplyAdagrad extends RawOp implements Operan private Output out; - private ApplyAdagrad(Operation operation) { - super(operation); + public ApplyAdagrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -162,6 +168,9 @@ public Options updateSlots(Boolean updateSlots) { } } + @OpInputsMetadata( + outputsClass = ApplyAdagrad.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradDa.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradDa.java index a3f60e0471f..e6668ee1593 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradDa.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradDa.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -37,6 +39,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAdagradDa.OP_NAME, + inputsClass = ApplyAdagradDa.Inputs.class +) @Operator( group = "train" ) @@ -48,8 +54,8 @@ public final class ApplyAdagradDa extends RawOp implements Oper private Output out; - private ApplyAdagradDa(Operation operation) { - super(operation); + public ApplyAdagradDa(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -142,6 +148,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyAdagradDa.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java index f26f0ca7e82..61b3e118bee 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAdagradV2.OP_NAME, + inputsClass = ApplyAdagradV2.Inputs.class +) public final class ApplyAdagradV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -45,8 +51,8 @@ public final class ApplyAdagradV2 extends RawOp implements Oper private Output out; - private ApplyAdagradV2(Operation operation) { - super(operation); + public ApplyAdagradV2(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -160,6 +166,9 @@ public Options updateSlots(Boolean updateSlots) { } } + @OpInputsMetadata( + outputsClass = ApplyAdagradV2.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdam.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdam.java index ae419bd1640..689acdd1364 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdam.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdam.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAdam.OP_NAME, + inputsClass = ApplyAdam.Inputs.class +) @Operator( group = "train" ) @@ -51,8 +57,8 @@ public final class ApplyAdam extends RawOp implements Operand out; - private ApplyAdam(Operation operation) { - super(operation); + public ApplyAdam(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -177,6 +183,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ApplyAdam.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAddSign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAddSign.java index 5eb80c556c3..8daf7e8ee25 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAddSign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAddSign.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyAddSign.OP_NAME, + inputsClass = ApplyAddSign.Inputs.class +) @Operator( group = "train" ) @@ -50,8 +56,8 @@ public final class ApplyAddSign extends RawOp implements Operan private Output out; - private ApplyAddSign(Operation operation) { - super(operation); + public ApplyAddSign(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyAddSign.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyCenteredRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyCenteredRmsProp.java index 5872ccce7b1..e6dd1402cb2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyCenteredRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyCenteredRmsProp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -50,6 +52,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyCenteredRmsProp.OP_NAME, + inputsClass = ApplyCenteredRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -61,8 +67,8 @@ public final class ApplyCenteredRmsProp extends RawOp implement private Output out; - private ApplyCenteredRmsProp(Operation operation) { - super(operation); + public ApplyCenteredRmsProp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -159,6 +165,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyCenteredRmsProp.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyFtrl.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyFtrl.java index af10fab7ed5..c977bdb6ce2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyFtrl.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyFtrl.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyFtrl.OP_NAME, + inputsClass = ApplyFtrl.Inputs.class +) @Operator( group = "train" ) @@ -54,8 +60,8 @@ public final class ApplyFtrl extends RawOp implements Operand out; - private ApplyFtrl(Operation operation) { - super(operation); + public ApplyFtrl(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -178,6 +184,9 @@ public Options multiplyLinearByLr(Boolean multiplyLinearByLr) { } } + @OpInputsMetadata( + outputsClass = ApplyFtrl.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyGradientDescent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyGradientDescent.java index 0e459aefa68..dd732971803 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyGradientDescent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyGradientDescent.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyGradientDescent.OP_NAME, + inputsClass = ApplyGradientDescent.Inputs.class +) @Operator( group = "train" ) @@ -47,8 +53,8 @@ public final class ApplyGradientDescent extends RawOp implement private Output out; - private ApplyGradientDescent(Operation operation) { - super(operation); + public ApplyGradientDescent(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -130,6 +136,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyGradientDescent.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyMomentum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyMomentum.java index cf22b9bcbed..024f1d554ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyMomentum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyMomentum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyMomentum.OP_NAME, + inputsClass = ApplyMomentum.Inputs.class +) @Operator( group = "train" ) @@ -50,8 +56,8 @@ public final class ApplyMomentum extends RawOp implements Opera private Output out; - private ApplyMomentum(Operation operation) { - super(operation); + public ApplyMomentum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -169,6 +175,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ApplyMomentum.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyPowerSign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyPowerSign.java index 88fd9980c94..6c0d708ac35 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyPowerSign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyPowerSign.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyPowerSign.OP_NAME, + inputsClass = ApplyPowerSign.Inputs.class +) @Operator( group = "train" ) @@ -50,8 +56,8 @@ public final class ApplyPowerSign extends RawOp implements Oper private Output out; - private ApplyPowerSign(Operation operation) { - super(operation); + public ApplyPowerSign(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -144,6 +150,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyPowerSign.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalAdagrad.java index aa2dc292c1a..f9307774264 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalAdagrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyProximalAdagrad.OP_NAME, + inputsClass = ApplyProximalAdagrad.Inputs.class +) @Operator( group = "train" ) @@ -50,8 +56,8 @@ public final class ApplyProximalAdagrad extends RawOp implement private Output out; - private ApplyProximalAdagrad(Operation operation) { - super(operation); + public ApplyProximalAdagrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -140,6 +146,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyProximalAdagrad.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalGradientDescent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalGradientDescent.java index d1c6ab76cea..b30f1c5401d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalGradientDescent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyProximalGradientDescent.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyProximalGradientDescent.OP_NAME, + inputsClass = ApplyProximalGradientDescent.Inputs.class +) @Operator( group = "train" ) @@ -49,8 +55,8 @@ public final class ApplyProximalGradientDescent extends RawOp i private Output out; - private ApplyProximalGradientDescent(Operation operation) { - super(operation); + public ApplyProximalGradientDescent(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -137,6 +143,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyProximalGradientDescent.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyRmsProp.java index 3c9ad0c5087..7bd5e6b2111 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyRmsProp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +46,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = ApplyRmsProp.OP_NAME, + inputsClass = ApplyRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -55,8 +61,8 @@ public final class ApplyRmsProp extends RawOp implements Operan private Output out; - private ApplyRmsProp(Operation operation) { - super(operation); + public ApplyRmsProp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -151,6 +157,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ApplyRmsProp.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/BatchMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/BatchMatMul.java index 2efb6c587ac..50ec0d48706 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/BatchMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/BatchMatMul.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -57,6 +59,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = BatchMatMul.OP_NAME, + inputsClass = BatchMatMul.Inputs.class +) @Operator( group = "train" ) @@ -68,8 +74,8 @@ public final class BatchMatMul extends RawOp implements Operand private Output output; - private BatchMatMul(Operation operation) { - super(operation); + public BatchMatMul(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -175,6 +181,9 @@ public Options adjY(Boolean adjY) { } } + @OpInputsMetadata( + outputsClass = BatchMatMul.class + ) public static class Inputs extends RawOpInputs> { /** * 2-D or higher with shape {@code [..., r_x, c_x]}. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java index ef4a34b1a9a..4912f05ce64 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; /** * Computes the static batch size of a dataset sans partial batches. */ +@OpMetadata( + opType = ComputeBatchSize.OP_NAME, + inputsClass = ComputeBatchSize.Inputs.class +) public final class ComputeBatchSize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class ComputeBatchSize extends RawOp implements Operand { private Output batchSize; - private ComputeBatchSize(Operation operation) { - super(operation); + public ComputeBatchSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; batchSize = operation.output(outputIdx++); } @@ -77,6 +83,9 @@ public Output asOutput() { return batchSize; } + @OpInputsMetadata( + outputsClass = ComputeBatchSize.class + ) public static class Inputs extends RawOpInputs { /** * The inputDataset input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ConditionalAccumulator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ConditionalAccumulator.java index ae58b09075e..c20b316f1ef 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ConditionalAccumulator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ConditionalAccumulator.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -43,6 +45,10 @@ * resets the aggregate to 0, and increments the global_step recorded by * the accumulator. */ +@OpMetadata( + opType = ConditionalAccumulator.OP_NAME, + inputsClass = ConditionalAccumulator.Inputs.class +) @Operator( group = "train" ) @@ -54,8 +60,8 @@ public final class ConditionalAccumulator extends RawOp implements Operand handle; - private ConditionalAccumulator(Operation operation) { - super(operation); + public ConditionalAccumulator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -189,6 +195,9 @@ public Options reductionType(String reductionType) { } } + @OpInputsMetadata( + outputsClass = ConditionalAccumulator.class + ) public static class Inputs extends RawOpInputs { /** * The type of the value being accumulated. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/GenerateVocabRemapping.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/GenerateVocabRemapping.java index d5246b11201..0ee41c32d14 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/GenerateVocabRemapping.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/GenerateVocabRemapping.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -59,6 +61,10 @@ * use the corresponding index_table_from_file() as the FeatureColumn framework * does (as opposed to tf.feature_to_id(), which uses a CuckooTable). */ +@OpMetadata( + opType = GenerateVocabRemapping.OP_NAME, + inputsClass = GenerateVocabRemapping.Inputs.class +) @Operator( group = "train" ) @@ -72,8 +78,8 @@ public final class GenerateVocabRemapping extends RawOp { private Output numPresent; - private GenerateVocabRemapping(Operation operation) { - super(operation); + public GenerateVocabRemapping(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; remapping = operation.output(outputIdx++); numPresent = operation.output(outputIdx++); @@ -163,6 +169,9 @@ public Options oldVocabSize(Long oldVocabSize) { } } + @OpInputsMetadata( + outputsClass = GenerateVocabRemapping.class + ) public static class Inputs extends RawOpInputs { /** * Path to the new vocab file. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/MergeV2Checkpoints.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/MergeV2Checkpoints.java index 2d45a0379dc..38256a613e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/MergeV2Checkpoints.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/MergeV2Checkpoints.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; @@ -38,6 +40,10 @@ * path in the input checkpoint_prefixes. This is useful when those paths are non * user-facing temporary locations. */ +@OpMetadata( + opType = MergeV2Checkpoints.OP_NAME, + inputsClass = MergeV2Checkpoints.Inputs.class +) @Operator( group = "train" ) @@ -47,8 +53,8 @@ public final class MergeV2Checkpoints extends RawOp { */ public static final String OP_NAME = "MergeV2Checkpoints"; - private MergeV2Checkpoints(Operation operation) { - super(operation); + public MergeV2Checkpoints(Operation operation) { + super(operation, OP_NAME); } /** @@ -110,6 +116,9 @@ public Options deleteOldDirs(Boolean deleteOldDirs) { } } + @OpInputsMetadata( + outputsClass = MergeV2Checkpoints.class + ) public static class Inputs extends RawOpInputs { /** * prefixes of V2 checkpoints to merge. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/NegTrain.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/NegTrain.java index 491a6c2da92..a997c7064ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/NegTrain.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/NegTrain.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -34,6 +36,10 @@ /** * Training via negative sampling. */ +@OpMetadata( + opType = NegTrain.OP_NAME, + inputsClass = NegTrain.Inputs.class +) @Operator( group = "train" ) @@ -43,8 +49,8 @@ public final class NegTrain extends RawOp { */ public static final String OP_NAME = "NegTrain"; - private NegTrain(Operation operation) { - super(operation); + public NegTrain(Operation operation) { + super(operation, OP_NAME); } /** @@ -81,6 +87,9 @@ public static NegTrain create(Scope scope, Operand wIn, Operand { /** * input word embedding. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/PreventGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/PreventGradient.java index 884fffc0487..3d9900c40ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/PreventGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/PreventGradient.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = PreventGradient.OP_NAME, + inputsClass = PreventGradient.Inputs.class +) @Operator( group = "train" ) @@ -53,8 +59,8 @@ public final class PreventGradient extends RawOp implements Ope private Output output; - private PreventGradient(Operation operation) { - super(operation); + public PreventGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -132,6 +138,9 @@ public Options message(String message) { } } + @OpInputsMetadata( + outputsClass = PreventGradient.class + ) public static class Inputs extends RawOpInputs> { /** * any tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java index db829374d76..6a425a7d10f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * Applies a gradient to a given accumulator. * Does not add if local_step is lesser than the accumulator's global_step. */ +@OpMetadata( + opType = ResourceAccumulatorApplyGradient.OP_NAME, + inputsClass = ResourceAccumulatorApplyGradient.Inputs.class +) public final class ResourceAccumulatorApplyGradient extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ResourceAccumulatorApplyGradient"; - private ResourceAccumulatorApplyGradient(Operation operation) { - super(operation); + public ResourceAccumulatorApplyGradient(Operation operation) { + super(operation, OP_NAME); } /** @@ -66,6 +72,9 @@ public static ResourceAccumulatorApplyGradient create(Scope scope, return new ResourceAccumulatorApplyGradient(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceAccumulatorApplyGradient.class + ) public static class Inputs extends RawOpInputs { /** * The handle to a accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java index c9cc0a3c4a2..c5e181665f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; /** * Returns the number of gradients aggregated in the given accumulators. */ +@OpMetadata( + opType = ResourceAccumulatorNumAccumulated.OP_NAME, + inputsClass = ResourceAccumulatorNumAccumulated.Inputs.class +) public final class ResourceAccumulatorNumAccumulated extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -41,8 +47,8 @@ public final class ResourceAccumulatorNumAccumulated extends RawOp implements Op private Output numAccumulated; - private ResourceAccumulatorNumAccumulated(Operation operation) { - super(operation); + public ResourceAccumulatorNumAccumulated(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; numAccumulated = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return numAccumulated; } + @OpInputsMetadata( + outputsClass = ResourceAccumulatorNumAccumulated.class + ) public static class Inputs extends RawOpInputs { /** * The handle to an accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java index 23141dd5d10..226492c11c3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * Logs warning if the accumulator's value is already higher than * new_global_step. */ +@OpMetadata( + opType = ResourceAccumulatorSetGlobalStep.OP_NAME, + inputsClass = ResourceAccumulatorSetGlobalStep.Inputs.class +) public final class ResourceAccumulatorSetGlobalStep extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ResourceAccumulatorSetGlobalStep"; - private ResourceAccumulatorSetGlobalStep(Operation operation) { - super(operation); + public ResourceAccumulatorSetGlobalStep(Operation operation) { + super(operation, OP_NAME); } /** @@ -63,6 +69,9 @@ public static ResourceAccumulatorSetGlobalStep create(Scope scope, return new ResourceAccumulatorSetGlobalStep(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = ResourceAccumulatorSetGlobalStep.class + ) public static class Inputs extends RawOpInputs { /** * The handle to an accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java index 1cfc15beb3c..cbe1227ee7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code average} output */ +@OpMetadata( + opType = ResourceAccumulatorTakeGradient.OP_NAME, + inputsClass = ResourceAccumulatorTakeGradient.Inputs.class +) public final class ResourceAccumulatorTakeGradient extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -50,8 +56,8 @@ public final class ResourceAccumulatorTakeGradient extends RawO private Output average; - private ResourceAccumulatorTakeGradient(Operation operation) { - super(operation); + public ResourceAccumulatorTakeGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; average = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return average; } + @OpInputsMetadata( + outputsClass = ResourceAccumulatorTakeGradient.class + ) public static class Inputs extends RawOpInputs> { /** * The handle to an accumulator. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java index 336031fd4ae..2323ac98843 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,14 +37,18 @@ * v_t <- max(beta2 * v_{t-1}, abs(g)) * variable <- variable - learning_rate / (1 - beta1^t) * m_t / (v_t + epsilon) */ +@OpMetadata( + opType = ResourceApplyAdaMax.OP_NAME, + inputsClass = ResourceApplyAdaMax.Inputs.class +) public final class ResourceApplyAdaMax extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ResourceApplyAdaMax"; - private ResourceApplyAdaMax(Operation operation) { - super(operation); + public ResourceApplyAdaMax(Operation operation) { + super(operation, OP_NAME); } /** @@ -124,6 +130,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAdaMax.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdadelta.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdadelta.java index 11c2202c18b..2b2a9be5d92 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdadelta.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdadelta.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * update_accum = rho() * update_accum + (1 - rho()) * update.square(); * var -= update; */ +@OpMetadata( + opType = ResourceApplyAdadelta.OP_NAME, + inputsClass = ResourceApplyAdadelta.Inputs.class +) @Operator( group = "train" ) @@ -46,8 +52,8 @@ public final class ResourceApplyAdadelta extends RawOp { */ public static final String OP_NAME = "ResourceApplyAdadelta"; - private ResourceApplyAdadelta(Operation operation) { - super(operation); + public ResourceApplyAdadelta(Operation operation) { + super(operation, OP_NAME); } /** @@ -123,6 +129,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAdadelta.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java index 29042e8ccd3..512af640a14 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -34,14 +36,18 @@ * accum += grad * grad * var -= lr * grad * (1 / (sqrt(accum) + epsilon)) */ +@OpMetadata( + opType = ResourceApplyAdagrad.OP_NAME, + inputsClass = ResourceApplyAdagrad.Inputs.class +) public final class ResourceApplyAdagrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ResourceApplyAdagradV2"; - private ResourceApplyAdagrad(Operation operation) { - super(operation); + public ResourceApplyAdagrad(Operation operation) { + super(operation, OP_NAME); } /** @@ -140,6 +146,9 @@ public Options updateSlots(Boolean updateSlots) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAdagrad.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagradDa.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagradDa.java index b153e0c2cf6..f2d80bafeac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagradDa.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagradDa.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -34,6 +36,10 @@ /** * Update '*var' according to the proximal adagrad scheme. */ +@OpMetadata( + opType = ResourceApplyAdagradDa.OP_NAME, + inputsClass = ResourceApplyAdagradDa.Inputs.class +) @Operator( group = "train" ) @@ -43,8 +49,8 @@ public final class ResourceApplyAdagradDa extends RawOp { */ public static final String OP_NAME = "ResourceApplyAdagradDA"; - private ResourceApplyAdagradDa(Operation operation) { - super(operation); + public ResourceApplyAdagradDa(Operation operation) { + super(operation, OP_NAME); } /** @@ -122,6 +128,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAdagradDa.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdam.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdam.java index d58a5e916f1..a0fa6645311 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdam.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdam.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -37,6 +39,10 @@ * $$v_t := \beta_2 * v_{t-1} + (1 - \beta_2) * g * g$$ * $$\text{variable} := \text{variable} - \text{lr}_t * m_t / (\sqrt{v_t} + \epsilon)$$ */ +@OpMetadata( + opType = ResourceApplyAdam.OP_NAME, + inputsClass = ResourceApplyAdam.Inputs.class +) @Operator( group = "train" ) @@ -46,8 +52,8 @@ public final class ResourceApplyAdam extends RawOp { */ public static final String OP_NAME = "ResourceApplyAdam"; - private ResourceApplyAdam(Operation operation) { - super(operation); + public ResourceApplyAdam(Operation operation) { + super(operation, OP_NAME); } /** @@ -157,6 +163,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAdam.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdamWithAmsgrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdamWithAmsgrad.java index 38df623446b..ad21624b125 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdamWithAmsgrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdamWithAmsgrad.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * $$\hat{v}t := max{\hat{v}{t-1}, v_t}$$ * $$\text{variable} := \text{variable} - \text{lr}_t * m_t / (\sqrt{\hat{v}_t} + \epsilon)$$ */ +@OpMetadata( + opType = ResourceApplyAdamWithAmsgrad.OP_NAME, + inputsClass = ResourceApplyAdamWithAmsgrad.Inputs.class +) @Operator( group = "train" ) @@ -47,8 +53,8 @@ public final class ResourceApplyAdamWithAmsgrad extends RawOp { */ public static final String OP_NAME = "ResourceApplyAdamWithAmsgrad"; - private ResourceApplyAdamWithAmsgrad(Operation operation) { - super(operation); + public ResourceApplyAdamWithAmsgrad(Operation operation) { + super(operation, OP_NAME); } /** @@ -134,6 +140,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAdamWithAmsgrad.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAddSign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAddSign.java index d034cd3bf0d..250296eb0cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAddSign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAddSign.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * update <- (alpha + sign_decay * sign(g) *sign(m)) * g * variable <- variable - lr_t * update */ +@OpMetadata( + opType = ResourceApplyAddSign.OP_NAME, + inputsClass = ResourceApplyAddSign.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class ResourceApplyAddSign extends RawOp { */ public static final String OP_NAME = "ResourceApplyAddSign"; - private ResourceApplyAddSign(Operation operation) { - super(operation); + public ResourceApplyAddSign(Operation operation) { + super(operation, OP_NAME); } /** @@ -123,6 +129,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyAddSign.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyCenteredRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyCenteredRmsProp.java index 55ce63a208c..ae523e0de04 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyCenteredRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyCenteredRmsProp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -47,6 +49,10 @@ * mom <- momentum * mom_{t-1} + lr * grad / sqrt(ms - mg * mg + epsilon) * var <- var - mom */ +@OpMetadata( + opType = ResourceApplyCenteredRmsProp.OP_NAME, + inputsClass = ResourceApplyCenteredRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -56,8 +62,8 @@ public final class ResourceApplyCenteredRmsProp extends RawOp { */ public static final String OP_NAME = "ResourceApplyCenteredRMSProp"; - private ResourceApplyCenteredRmsProp(Operation operation) { - super(operation); + public ResourceApplyCenteredRmsProp(Operation operation) { + super(operation, OP_NAME); } /** @@ -139,6 +145,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyCenteredRmsProp.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyFtrl.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyFtrl.java index 8d3540eda02..66b8fbb0c4a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyFtrl.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyFtrl.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * var = (sign(linear) * l1 - linear) / quadratic if |linear| > l1 else 0.0 * accum = accum_new */ +@OpMetadata( + opType = ResourceApplyFtrl.OP_NAME, + inputsClass = ResourceApplyFtrl.Inputs.class +) @Operator( group = "train" ) @@ -49,8 +55,8 @@ public final class ResourceApplyFtrl extends RawOp { */ public static final String OP_NAME = "ResourceApplyFtrlV2"; - private ResourceApplyFtrl(Operation operation) { - super(operation); + public ResourceApplyFtrl(Operation operation) { + super(operation, OP_NAME); } /** @@ -158,6 +164,9 @@ public Options multiplyLinearByLr(Boolean multiplyLinearByLr) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyFtrl.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyGradientDescent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyGradientDescent.java index f33601521df..c5d9596e87c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyGradientDescent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyGradientDescent.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -33,6 +35,10 @@ /** * Update '*var' by subtracting 'alpha' * 'delta' from it. */ +@OpMetadata( + opType = ResourceApplyGradientDescent.OP_NAME, + inputsClass = ResourceApplyGradientDescent.Inputs.class +) @Operator( group = "train" ) @@ -42,8 +48,8 @@ public final class ResourceApplyGradientDescent extends RawOp { */ public static final String OP_NAME = "ResourceApplyGradientDescent"; - private ResourceApplyGradientDescent(Operation operation) { - super(operation); + public ResourceApplyGradientDescent(Operation operation) { + super(operation, OP_NAME); } /** @@ -109,6 +115,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyGradientDescent.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyKerasMomentum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyKerasMomentum.java index 39edfec6216..c5a0f246da4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyKerasMomentum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyKerasMomentum.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ *

    accum = accum * momentum - lr * grad * var += accum */ +@OpMetadata( + opType = ResourceApplyKerasMomentum.OP_NAME, + inputsClass = ResourceApplyKerasMomentum.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class ResourceApplyKerasMomentum extends RawOp { */ public static final String OP_NAME = "ResourceApplyKerasMomentum"; - private ResourceApplyKerasMomentum(Operation operation) { - super(operation); + public ResourceApplyKerasMomentum(Operation operation) { + super(operation, OP_NAME); } /** @@ -149,6 +155,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyKerasMomentum.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyMomentum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyMomentum.java index 6bb224aa4e4..432fbdad8c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyMomentum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyMomentum.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ *

    accum = accum * momentum + grad * var -= lr * accum */ +@OpMetadata( + opType = ResourceApplyMomentum.OP_NAME, + inputsClass = ResourceApplyMomentum.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class ResourceApplyMomentum extends RawOp { */ public static final String OP_NAME = "ResourceApplyMomentum"; - private ResourceApplyMomentum(Operation operation) { - super(operation); + public ResourceApplyMomentum(Operation operation) { + super(operation, OP_NAME); } /** @@ -149,6 +155,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyMomentum.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyPowerSign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyPowerSign.java index 3df2066af31..535cc617988 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyPowerSign.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyPowerSign.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * update <- exp(logbase * sign_decay * sign(g) * sign(m_t)) * g * variable <- variable - lr_t * update */ +@OpMetadata( + opType = ResourceApplyPowerSign.OP_NAME, + inputsClass = ResourceApplyPowerSign.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class ResourceApplyPowerSign extends RawOp { */ public static final String OP_NAME = "ResourceApplyPowerSign"; - private ResourceApplyPowerSign(Operation operation) { - super(operation); + public ResourceApplyPowerSign(Operation operation) { + super(operation, OP_NAME); } /** @@ -123,6 +129,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyPowerSign.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalAdagrad.java index 5e251235af1..6805a4b1a0c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalAdagrad.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * prox_v = var - lr * grad * (1 / sqrt(accum)) * var = sign(prox_v)/(1+lrl2) * max{|prox_v|-lrl1,0} */ +@OpMetadata( + opType = ResourceApplyProximalAdagrad.OP_NAME, + inputsClass = ResourceApplyProximalAdagrad.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class ResourceApplyProximalAdagrad extends RawOp { */ public static final String OP_NAME = "ResourceApplyProximalAdagrad"; - private ResourceApplyProximalAdagrad(Operation operation) { - super(operation); + public ResourceApplyProximalAdagrad(Operation operation) { + super(operation, OP_NAME); } /** @@ -119,6 +125,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyProximalAdagrad.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalGradientDescent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalGradientDescent.java index 1a3da62fbf4..e635dbe42c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalGradientDescent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyProximalGradientDescent.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * prox_v = var - alpha * delta * var = sign(prox_v)/(1+alphal2) * max{|prox_v|-alphal1,0} */ +@OpMetadata( + opType = ResourceApplyProximalGradientDescent.OP_NAME, + inputsClass = ResourceApplyProximalGradientDescent.Inputs.class +) @Operator( group = "train" ) @@ -44,8 +50,8 @@ public final class ResourceApplyProximalGradientDescent extends RawOp { */ public static final String OP_NAME = "ResourceApplyProximalGradientDescent"; - private ResourceApplyProximalGradientDescent(Operation operation) { - super(operation); + public ResourceApplyProximalGradientDescent(Operation operation) { + super(operation, OP_NAME); } /** @@ -116,6 +122,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyProximalGradientDescent.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyRmsProp.java index 6d2fe91d598..5d2b0aa662f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyRmsProp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +43,10 @@ * mom <- momentum * mom_{t-1} + lr * grad / sqrt(ms + epsilon) * var <- var - mom */ +@OpMetadata( + opType = ResourceApplyRmsProp.OP_NAME, + inputsClass = ResourceApplyRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -50,8 +56,8 @@ public final class ResourceApplyRmsProp extends RawOp { */ public static final String OP_NAME = "ResourceApplyRMSProp"; - private ResourceApplyRmsProp(Operation operation) { - super(operation); + public ResourceApplyRmsProp(Operation operation) { + super(operation, OP_NAME); } /** @@ -131,6 +137,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceApplyRmsProp.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java index 4cc51c55289..d0a4c09b894 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +45,10 @@ * This is a resource version of ConditionalAccumulator that will work in TF2.0 * with tf.cond version 2. */ +@OpMetadata( + opType = ResourceConditionalAccumulator.OP_NAME, + inputsClass = ResourceConditionalAccumulator.Inputs.class +) public final class ResourceConditionalAccumulator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -52,8 +58,8 @@ public final class ResourceConditionalAccumulator extends RawOp implements Opera private Output handle; @SuppressWarnings("unchecked") - private ResourceConditionalAccumulator(Operation operation) { - super(operation); + public ResourceConditionalAccumulator(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; handle = operation.output(outputIdx++); } @@ -188,6 +194,9 @@ public Options reductionType(String reductionType) { } } + @OpInputsMetadata( + outputsClass = ResourceConditionalAccumulator.class + ) public static class Inputs extends RawOpInputs { /** * The type of the value being accumulated. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdadelta.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdadelta.java index 01f0a739f36..5c9b4640376 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdadelta.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdadelta.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -34,6 +36,10 @@ /** * var: Should be from a Variable(). */ +@OpMetadata( + opType = ResourceSparseApplyAdadelta.OP_NAME, + inputsClass = ResourceSparseApplyAdadelta.Inputs.class +) @Operator( group = "train" ) @@ -43,8 +49,8 @@ public final class ResourceSparseApplyAdadelta extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyAdadelta"; - private ResourceSparseApplyAdadelta(Operation operation) { - super(operation); + public ResourceSparseApplyAdadelta(Operation operation) { + super(operation, OP_NAME); } /** @@ -122,6 +128,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyAdadelta.class + ) public static class Inputs extends RawOpInputs { /** * The var input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagrad.java index aaa5be414ea..8de948f6d86 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagrad.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * accum += grad * grad * var -= lr * grad * (1 / sqrt(accum)) */ +@OpMetadata( + opType = ResourceSparseApplyAdagrad.OP_NAME, + inputsClass = ResourceSparseApplyAdagrad.Inputs.class +) @Operator( group = "train" ) @@ -46,8 +52,8 @@ public final class ResourceSparseApplyAdagrad extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyAdagrad"; - private ResourceSparseApplyAdagrad(Operation operation) { - super(operation); + public ResourceSparseApplyAdagrad(Operation operation) { + super(operation, OP_NAME); } /** @@ -146,6 +152,9 @@ public Options updateSlots(Boolean updateSlots) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyAdagrad.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradDa.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradDa.java index 6a410002f3a..bd748830dde 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradDa.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradDa.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -35,6 +37,10 @@ /** * Update entries in '*var' and '*accum' according to the proximal adagrad scheme. */ +@OpMetadata( + opType = ResourceSparseApplyAdagradDa.OP_NAME, + inputsClass = ResourceSparseApplyAdagradDa.Inputs.class +) @Operator( group = "train" ) @@ -44,8 +50,8 @@ public final class ResourceSparseApplyAdagradDa extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyAdagradDA"; - private ResourceSparseApplyAdagradDa(Operation operation) { - super(operation); + public ResourceSparseApplyAdagradDa(Operation operation) { + super(operation, OP_NAME); } /** @@ -126,6 +132,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyAdagradDa.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java index 423505a39b5..594f0e4a1fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -36,14 +38,18 @@ * accum += grad * grad * var -= lr * grad * (1 / sqrt(accum)) */ +@OpMetadata( + opType = ResourceSparseApplyAdagradV2.OP_NAME, + inputsClass = ResourceSparseApplyAdagradV2.Inputs.class +) public final class ResourceSparseApplyAdagradV2 extends RawOp { /** * The name of this op, as known by TensorFlow core engine */ public static final String OP_NAME = "ResourceSparseApplyAdagradV2"; - private ResourceSparseApplyAdagradV2(Operation operation) { - super(operation); + public ResourceSparseApplyAdagradV2(Operation operation) { + super(operation, OP_NAME); } /** @@ -144,6 +150,9 @@ public Options updateSlots(Boolean updateSlots) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyAdagradV2.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyCenteredRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyCenteredRmsProp.java index 2789f2fb191..36f3d3f1aa8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyCenteredRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyCenteredRmsProp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -47,6 +49,10 @@ * mom <- momentum * mom_{t-1} + lr * grad / sqrt(ms + epsilon) * var <- var - mom */ +@OpMetadata( + opType = ResourceSparseApplyCenteredRmsProp.OP_NAME, + inputsClass = ResourceSparseApplyCenteredRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -56,8 +62,8 @@ public final class ResourceSparseApplyCenteredRmsProp extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyCenteredRMSProp"; - private ResourceSparseApplyCenteredRmsProp(Operation operation) { - super(operation); + public ResourceSparseApplyCenteredRmsProp(Operation operation) { + super(operation, OP_NAME); } /** @@ -141,6 +147,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyCenteredRmsProp.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyFtrl.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyFtrl.java index cfa21d59dc0..2e76781ef93 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyFtrl.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyFtrl.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * var = (sign(linear) * l1 - linear) / quadratic if |linear| > l1 else 0.0 * accum = accum_new */ +@OpMetadata( + opType = ResourceSparseApplyFtrl.OP_NAME, + inputsClass = ResourceSparseApplyFtrl.Inputs.class +) @Operator( group = "train" ) @@ -51,8 +57,8 @@ public final class ResourceSparseApplyFtrl extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyFtrlV2"; - private ResourceSparseApplyFtrl(Operation operation) { - super(operation); + public ResourceSparseApplyFtrl(Operation operation) { + super(operation, OP_NAME); } /** @@ -162,6 +168,9 @@ public Options multiplyLinearByLr(Boolean multiplyLinearByLr) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyFtrl.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyKerasMomentum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyKerasMomentum.java index 94c2ad9bd17..50a6fbbf44d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyKerasMomentum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyKerasMomentum.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ *

    accum = accum * momentum - lr * grad * var += accum */ +@OpMetadata( + opType = ResourceSparseApplyKerasMomentum.OP_NAME, + inputsClass = ResourceSparseApplyKerasMomentum.Inputs.class +) @Operator( group = "train" ) @@ -47,8 +53,8 @@ public final class ResourceSparseApplyKerasMomentum extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyKerasMomentum"; - private ResourceSparseApplyKerasMomentum(Operation operation) { - super(operation); + public ResourceSparseApplyKerasMomentum(Operation operation) { + super(operation, OP_NAME); } /** @@ -153,6 +159,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyKerasMomentum.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyMomentum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyMomentum.java index 1b52c885e67..ab6a217555d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyMomentum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyMomentum.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ *

    accum = accum * momentum + grad * var -= lr * accum */ +@OpMetadata( + opType = ResourceSparseApplyMomentum.OP_NAME, + inputsClass = ResourceSparseApplyMomentum.Inputs.class +) @Operator( group = "train" ) @@ -47,8 +53,8 @@ public final class ResourceSparseApplyMomentum extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyMomentum"; - private ResourceSparseApplyMomentum(Operation operation) { - super(operation); + public ResourceSparseApplyMomentum(Operation operation) { + super(operation, OP_NAME); } /** @@ -153,6 +159,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyMomentum.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalAdagrad.java index 57b484d3b21..c89159095cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalAdagrad.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * prox_v -= lr * grad * (1 / sqrt(accum)) * var = sign(prox_v)/(1+lrl2) * max{|prox_v|-lrl1,0} */ +@OpMetadata( + opType = ResourceSparseApplyProximalAdagrad.OP_NAME, + inputsClass = ResourceSparseApplyProximalAdagrad.Inputs.class +) @Operator( group = "train" ) @@ -48,8 +54,8 @@ public final class ResourceSparseApplyProximalAdagrad extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyProximalAdagrad"; - private ResourceSparseApplyProximalAdagrad(Operation operation) { - super(operation); + public ResourceSparseApplyProximalAdagrad(Operation operation) { + super(operation, OP_NAME); } /** @@ -124,6 +130,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyProximalAdagrad.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalGradientDescent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalGradientDescent.java index 2d266131daf..2fcd22d2bf4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalGradientDescent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyProximalGradientDescent.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * prox_v = var - alpha * grad * var = sign(prox_v)/(1+alphal2) * max{|prox_v|-alphal1,0} */ +@OpMetadata( + opType = ResourceSparseApplyProximalGradientDescent.OP_NAME, + inputsClass = ResourceSparseApplyProximalGradientDescent.Inputs.class +) @Operator( group = "train" ) @@ -46,8 +52,8 @@ public final class ResourceSparseApplyProximalGradientDescent extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyProximalGradientDescent"; - private ResourceSparseApplyProximalGradientDescent(Operation operation) { - super(operation); + public ResourceSparseApplyProximalGradientDescent(Operation operation) { + super(operation, OP_NAME); } /** @@ -120,6 +126,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyProximalGradientDescent.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyRmsProp.java index ee73c441b8f..149a8003b46 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyRmsProp.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * mom <- momentum * mom_{t-1} + lr * grad / sqrt(ms + epsilon) * var <- var - mom */ +@OpMetadata( + opType = ResourceSparseApplyRmsProp.OP_NAME, + inputsClass = ResourceSparseApplyRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -51,8 +57,8 @@ public final class ResourceSparseApplyRmsProp extends RawOp { */ public static final String OP_NAME = "ResourceSparseApplyRMSProp"; - private ResourceSparseApplyRmsProp(Operation operation) { - super(operation); + public ResourceSparseApplyRmsProp(Operation operation) { + super(operation, OP_NAME); } /** @@ -134,6 +140,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = ResourceSparseApplyRmsProp.class + ) public static class Inputs extends RawOpInputs { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Restore.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Restore.java index 07682b56434..e2c180213e2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Restore.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Restore.java @@ -30,6 +30,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -51,6 +53,10 @@ * strings and correspondingly well-formed. *

    Callers must ensure all the named tensors are indeed stored in the checkpoint. */ +@OpMetadata( + opType = Restore.OP_NAME, + inputsClass = Restore.Inputs.class +) @Operator( group = "train" ) @@ -63,8 +69,8 @@ public final class Restore extends RawOp implements Iterable> { private List> tensors; @SuppressWarnings("unchecked") - private Restore(Operation operation) { - super(operation); + public Restore(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int tensorsLength = operation.outputListLength("tensors"); tensors = Arrays.asList(operation.outputList(outputIdx, tensorsLength)); @@ -112,6 +118,9 @@ public Iterator> iterator() { return (Iterator) tensors.iterator(); } + @OpInputsMetadata( + outputsClass = Restore.class + ) public static class Inputs extends RawOpInputs { /** * Must have a single element. The prefix of a V2 checkpoint. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/RestoreSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/RestoreSlice.java index ee5496926cc..0237357a838 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/RestoreSlice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/RestoreSlice.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -43,6 +45,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = RestoreSlice.OP_NAME, + inputsClass = RestoreSlice.Inputs.class +) @Operator( group = "train" ) @@ -54,8 +60,8 @@ public final class RestoreSlice extends RawOp implements Operan private Output tensor; - private RestoreSlice(Operation operation) { - super(operation); + public RestoreSlice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tensor = operation.output(outputIdx++); } @@ -143,6 +149,9 @@ public Options preferredShard(Long preferredShard) { } } + @OpInputsMetadata( + outputsClass = RestoreSlice.class + ) public static class Inputs extends RawOpInputs> { /** * Must have a single element. The pattern of the files from diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Save.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Save.java index 57b0d9dc6df..d0386275ef4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Save.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/Save.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -37,6 +39,10 @@ * specific slices of full tensors, "shape_and_slices" should be non-empty strings * and correspondingly well-formed. */ +@OpMetadata( + opType = Save.OP_NAME, + inputsClass = Save.Inputs.class +) @Operator( group = "train" ) @@ -46,8 +52,8 @@ public final class Save extends RawOp { */ public static final String OP_NAME = "SaveV2"; - private Save(Operation operation) { - super(operation); + public Save(Operation operation) { + super(operation, OP_NAME); } /** @@ -75,6 +81,9 @@ public static Save create(Scope scope, Operand prefix, Operand return new Save(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = Save.class + ) public static class Inputs extends RawOpInputs { /** * Must have a single element. The prefix of the V2 checkpoint to which we diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SaveSlices.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SaveSlices.java index e1b2ab7325a..ff034d1d6fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SaveSlices.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SaveSlices.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TString; @@ -54,6 +56,10 @@ * *

    See also {@code Save}. */ +@OpMetadata( + opType = SaveSlices.OP_NAME, + inputsClass = SaveSlices.Inputs.class +) @Operator( group = "train" ) @@ -63,8 +69,8 @@ public final class SaveSlices extends RawOp { */ public static final String OP_NAME = "SaveSlices"; - private SaveSlices(Operation operation) { - super(operation); + public SaveSlices(Operation operation) { + super(operation, OP_NAME); } /** @@ -92,6 +98,9 @@ public static SaveSlices create(Scope scope, Operand filename, return new SaveSlices(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = SaveSlices.class + ) public static class Inputs extends RawOpInputs { /** * Must have a single element. The name of the file to which we write the diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaFprint.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaFprint.java index 0d7361ffed2..d37ce836e43 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaFprint.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaFprint.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -34,6 +36,10 @@ /** * Computes fingerprints of the input strings. */ +@OpMetadata( + opType = SdcaFprint.OP_NAME, + inputsClass = SdcaFprint.Inputs.class +) @Operator( group = "train" ) @@ -45,8 +51,8 @@ public final class SdcaFprint extends RawOp implements Operand { private Output output; - private SdcaFprint(Operation operation) { - super(operation); + public SdcaFprint(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -82,6 +88,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SdcaFprint.class + ) public static class Inputs extends RawOpInputs { /** * vector of strings to compute fingerprints on. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java index 3d271080556..83990a329e2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -48,6 +50,10 @@ *

    Stochastic Dual Coordinate Ascent with Adaptive Probabilities .
    * Dominik Csiba, Zheng Qu, Peter Richtarik. 2015 */ +@OpMetadata( + opType = SdcaOptimizer.OP_NAME, + inputsClass = SdcaOptimizer.Inputs.class +) public final class SdcaOptimizer extends RawOp { /** * The name of this op, as known by TensorFlow core engine @@ -61,8 +67,8 @@ public final class SdcaOptimizer extends RawOp { private List> outDeltaDenseWeights; @SuppressWarnings("unchecked") - private SdcaOptimizer(Operation operation) { - super(operation); + public SdcaOptimizer(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outExampleStateData = operation.output(outputIdx++); int outDeltaSparseWeightsLength = operation.outputListLength("out_delta_sparse_weights"); @@ -200,6 +206,9 @@ public Options adaptive(Boolean adaptive) { } } + @OpInputsMetadata( + outputsClass = SdcaOptimizer.class + ) public static class Inputs extends RawOpInputs { /** * a list of vectors which contain example indices. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaShrinkL1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaShrinkL1.java index ac043e371b1..eda1af88933 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaShrinkL1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaShrinkL1.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** * Applies L1 regularization shrink step on the parameters. */ +@OpMetadata( + opType = SdcaShrinkL1.OP_NAME, + inputsClass = SdcaShrinkL1.Inputs.class +) @Operator( group = "train" ) @@ -42,8 +48,8 @@ public final class SdcaShrinkL1 extends RawOp { */ public static final String OP_NAME = "SdcaShrinkL1"; - private SdcaShrinkL1(Operation operation) { - super(operation); + public SdcaShrinkL1(Operation operation) { + super(operation, OP_NAME); } /** @@ -68,6 +74,9 @@ public static SdcaShrinkL1 create(Scope scope, Iterable> weigh return new SdcaShrinkL1(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = SdcaShrinkL1.class + ) public static class Inputs extends RawOpInputs { /** * a list of vectors where each value is the weight associated with a diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdadelta.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdadelta.java index 5bb9b3f3d39..1ae14de5884 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdadelta.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdadelta.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -37,6 +39,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyAdadelta.OP_NAME, + inputsClass = SparseApplyAdadelta.Inputs.class +) @Operator( group = "train" ) @@ -48,8 +54,8 @@ public final class SparseApplyAdadelta extends RawOp implements private Output out; - private SparseApplyAdadelta(Operation operation) { - super(operation); + public SparseApplyAdadelta(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -142,6 +148,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = SparseApplyAdadelta.class + ) public static class Inputs extends RawOpInputs> { /** * The var input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java index 7485978329f..5f243bee63e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyAdagrad.OP_NAME, + inputsClass = SparseApplyAdagrad.Inputs.class +) public final class SparseApplyAdagrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine @@ -47,8 +53,8 @@ public final class SparseApplyAdagrad extends RawOp implements private Output out; - private SparseApplyAdagrad(Operation operation) { - super(operation); + public SparseApplyAdagrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -165,6 +171,9 @@ public Options updateSlots(Boolean updateSlots) { } } + @OpInputsMetadata( + outputsClass = SparseApplyAdagrad.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagradDa.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagradDa.java index b07ad601f9f..7898f903fd9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagradDa.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagradDa.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt64; @@ -38,6 +40,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyAdagradDa.OP_NAME, + inputsClass = SparseApplyAdagradDa.Inputs.class +) @Operator( group = "train" ) @@ -49,8 +55,8 @@ public final class SparseApplyAdagradDa extends RawOp implement private Output out; - private SparseApplyAdagradDa(Operation operation) { - super(operation); + public SparseApplyAdagradDa(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -146,6 +152,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = SparseApplyAdagradDa.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyCenteredRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyCenteredRmsProp.java index 02d36702a5e..bbe031b0f37 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyCenteredRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyCenteredRmsProp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -50,6 +52,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyCenteredRmsProp.OP_NAME, + inputsClass = SparseApplyCenteredRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -61,8 +67,8 @@ public final class SparseApplyCenteredRmsProp extends RawOp imp private Output out; - private SparseApplyCenteredRmsProp(Operation operation) { - super(operation); + public SparseApplyCenteredRmsProp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -162,6 +168,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = SparseApplyCenteredRmsProp.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyFtrl.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyFtrl.java index b093d454dd1..5627f95b0fb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyFtrl.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyFtrl.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -45,6 +47,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyFtrl.OP_NAME, + inputsClass = SparseApplyFtrl.Inputs.class +) @Operator( group = "train" ) @@ -56,8 +62,8 @@ public final class SparseApplyFtrl extends RawOp implements Ope private Output out; - private SparseApplyFtrl(Operation operation) { - super(operation); + public SparseApplyFtrl(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -183,6 +189,9 @@ public Options multiplyLinearByLr(Boolean multiplyLinearByLr) { } } + @OpInputsMetadata( + outputsClass = SparseApplyFtrl.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyMomentum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyMomentum.java index 9f36c04de51..1c78b7c93a7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyMomentum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyMomentum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +43,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyMomentum.OP_NAME, + inputsClass = SparseApplyMomentum.Inputs.class +) @Operator( group = "train" ) @@ -52,8 +58,8 @@ public final class SparseApplyMomentum extends RawOp implements private Output out; - private SparseApplyMomentum(Operation operation) { - super(operation); + public SparseApplyMomentum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -174,6 +180,9 @@ public Options useNesterov(Boolean useNesterov) { } } + @OpInputsMetadata( + outputsClass = SparseApplyMomentum.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalAdagrad.java index d0f3ef64c4d..eff6fe8315b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalAdagrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyProximalAdagrad.OP_NAME, + inputsClass = SparseApplyProximalAdagrad.Inputs.class +) @Operator( group = "train" ) @@ -53,8 +59,8 @@ public final class SparseApplyProximalAdagrad extends RawOp imp private Output out; - private SparseApplyProximalAdagrad(Operation operation) { - super(operation); + public SparseApplyProximalAdagrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -145,6 +151,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = SparseApplyProximalAdagrad.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalGradientDescent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalGradientDescent.java index ee51c62e9bb..15a1696be37 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalGradientDescent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyProximalGradientDescent.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyProximalGradientDescent.OP_NAME, + inputsClass = SparseApplyProximalGradientDescent.Inputs.class +) @Operator( group = "train" ) @@ -51,8 +57,8 @@ public final class SparseApplyProximalGradientDescent extends R private Output out; - private SparseApplyProximalGradientDescent(Operation operation) { - super(operation); + public SparseApplyProximalGradientDescent(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -141,6 +147,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = SparseApplyProximalGradientDescent.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyRmsProp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyRmsProp.java index af67871862b..6cd2563cc64 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyRmsProp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyRmsProp.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -45,6 +47,10 @@ * * @param data type for {@code out} output */ +@OpMetadata( + opType = SparseApplyRmsProp.OP_NAME, + inputsClass = SparseApplyRmsProp.Inputs.class +) @Operator( group = "train" ) @@ -56,8 +62,8 @@ public final class SparseApplyRmsProp extends RawOp implements private Output out; - private SparseApplyRmsProp(Operation operation) { - super(operation); + public SparseApplyRmsProp(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; out = operation.output(outputIdx++); } @@ -154,6 +160,9 @@ public Options useLocking(Boolean useLocking) { } } + @OpInputsMetadata( + outputsClass = SparseApplyRmsProp.class + ) public static class Inputs extends RawOpInputs> { /** * Should be from a Variable(). diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SymbolicGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SymbolicGradient.java index 205771ec423..ff194016743 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SymbolicGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SymbolicGradient.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * Computes the gradient function for function f via backpropagation. */ +@OpMetadata( + opType = SymbolicGradient.OP_NAME, + inputsClass = SymbolicGradient.Inputs.class +) @Operator( group = "train" ) @@ -50,8 +56,8 @@ public final class SymbolicGradient extends RawOp implements Iterable> output; @SuppressWarnings("unchecked") - private SymbolicGradient(Operation operation) { - super(operation); + public SymbolicGradient(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -107,6 +113,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = SymbolicGradient.class + ) public static class Inputs extends RawOpInputs { /** * a list of input tensors of size N + M; diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/TileGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/TileGrad.java index ccfe689fd89..86cd960293c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/TileGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/TileGrad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = TileGrad.OP_NAME, + inputsClass = TileGrad.Inputs.class +) @Operator( group = "train" ) @@ -51,8 +57,8 @@ public final class TileGrad extends RawOp implements Operand private Output output; - private TileGrad(Operation operation) { - super(operation); + public TileGrad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = TileGrad.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/BroadcastHelper.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/BroadcastHelper.java index 31b51e9ef81..7f1ed2e0693 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/BroadcastHelper.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/BroadcastHelper.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code lhs_output} output */ +@OpMetadata( + opType = BroadcastHelper.OP_NAME, + inputsClass = BroadcastHelper.Inputs.class +) @Operator( group = "xla" ) @@ -53,8 +59,8 @@ public final class BroadcastHelper extends RawOp { private Output rhsOutput; - private BroadcastHelper(Operation operation) { - super(operation); + public BroadcastHelper(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; lhsOutput = operation.output(outputIdx++); rhsOutput = operation.output(outputIdx++); @@ -100,6 +106,9 @@ public Output rhsOutput() { return rhsOutput; } + @OpInputsMetadata( + outputsClass = BroadcastHelper.class + ) public static class Inputs extends RawOpInputs> { /** * the LHS input tensor diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ClusterOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ClusterOutput.java index ad111378f34..4ae6908f1aa 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ClusterOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ClusterOutput.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code outputs} output */ +@OpMetadata( + opType = ClusterOutput.OP_NAME, + inputsClass = ClusterOutput.Inputs.class +) @Operator( group = "xla" ) @@ -47,8 +53,8 @@ public final class ClusterOutput extends RawOp implements Opera private Output outputs; - private ClusterOutput(Operation operation) { - super(operation); + public ClusterOutput(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; outputs = operation.output(outputIdx++); } @@ -84,6 +90,9 @@ public Output asOutput() { return outputs; } + @OpInputsMetadata( + outputsClass = ClusterOutput.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Conv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Conv.java index e8df7159368..0c57763cd5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Conv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Conv.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Conv.OP_NAME, + inputsClass = Conv.Inputs.class +) @Operator( group = "xla" ) @@ -51,8 +57,8 @@ public final class Conv extends RawOp implements Operand { private Output output; - private Conv(Operation operation) { - super(operation); + public Conv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -111,6 +117,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Conv.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dequantize.java index 03cece9dc1e..7fc5e4adb2f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dequantize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBfloat16; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * Takes the packed uint32 input and unpacks the input to uint8 to do * Dequantization on device. */ +@OpMetadata( + opType = Dequantize.OP_NAME, + inputsClass = Dequantize.Inputs.class +) @Operator( group = "xla" ) @@ -46,8 +52,8 @@ public final class Dequantize extends RawOp implements Operand { private Output output; - private Dequantize(Operation operation) { - super(operation); + public Dequantize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Dequantize.class + ) public static class Inputs extends RawOpInputs { /** * Input tensors whose types is uint32, shape is [d0, ..., dn]. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dot.java index 5e5c78e362c..16af0ba81e9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Dot.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Dot.OP_NAME, + inputsClass = Dot.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class Dot extends RawOp implements Operand { private Output output; - private Dot(Operation operation) { - super(operation); + public Dot(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Dot.class + ) public static class Inputs extends RawOpInputs> { /** * the LHS tensor diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicSlice.java index 65f082eabc2..6062f75e5ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicSlice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicSlice.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DynamicSlice.OP_NAME, + inputsClass = DynamicSlice.Inputs.class +) @Operator( group = "xla" ) @@ -55,8 +61,8 @@ public final class DynamicSlice extends RawOp implements Operan private Output output; - private DynamicSlice(Operation operation) { - super(operation); + public DynamicSlice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = DynamicSlice.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicUpdateSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicUpdateSlice.java index e1a4c0e7574..54ee6f4b457 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicUpdateSlice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/DynamicUpdateSlice.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +46,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = DynamicUpdateSlice.OP_NAME, + inputsClass = DynamicUpdateSlice.Inputs.class +) @Operator( group = "xla" ) @@ -55,8 +61,8 @@ public final class DynamicUpdateSlice extends RawOp implements private Output output; - private DynamicUpdateSlice(Operation operation) { - super(operation); + public DynamicUpdateSlice(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = DynamicUpdateSlice.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Einsum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Einsum.java index 3048c1e6f06..72d482e1fe6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Einsum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Einsum.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ * * @param data type for {@code product} output */ +@OpMetadata( + opType = Einsum.OP_NAME, + inputsClass = Einsum.Inputs.class +) @Operator( group = "xla" ) @@ -49,8 +55,8 @@ public final class Einsum extends RawOp implements Operand { private Output product; - private Einsum(Operation operation) { - super(operation); + public Einsum(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; product = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return product; } + @OpInputsMetadata( + outputsClass = Einsum.class + ) public static class Inputs extends RawOpInputs> { /** * The a input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Gather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Gather.java index d8b7cf0421a..63f75f76011 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Gather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Gather.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -38,6 +40,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Gather.OP_NAME, + inputsClass = Gather.Inputs.class +) @Operator( group = "xla" ) @@ -49,8 +55,8 @@ public final class Gather extends RawOp implements Operand { private Output output; - private Gather(Operation operation) { - super(operation); + public Gather(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -97,6 +103,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Gather.class + ) public static class Inputs extends RawOpInputs> { /** * The array we're gathering from. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/If.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/If.java index 31ac4815360..4422e0bd391 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/If.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/If.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * output = cond ? then_branch(inputs) : else_branch(inputs). */ +@OpMetadata( + opType = If.OP_NAME, + inputsClass = If.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class If extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private If(Operation operation) { - super(operation); + public If(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -103,6 +109,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = If.class + ) public static class Inputs extends RawOpInputs { /** * A boolean scalar. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/KeyValueSort.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/KeyValueSort.java index ad72e80c6f0..72d4941485c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/KeyValueSort.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/KeyValueSort.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +44,10 @@ * * @param data type for {@code sorted_values} output */ +@OpMetadata( + opType = KeyValueSort.OP_NAME, + inputsClass = KeyValueSort.Inputs.class +) @Operator( group = "xla" ) @@ -55,8 +61,8 @@ public final class KeyValueSort extends RawO private Output sortedValues; - private KeyValueSort(Operation operation) { - super(operation); + public KeyValueSort(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; sortedKeys = operation.output(outputIdx++); sortedValues = operation.output(outputIdx++); @@ -101,6 +107,9 @@ public Output sortedValues() { return sortedValues; } + @OpInputsMetadata( + outputsClass = KeyValueSort.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type K. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Pad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Pad.java index f28df38bbe0..ae2788bebe2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Pad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Pad.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Pad.OP_NAME, + inputsClass = Pad.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class Pad extends RawOp implements Operand { private Output output; - private Pad(Operation operation) { - super(operation); + public Pad(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -102,6 +108,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Pad.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Recv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Recv.java index a660cd82f4c..72ecccaa144 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Recv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Recv.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code tensor} output */ +@OpMetadata( + opType = Recv.OP_NAME, + inputsClass = Recv.Inputs.class +) @Operator( group = "xla" ) @@ -51,8 +57,8 @@ public final class Recv extends RawOp implements Operand { private Output tensor; - private Recv(Operation operation) { - super(operation); + public Recv(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; tensor = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return tensor; } + @OpInputsMetadata( + outputsClass = Recv.class + ) public static class Inputs extends RawOpInputs> { /** * The type of the tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Reduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Reduce.java index 3e25c7db1c0..8da34221d7b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Reduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Reduce.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Reduce.OP_NAME, + inputsClass = Reduce.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class Reduce extends RawOp implements Operand { private Output output; - private Reduce(Operation operation) { - super(operation); + public Reduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -98,6 +104,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Reduce.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReduceWindow.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReduceWindow.java index 41cf0d5cfbe..39385fc184c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReduceWindow.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReduceWindow.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = ReduceWindow.OP_NAME, + inputsClass = ReduceWindow.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class ReduceWindow extends RawOp implements Operan private Output output; - private ReduceWindow(Operation operation) { - super(operation); + public ReduceWindow(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = ReduceWindow.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/RemoveDynamicDimensionSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/RemoveDynamicDimensionSize.java index b657a92ee72..56f0bf33738 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/RemoveDynamicDimensionSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/RemoveDynamicDimensionSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = RemoveDynamicDimensionSize.OP_NAME, + inputsClass = RemoveDynamicDimensionSize.Inputs.class +) @Operator( group = "xla" ) @@ -52,8 +58,8 @@ public final class RemoveDynamicDimensionSize extends RawOp imp private Output output; - private RemoveDynamicDimensionSize(Operation operation) { - super(operation); + public RemoveDynamicDimensionSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -92,6 +98,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = RemoveDynamicDimensionSize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReplicaId.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReplicaId.java index 9b12ea45151..9b061954c33 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReplicaId.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReplicaId.java @@ -27,12 +27,18 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; /** * Replica ID. */ +@OpMetadata( + opType = ReplicaId.OP_NAME, + inputsClass = ReplicaId.Inputs.class +) @Operator( group = "xla" ) @@ -44,8 +50,8 @@ public final class ReplicaId extends RawOp implements Operand { private Output id; - private ReplicaId(Operation operation) { - super(operation); + public ReplicaId(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; id = operation.output(outputIdx++); } @@ -78,6 +84,9 @@ public Output asOutput() { return id; } + @OpInputsMetadata( + outputsClass = ReplicaId.class + ) public static class Inputs extends RawOpInputs { public Inputs(GraphOperation op) { super(new ReplicaId(op), op, Arrays.asList()); diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Scatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Scatter.java index f5114c2979a..0f9028d7e35 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Scatter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Scatter.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Scatter.OP_NAME, + inputsClass = Scatter.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class Scatter extends RawOp implements Operand private Output output; - private Scatter(Operation operation) { - super(operation); + public Scatter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -101,6 +107,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Scatter.class + ) public static class Inputs extends RawOpInputs> { /** * Array to be scattered into. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelectAndScatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelectAndScatter.java index 715c01a5a53..1da6395144c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelectAndScatter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelectAndScatter.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TNumber; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SelectAndScatter.OP_NAME, + inputsClass = SelectAndScatter.Inputs.class +) @Operator( group = "xla" ) @@ -51,8 +57,8 @@ public final class SelectAndScatter extends RawOp implements Op private Output output; - private SelectAndScatter(Operation operation) { - super(operation); + public SelectAndScatter(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -105,6 +111,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SelectAndScatter.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelfAdjointEig.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelfAdjointEig.java index 0c190e36fe0..4d089c4f56f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelfAdjointEig.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SelfAdjointEig.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code w} output */ +@OpMetadata( + opType = SelfAdjointEig.OP_NAME, + inputsClass = SelfAdjointEig.Inputs.class +) @Operator( group = "xla" ) @@ -53,8 +59,8 @@ public final class SelfAdjointEig extends RawOp { private Output v; - private SelfAdjointEig(Operation operation) { - super(operation); + public SelfAdjointEig(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; w = operation.output(outputIdx++); v = operation.output(outputIdx++); @@ -108,6 +114,9 @@ public Output v() { return v; } + @OpInputsMetadata( + outputsClass = SelfAdjointEig.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Send.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Send.java index bb5010128fa..18bdc7d2ba0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Send.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Send.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -35,6 +37,10 @@ * documented at * https://www.tensorflow.org/performance/xla/operation_semantics#send . */ +@OpMetadata( + opType = Send.OP_NAME, + inputsClass = Send.Inputs.class +) @Operator( group = "xla" ) @@ -44,8 +50,8 @@ public final class Send extends RawOp { */ public static final String OP_NAME = "XlaSend"; - private Send(Operation operation) { - super(operation); + public Send(Operation operation) { + super(operation, OP_NAME); } /** @@ -66,6 +72,9 @@ public static Send create(Scope scope, Operand tensor, String t return new Send(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = Send.class + ) public static class Inputs extends RawOpInputs { /** * The tensor to send. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SetDynamicDimensionSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SetDynamicDimensionSize.java index 00c232c5319..79ad9f2524c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SetDynamicDimensionSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SetDynamicDimensionSize.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -41,6 +43,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SetDynamicDimensionSize.OP_NAME, + inputsClass = SetDynamicDimensionSize.Inputs.class +) @Operator( group = "xla" ) @@ -52,8 +58,8 @@ public final class SetDynamicDimensionSize extends RawOp implem private Output output; - private SetDynamicDimensionSize(Operation operation) { - super(operation); + public SetDynamicDimensionSize(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -94,6 +100,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SetDynamicDimensionSize.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sharding.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sharding.java index d39f3f93f21..6fb001ae681 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sharding.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sharding.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Sharding.OP_NAME, + inputsClass = Sharding.Inputs.class +) @Operator( group = "xla" ) @@ -47,8 +53,8 @@ public final class Sharding extends RawOp implements Operand private Output output; - private Sharding(Operation operation) { - super(operation); + public Sharding(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -124,6 +130,9 @@ public Options sharding(String sharding) { } } + @OpInputsMetadata( + outputsClass = Sharding.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sort.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sort.java index 70056c2c7c1..470d2063e5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sort.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Sort.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = Sort.OP_NAME, + inputsClass = Sort.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class Sort extends RawOp implements Operand { private Output output; - private Sort(Operation operation) { - super(operation); + public Sort(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -87,6 +93,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = Sort.class + ) public static class Inputs extends RawOpInputs> { /** * A {@code Tensor} of type T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdFullToShardShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdFullToShardShape.java index b717ed45b8b..e8cfd9a709d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdFullToShardShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdFullToShardShape.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SpmdFullToShardShape.OP_NAME, + inputsClass = SpmdFullToShardShape.Inputs.class +) @Operator( group = "xla" ) @@ -51,8 +57,8 @@ public final class SpmdFullToShardShape extends RawOp implement private Output output; - private SpmdFullToShardShape(Operation operation) { - super(operation); + public SpmdFullToShardShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -91,6 +97,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SpmdFullToShardShape.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdShardToFullShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdShardToFullShape.java index 3a86c8b7b26..5cba20d5891 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdShardToFullShape.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SpmdShardToFullShape.java @@ -28,6 +28,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +42,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = SpmdShardToFullShape.OP_NAME, + inputsClass = SpmdShardToFullShape.Inputs.class +) @Operator( group = "xla" ) @@ -51,8 +57,8 @@ public final class SpmdShardToFullShape extends RawOp implement private Output output; - private SpmdShardToFullShape(Operation operation) { - super(operation); + public SpmdShardToFullShape(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -93,6 +99,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = SpmdShardToFullShape.class + ) public static class Inputs extends RawOpInputs> { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Svd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Svd.java index e6049fbf8fd..60ef26b9e5e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Svd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/Svd.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ * * @param data type for {@code s} output */ +@OpMetadata( + opType = Svd.OP_NAME, + inputsClass = Svd.Inputs.class +) @Operator( group = "xla" ) @@ -54,8 +60,8 @@ public final class Svd extends RawOp { private Output v; - private Svd(Operation operation) { - super(operation); + public Svd(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; s = operation.output(outputIdx++); u = operation.output(outputIdx++); @@ -117,6 +123,9 @@ public Output v() { return v; } + @OpInputsMetadata( + outputsClass = Svd.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/While.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/While.java index 80002335b2b..0fe6c0598d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/While.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/While.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * output = input; While (Cond(output)) { output = Body(output) } */ +@OpMetadata( + opType = While.OP_NAME, + inputsClass = While.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class While extends RawOp implements Iterable> { private List> output; @SuppressWarnings("unchecked") - private While(Operation operation) { - super(operation); + public While(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList(operation.outputList(outputIdx, outputLength)); @@ -102,6 +108,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = While.class + ) public static class Inputs extends RawOpInputs { /** * A list of input tensors whose types are T. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaHostCompute.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaHostCompute.java index fcb5d6234e0..603979a7195 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaHostCompute.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaHostCompute.java @@ -32,6 +32,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -39,6 +41,10 @@ /** * A pseudo-op to represent host-side computation in an XLA program. */ +@OpMetadata( + opType = XlaHostCompute.OP_NAME, + inputsClass = XlaHostCompute.Inputs.class +) @Operator( group = "xla" ) @@ -51,8 +57,8 @@ public final class XlaHostCompute extends RawOp implements Iterable> outputs; @SuppressWarnings("unchecked") - private XlaHostCompute(Operation operation) { - super(operation); + public XlaHostCompute(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList(operation.outputList(outputIdx, outputsLength)); @@ -229,6 +235,9 @@ public Options tpuCore(Long tpuCore) { } } + @OpInputsMetadata( + outputsClass = XlaHostCompute.class + ) public static class Inputs extends RawOpInputs { /** * A list of tensors that will be sent to the host. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaLaunch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaLaunch.java index d8c864cea8f..5098d6d7190 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaLaunch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaLaunch.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +40,10 @@ /** * XLA Launch Op. For use by the XLA JIT only. */ +@OpMetadata( + opType = XlaLaunch.OP_NAME, + inputsClass = XlaLaunch.Inputs.class +) @Operator( group = "xla" ) @@ -50,8 +56,8 @@ public final class XlaLaunch extends RawOp implements Iterable> { private List> results; @SuppressWarnings("unchecked") - private XlaLaunch(Operation operation) { - super(operation); + public XlaLaunch(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int resultsLength = operation.outputListLength("results"); results = Arrays.asList(operation.outputList(outputIdx, resultsLength)); @@ -99,6 +105,9 @@ public Iterator> iterator() { return (Iterator) results.iterator(); } + @OpInputsMetadata( + outputsClass = XlaLaunch.class + ) public static class Inputs extends RawOpInputs { /** * The constants input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvFromHost.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvFromHost.java index 289a7ff2853..1be2ef10874 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvFromHost.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvFromHost.java @@ -29,6 +29,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = XlaRecvFromHost.OP_NAME, + inputsClass = XlaRecvFromHost.Inputs.class +) @Operator( group = "xla" ) @@ -53,8 +59,8 @@ public final class XlaRecvFromHost extends RawOp implements Ope private Output output; - private XlaRecvFromHost(Operation operation) { - super(operation); + public XlaRecvFromHost(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -95,6 +101,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = XlaRecvFromHost.class + ) public static class Inputs extends RawOpInputs> { /** * The Toutput attribute diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendToHost.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendToHost.java index 41478534837..535215c5a70 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendToHost.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendToHost.java @@ -26,6 +26,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -36,6 +38,10 @@ * Tinput: element type for input. * key: A unique identifier for this region used to match up host transfers. */ +@OpMetadata( + opType = XlaSendToHost.OP_NAME, + inputsClass = XlaSendToHost.Inputs.class +) @Operator( group = "xla" ) @@ -45,8 +51,8 @@ public final class XlaSendToHost extends RawOp { */ public static final String OP_NAME = "XlaSendToHost"; - private XlaSendToHost(Operation operation) { - super(operation); + public XlaSendToHost(Operation operation) { + super(operation, OP_NAME); } /** @@ -67,6 +73,9 @@ public static XlaSendToHost create(Scope scope, Operand input, return new XlaSendToHost(opBuilder.build()); } + @OpInputsMetadata( + outputsClass = XlaSendToHost.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSetBound.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSetBound.java index 7c51caf3b3b..630988d4d4a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSetBound.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSetBound.java @@ -27,6 +27,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; @@ -36,6 +38,10 @@ * returns the same value. * */ +@OpMetadata( + opType = XlaSetBound.OP_NAME, + inputsClass = XlaSetBound.Inputs.class +) @Operator( group = "xla" ) @@ -47,8 +53,8 @@ public final class XlaSetBound extends RawOp implements Operand { private Output output; - private XlaSetBound(Operation operation) { - super(operation); + public XlaSetBound(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } @@ -85,6 +91,9 @@ public Output asOutput() { return output; } + @OpInputsMetadata( + outputsClass = XlaSetBound.class + ) public static class Inputs extends RawOpInputs { /** * The input input diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicReduce.java index 15523c32a6e..4ae22b1c842 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicReduce.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +44,10 @@ * * @param data type for {@code output} output */ +@OpMetadata( + opType = XlaVariadicReduce.OP_NAME, + inputsClass = XlaVariadicReduce.Inputs.class +) @Operator( group = "xla" ) @@ -54,8 +60,8 @@ public final class XlaVariadicReduce extends RawOp implements I private List> output; @SuppressWarnings("unchecked") - private XlaVariadicReduce(Operation operation) { - super(operation); + public XlaVariadicReduce(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputLength = operation.outputListLength("output"); output = Arrays.asList((Output[]) operation.outputList(outputIdx, outputLength)); @@ -106,6 +112,9 @@ public Iterator> iterator() { return (Iterator) output.iterator(); } + @OpInputsMetadata( + outputsClass = XlaVariadicReduce.class + ) public static class Inputs extends RawOpInputs> { /** * the input tensor(s) diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicSort.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicSort.java index 8570842977b..1f7bfc1d7fb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicSort.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaVariadicSort.java @@ -31,6 +31,8 @@ import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.framework.DataType; import org.tensorflow.types.TInt32; @@ -43,6 +45,10 @@ *

    Sorts one or more tensors, with support for custom comparator, dimension, and * is_stable attributes. */ +@OpMetadata( + opType = XlaVariadicSort.OP_NAME, + inputsClass = XlaVariadicSort.Inputs.class +) @Operator( group = "xla" ) @@ -55,8 +61,8 @@ public final class XlaVariadicSort extends RawOp implements Iterable> outputs; @SuppressWarnings("unchecked") - private XlaVariadicSort(Operation operation) { - super(operation); + public XlaVariadicSort(Operation operation) { + super(operation, OP_NAME); int outputIdx = 0; int outputsLength = operation.outputListLength("outputs"); outputs = Arrays.asList(operation.outputList(outputIdx, outputsLength)); @@ -103,6 +109,9 @@ public Iterator> iterator() { return (Iterator) outputs.iterator(); } + @OpInputsMetadata( + outputsClass = XlaVariadicSort.class + ) public static class Inputs extends RawOpInputs { /** * A list of {@code Tensor} of identical shape but possibly different types. diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/BaseGradientAdapter.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/BaseGradientAdapter.java new file mode 100644 index 00000000000..41ee090e15d --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/BaseGradientAdapter.java @@ -0,0 +1,90 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ +package org.tensorflow; + +import java.util.ArrayList; +import java.util.List; +import org.bytedeco.javacpp.PointerScope; +import org.tensorflow.internal.c_api.GradFunc; +import org.tensorflow.internal.c_api.NativeOutput; +import org.tensorflow.internal.c_api.NativeOutputVector; +import org.tensorflow.internal.c_api.Node; +import org.tensorflow.internal.c_api.TF_Operation; + +/** Helper base class for custom gradient adapters INTERNAL USE ONLY */ +public abstract class BaseGradientAdapter extends GradFunc { + + protected BaseGradientAdapter() { + super(); + } + + /** + * Convert an array of native outputs to a list of {@link Output}s. + * + * @param g the graph the outputs are in + * @param nativeOutputs the native outputs to convert + * @return a list of Outputs + */ + protected static List> fromNativeOutputs(Graph g, NativeOutputVector nativeOutputs) { + List> gradInputs = new ArrayList<>((int) nativeOutputs.size()); + for (int i = 0; i < nativeOutputs.size(); i++) { + NativeOutput output = nativeOutputs.get(i); + gradInputs.add(new Output<>(getGraphOp(g, output.node()), output.index())); + } + return gradInputs; + } + + /** + * Put the Java outputs into the array of native outputs, resizing it to the necessary size. + * + * @param outputs the outputs to put + * @param nativeOutputs the native array to put the outputs into + */ + protected static void putToNativeOutputs( + List> outputs, NativeOutputVector nativeOutputs) { + nativeOutputs.resize(outputs.size()); + for (int i = 0; i < outputs.size(); i++) { + Output output = outputs.get(i).asOutput(); + Node node = ((GraphOperation) output.op()).getUnsafeNativeHandle().node(); + nativeOutputs.put(i, new NativeOutput(node, output.index())); + } + } + + /** + * Make a {@link GraphOperation} from a native {@link Node} + * + * @param g the graph the operation is in + * @param node the native node + * @return a graph operation with the underlying native node + */ + protected static GraphOperation getGraphOp(Graph g, Node node) { + try (PointerScope scope = new PointerScope(); + Graph.Reference ref = g.ref()) { + return new GraphOperation(g, new TF_Operation(node)); + } + } + + /** + * Use builders without locking. This should only be used during custom gradient building. + * + *

    The graph locks are not re-entrant, so attempting to add an op to a graph that has been + * locked by the gradient builder will fail without this. + */ + protected static void useDangerousLockedBuilders(Graph g, boolean dangerous) { + g.setDangerousGradientBuilder(dangerous); + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/EagerSession.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/EagerSession.java index f141e9dc551..ba90aa53f10 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/EagerSession.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/EagerSession.java @@ -29,6 +29,7 @@ import org.tensorflow.internal.c_api.TFE_Context; import org.tensorflow.internal.c_api.TFE_ContextOptions; import org.tensorflow.internal.c_api.TF_Status; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Scope; import org.tensorflow.op.core.Assign; import org.tensorflow.op.core.Placeholder; @@ -397,7 +398,7 @@ void detach(Pointer... resources) { private final WeakPointerScope nativeResources; private TFE_Context nativeHandle; - private final Scope baseScope = new Scope(this); + private final Scope baseScope = new OpScope(this); private EagerSession(Options options) { this.nativeResources = new WeakPointerScope(); diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Graph.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Graph.java index 83aeb1a409b..18d8186d0b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Graph.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Graph.java @@ -39,6 +39,7 @@ import java.util.List; import java.util.Queue; import java.util.Set; +import java.util.WeakHashMap; import java.util.stream.Collectors; import org.bytedeco.javacpp.BytePointer; import org.bytedeco.javacpp.Pointer; @@ -46,6 +47,7 @@ import org.bytedeco.javacpp.PointerScope; import org.bytedeco.javacpp.SizeTPointer; import org.tensorflow.exceptions.TensorFlowException; +import org.tensorflow.internal.c_api.NativeGraphPointer; import org.tensorflow.internal.c_api.TF_Buffer; import org.tensorflow.internal.c_api.TF_Function; import org.tensorflow.internal.c_api.TF_Graph; @@ -56,6 +58,7 @@ import org.tensorflow.internal.c_api.TF_WhileParams; import org.tensorflow.ndarray.StdArrays; import org.tensorflow.op.Op; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Ops; import org.tensorflow.op.Scope; import org.tensorflow.op.core.Constant; @@ -81,14 +84,14 @@ public final class Graph implements ExecutionEnvironment, AutoCloseable { /** Create an empty Graph. */ public Graph() { - nativeHandle = allocate(); - this.baseScope = new Scope(this); + this(allocate()); } /** Create a Graph from an existing handle (takes ownership). */ Graph(TF_Graph nativeHandle) { this.nativeHandle = nativeHandle; - this.baseScope = new Scope(this); + this.baseScope = new OpScope(this); + allGraphs.add(this); } Graph(TF_Graph nativeHandle, SaverDef saverDef) { @@ -119,6 +122,7 @@ public void close() { } delete(nativeHandle); nativeHandle = null; + allGraphs.remove(this); } } @@ -215,7 +219,7 @@ public Output outputOrThrow(String output) { *

    The order of iteration is unspecified. Consumers of the iterator will receive no * notification should the underlying graph change during iteration. */ - public Iterator operations() { + public Iterator operations() { return new OperationIterator(this); } @@ -366,12 +370,22 @@ public synchronized Set subgraphFrom(Set> inputs) { return downstream; } + /** + * Returns a builder to add {@link Operation}s to the Graph. + * + * @param type of the Operation (i.e., identifies the computation to be performed) + * @param name to refer to the created Operation in the graph. + * @param scope the scope to use for the operation + * @return an {@link OperationBuilder}, which will add the Operation to the graph when {@link + * OperationBuilder#build()} is invoked. If {@link OperationBuilder#build()} is not invoked, + * then some resources may leak. + */ @Override public GraphOperationBuilder opBuilder(String type, String name, Scope scope) { if (!isOpEnabled(type)) { throw new IllegalArgumentException("Op " + type + " is not valid in graph mode."); } - return new GraphOperationBuilder(this, type, name, scope); + return new GraphOperationBuilder(this, type, name, scope, dangerousGradientBuilder); } @Override @@ -652,7 +666,8 @@ public boolean hasInitializers() { * @param dx if not null, the partial derivatives of some loss function {@code L} w.r.t. {@code y} * @return the partial derivatives {@code dy} with the size of {@code x} */ - public Output[] addGradients(String prefix, Output[] y, Output[] x, Output[] dx) { + public synchronized Output[] addGradients( + String prefix, Output[] y, Output[] x, Output[] dx) { Output[] dy = new Output[x.length]; final TF_Operation[] yHandles = new TF_Operation[y.length]; final int[] yIndices = new int[y.length]; @@ -847,6 +862,8 @@ public Output[] whileLoop( * Return the {@link SaverDef} instance used to save the state of all variables present in this * graph. * + *

    + * *

    The first time this method is called it builds the {@link SaverDef}. If this graph already * contains a "save/restore_all" operation then it is assumed to contain all necessary saving and * restoring operations. If that operation does not exist then the graph is mutated to add all the @@ -882,9 +899,21 @@ synchronized SaverDef saverDef() { private SaverDef saverDef; private final Scope baseScope; + private boolean dangerousGradientBuilder; + private final Set initializers = Collections.synchronizedSet(new LinkedHashSet<>()); private boolean newInitializers = false; + /** + * Use builders without locking. This should only be used during custom gradient building. + * + *

    The graph locks are not re-entrant, so attempting to add an op to a graph that has been + * locked by the gradient builder will fail without this. + */ + synchronized void setDangerousGradientBuilder(boolean dangerous) { + dangerousGradientBuilder = dangerous; + } + // Related native objects (such as the TF_Operation object backing an Operation instance) // have a validity tied to that of the Graph. The handles to those native objects are not // valid after Graph.close() has been invoked. @@ -930,7 +959,7 @@ Reference ref() { return new Reference(); } - private static final class OperationIterator implements Iterator { + private static final class OperationIterator implements Iterator { OperationIterator(Graph g) { this.graph = g; @@ -964,8 +993,8 @@ public boolean hasNext() { } @Override - public Operation next() { - Operation rhett = this.operation; + public GraphOperation next() { + GraphOperation rhett = this.operation; this.advance(); return rhett; } @@ -976,7 +1005,7 @@ public void remove() { } private final Graph graph; - private Operation operation; + private GraphOperation operation; private int position; } @@ -1246,7 +1275,7 @@ private static SaverDef addVariableSaver(Graph graph) { List> varOutputs = new ArrayList<>(); List> varTypes = new ArrayList<>(); - for (Iterator iter = graph.operations(); iter.hasNext(); ) { + for (Iterator iter = graph.operations(); iter.hasNext(); ) { Operation op = iter.next(); if (op.type().equals("VariableV2")) { varNames.add(op.name()); @@ -1287,6 +1316,25 @@ private static SaverDef addVariableSaver(Graph graph) { .build(); } + private static Set allGraphs = + Collections.synchronizedSet(Collections.newSetFromMap(new WeakHashMap<>())); + + /** + * Find the graph with the matching underlying native pointer. + * + * @return the graph if there is one, else null. + */ + public static Graph findGraphForPointer(NativeGraphPointer pointer) { + for (Graph g : allGraphs) { + if (g.nativeHandle != null + && !g.nativeHandle.isNull() + && g.nativeHandle.graph().equals(pointer)) { + return g; + } + } + return null; + } + static { try { // Ensure that TensorFlow native library and classes are ready to be used diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperation.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperation.java index d811139e9a2..4c01774f759 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperation.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperation.java @@ -194,6 +194,7 @@ public String device() { /** Get the number of inputs to the op, not including control inputs. */ public int numInputs() { + requireHandle(unsafeNativeHandle); return TF_OperationNumInputs(getUnsafeNativeHandle()); } @@ -212,8 +213,7 @@ public Output input(int idx) { try (PointerScope scope = new PointerScope()) { TF_Input input = new TF_Input().oper(getUnsafeNativeHandle()).index(idx); TF_Output output = TF_OperationInput(input); - String opName = TF_OperationName(output.oper()).getString(); - return graph.operation(opName).output(output.index()); + return new GraphOperation(graph, output.oper()).output(output.index()); } } @@ -244,6 +244,7 @@ public Output[] inputList(int idx, int length) { /** Get the op's inputs, not including control inputs. */ public List> inputs() { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { int numInputs = numInputs(); TF_Output handles = new TF_Output(numInputs); @@ -269,6 +270,7 @@ public List> inputs() { * @param index the output to look for usages of */ public int numConsumers(int index) { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { TF_Output output = new TF_Output().oper(getUnsafeNativeHandle()).index(index); return TF_OperationOutputNumConsumers(output); @@ -282,6 +284,7 @@ public int numConsumers(int index) { * @param index the output to look for usages of */ public Set consumers(int index) { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { TF_Output output = new TF_Output().oper(getUnsafeNativeHandle()).index(index); int numConsumers = numConsumers(index); @@ -305,6 +308,7 @@ public Set consumers(int index) { * dependencies. */ public int numConsumers() { + requireHandle(unsafeNativeHandle); int all = 0; for (int i = 0; i < numOutputs(); i++) { all += numConsumers(i); @@ -316,6 +320,7 @@ public int numConsumers() { * Get the ops that use any of this op's outputs as an input, not including control dependencies. */ public Set consumers() { + requireHandle(unsafeNativeHandle); Set all = new LinkedHashSet<>(); for (int i = 0; i < numOutputs(); i++) { all.addAll(consumers(i)); @@ -325,6 +330,7 @@ public Set consumers() { /** Get the number of control inputs for this op. */ public int numControlInputs() { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { return TF_OperationNumControlInputs(getUnsafeNativeHandle()); } @@ -332,6 +338,7 @@ public int numControlInputs() { /** Get the control inputs of this op. */ public Set controlInputs() { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { int numInputs = numControlInputs(); PointerPointer handles = new PointerPointer<>(numInputs); @@ -350,6 +357,7 @@ public Set controlInputs() { /** Get the number of ops with this op as a control dependency. */ public int numControlConsumers() { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { return TF_OperationNumControlOutputs(getUnsafeNativeHandle()); } @@ -357,6 +365,7 @@ public int numControlConsumers() { /** Get the ops with this op as a control dependency. */ public Set controlConsumers() { + requireHandle(unsafeNativeHandle); try (PointerScope scope = new PointerScope()) { int numConsumers = numControlConsumers(); PointerPointer handles = new PointerPointer<>(numConsumers); @@ -373,7 +382,12 @@ public Set controlConsumers() { } } - TF_Operation getUnsafeNativeHandle() { + /** + * Get the native handle of this operation. + * + *

    No liveness or non-null checking is done, the operation may have been deallocated. + */ + public TF_Operation getUnsafeNativeHandle() { return unsafeNativeHandle; } diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperationBuilder.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperationBuilder.java index 63434a9638b..64995612d7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperationBuilder.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperationBuilder.java @@ -19,7 +19,10 @@ import static org.tensorflow.internal.c_api.global.tensorflow.TF_AddInput; import static org.tensorflow.internal.c_api.global.tensorflow.TF_AddInputList; import static org.tensorflow.internal.c_api.global.tensorflow.TF_FinishOperation; +import static org.tensorflow.internal.c_api.global.tensorflow.TF_FinishOperationLocked; import static org.tensorflow.internal.c_api.global.tensorflow.TF_NewOperation; +import static org.tensorflow.internal.c_api.global.tensorflow.TF_NewOperationLocked; +import static org.tensorflow.internal.c_api.global.tensorflow.TF_OperationName; import static org.tensorflow.internal.c_api.global.tensorflow.TF_SetAttrBool; import static org.tensorflow.internal.c_api.global.tensorflow.TF_SetAttrBoolList; import static org.tensorflow.internal.c_api.global.tensorflow.TF_SetAttrFloat; @@ -67,14 +70,17 @@ /** An {@link OperationBuilder} for adding {@link GraphOperation}s to a {@link Graph}. */ public final class GraphOperationBuilder implements OperationBuilder { - GraphOperationBuilder(Graph graph, String type, String name, Scope scope) { + GraphOperationBuilder( + Graph graph, String type, String name, Scope scope, boolean dangerousGradientBuilder) { this.graph = graph; this.scope = scope; - Graph.Reference r = graph.ref(); - try { - this.unsafeNativeHandle = allocate(r.nativeHandle(), type, name); - } finally { - r.close(); + this.dangerousGradientBuilder = dangerousGradientBuilder; + try (Graph.Reference r = graph.ref()) { + if (dangerousGradientBuilder) { + this.unsafeNativeHandle = allocateDangerousGradient(r.nativeHandle(), type, name); + } else { + this.unsafeNativeHandle = allocate(r.nativeHandle(), type, name); + } } } @@ -86,14 +92,17 @@ public final class GraphOperationBuilder implements OperationBuilder { @Override public GraphOperation build() { scope.apply(this); - Graph.Reference r = graph.ref(); - try { - GraphOperation op = new GraphOperation(graph, finish(unsafeNativeHandle)); + try (Graph.Reference r = graph.ref()) { + TF_Operation built; + if (dangerousGradientBuilder) { + built = finishDangerousGradient(r.nativeHandle(), unsafeNativeHandle); + } else { + built = finish(unsafeNativeHandle); + } + GraphOperation op = new GraphOperation(graph, built); unsafeNativeHandle = null; scope.onOpCreated(op); return op; - } finally { - r.close(); } } @@ -392,6 +401,14 @@ public OperationBuilder setAttr(String name, AttrValue value) { private final Graph graph; private final Scope scope; + /** + * Use builders without locking. This should only be used during custom gradient building. + * + *

    The graph locks are not re-entrant, so attempting to add an op to a graph that has been + * locked by the gradient builder will fail without this. + */ + private final boolean dangerousGradientBuilder; + private static void requireHandle(Pointer handle) { if (handle == null || handle.isNull()) { throw new IllegalStateException("Operation has already been built"); @@ -419,6 +436,20 @@ private static TF_OperationDescription allocate(TF_Graph graphHandle, String typ return TF_NewOperation(graphHandle, type, name); } + /** + * Use builders without locking. This should only be used during custom gradient building. + * + *

    The graph locks are not re-entrant, so attempting to add an op to a graph that has been + * locked by the gradient builder will fail without this. + */ + private static TF_OperationDescription allocateDangerousGradient( + TF_Graph graphHandle, String type, String name) { + if (graphHandle == null || graphHandle.isNull()) { + throw new IllegalStateException("close() has been called on the Graph"); + } + return TF_NewOperationLocked(graphHandle, type, name); + } + private static TF_Operation finish(TF_OperationDescription handle) { requireHandle(handle); @@ -430,6 +461,24 @@ private static TF_Operation finish(TF_OperationDescription handle) { } } + /** + * Use builders without locking. This should only be used during custom gradient building. + * + *

    The graph locks are not re-entrant, so attempting to add an op to a graph that has been + * locked by the gradient builder will fail without this. + */ + private static TF_Operation finishDangerousGradient(TF_Graph g, TF_OperationDescription handle) { + requireHandle(handle); + + try (PointerScope scope = new PointerScope()) { + TF_Status status = TF_Status.newStatus(); + TF_Operation op = TF_FinishOperationLocked(handle, status); + status.throwExceptionIfNotOK(); + g.name_map().erase(TF_OperationName(op)); + return op; + } + } + private static void addInput(TF_OperationDescription handle, TF_Operation opHandle, int index) { try (PointerScope scope = new PointerScope()) { TF_Output out = new TF_Output(); diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/TensorFlow.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/TensorFlow.java index 23f4c62bc7f..53748b82271 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/TensorFlow.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/TensorFlow.java @@ -1,18 +1,18 @@ /* Copyright 2019-2021 The TensorFlow Authors. All Rights Reserved. - Licensed under the Apache License, Version 2.0 (the "License"); - you may not use this file except in compliance with the License. - You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - - Unless required by applicable law or agreed to in writing, software - distributed under the License is distributed on an "AS IS" BASIS, - WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - See the License for the specific language governing permissions and - limitations under the License. - ======================================================================= - */ +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +======================================================================= +*/ package org.tensorflow; import static org.tensorflow.internal.c_api.global.tensorflow.TF_DeleteBuffer; @@ -23,13 +23,25 @@ import static org.tensorflow.internal.c_api.global.tensorflow.TF_Version; import com.google.protobuf.InvalidProtocolBufferException; +import java.util.Collections; +import java.util.IdentityHashMap; import java.util.Set; import java.util.stream.Collectors; +import org.bytedeco.javacpp.PointerPointer; import org.bytedeco.javacpp.PointerScope; import org.tensorflow.exceptions.TensorFlowException; +import org.tensorflow.internal.c_api.GradFunc; +import org.tensorflow.internal.c_api.GradOpRegistry; +import org.tensorflow.internal.c_api.NativeStatus; import org.tensorflow.internal.c_api.TF_Buffer; import org.tensorflow.internal.c_api.TF_Library; import org.tensorflow.internal.c_api.TF_Status; +import org.tensorflow.op.CustomGradient; +import org.tensorflow.op.RawCustomGradient; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.math.Add; import org.tensorflow.proto.framework.OpList; /** Static utility methods describing the TensorFlow runtime. */ @@ -138,4 +150,91 @@ private TensorFlow() {} throw e; } } + + /** + * Keeps references to custom gradient functions to prevent them from being deallocated. All + * access of this set should be synchronized on this class. + * + *

    Required for correctness + */ + private static final Set gradientFuncs = + Collections.newSetFromMap(new IdentityHashMap<>()); + + private static synchronized boolean hasGradient(String opType) { + try (PointerScope scope = new PointerScope()) { + NativeStatus status = + GradOpRegistry.Global().Lookup(opType, new GradFunc(new PointerPointer<>(1))); + return status.ok(); + } + } + + /** + * Register a custom gradient function for ops of {@code opType} type. + * + *

    Creates the gradient based off of a {@link GraphOperation}. To operate on the op input class + * instead use {@link CustomGradient}. + * + *

    Note that this only works with graph gradients, and will eventually be deprecated in favor + * of unified gradient support once it is fully supported by tensorflow core. + * + * @param opType the type of op to register the gradient for. Should usually be an {@code OP_NAME} + * field, i.e. {@link Add#OP_NAME}. + * @param gradient the gradient function to use + * @return {@code true} if the gradient was registered, {@code false} if there was already a + * gradient registered for this op + */ + public static synchronized boolean registerCustomGradient( + String opType, RawCustomGradient gradient) { + if (hasGradient(opType)) { + return false; + } + GradFunc g = RawCustomGradient.adapter(gradient); + GradOpRegistry.Global().Register(opType, g); + gradientFuncs.add(g); + return true; + } + + /** + * Register a custom gradient function for ops of {@code inputClass}'s op type. The actual op type + * is detected from the class's {@link OpInputsMetadata} annotation. As such, it only works on + * generated op classes or custom op classes with the correct annotations. To operate on the + * {@link org.tensorflow.GraphOperation} directly use {@link RawCustomGradient}. + * + * @param inputClass the inputs class of op to register the gradient for. + * @param gradient the gradient function to use + * @return {@code true} if the gradient was registered, {@code false} if there was already a + * gradient registered for this op + * @throws IllegalArgumentException if {@code inputClass} is not annotated with {@link + * OpInputsMetadata} or the op class is not annotated with {@link OpMetadata}. + */ + public static synchronized > boolean registerCustomGradient( + Class inputClass, CustomGradient gradient) { + OpInputsMetadata metadata = inputClass.getAnnotation(OpInputsMetadata.class); + + if (metadata == null) { + throw new IllegalArgumentException( + "Inputs Class " + + inputClass + + " does not have a OpInputsMetadata annotation. Was it generated by tensorflow/java? If it was, this is a bug."); + } + OpMetadata outputMetadata = metadata.outputsClass().getAnnotation(OpMetadata.class); + + if (outputMetadata == null) { + throw new IllegalArgumentException( + "Op Class " + + metadata.outputsClass() + + " does not have a OpMetadata annotation. Was it generated by tensorflow/java? If it was, this is a bug."); + } + + String opType = outputMetadata.opType(); + + if (hasGradient(opType)) { + return false; + } + + GradFunc g = CustomGradient.adapter(gradient, inputClass); + GradOpRegistry.Global().Register(opType, g); + gradientFuncs.add(g); + return true; + } } diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java index 369ccabce43..da9921daea3 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java @@ -1,4 +1,5 @@ -/* Copyright 2019-2021 The TensorFlow Authors. All Rights Reserved. +/* +Copyright 2019-2021 The TensorFlow Authors. All Rights Reserved. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. @@ -15,10 +16,17 @@ */ package org.tensorflow.internal.c_api.presets; +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; import java.util.List; import org.bytedeco.javacpp.ClassProperties; import org.bytedeco.javacpp.LoadEnabled; import org.bytedeco.javacpp.Loader; +import org.bytedeco.javacpp.annotation.Adapter; +import org.bytedeco.javacpp.annotation.Cast; import org.bytedeco.javacpp.annotation.NoException; import org.bytedeco.javacpp.annotation.Platform; import org.bytedeco.javacpp.annotation.Properties; @@ -46,8 +54,16 @@ // "tensorflow/c/env.h", "tensorflow/c/kernels.h", "tensorflow/c/ops.h", + "tensorflow_adapters.h", "tensorflow/c/eager/c_api.h", - "tensorflow/c/eager/c_api_experimental.h" + "tensorflow/c/eager/c_api_experimental.h", + "tensorflow/cc/framework/scope.h", + "tensorflow/cc/framework/grad_op_registry.h", + "tensorflow/core/platform/status.h", + "tensorflow/core/graph/graph.h", + "tensorflow/c/tf_status_helper.h", + "tensorflow/cc/framework/ops.h", + "tensorflow/c/c_api_internal.h", }, link = "tensorflow_cc@.2", preload = {"iomp5", "mklml", "mklml_intel", "tensorflow_framework@.2"}, @@ -272,12 +288,60 @@ public void init(ClassProperties properties) { @Override public void map(InfoMap infoMap) { infoMap + .put( + new Info("c_api_internal.h") + .linePatterns( + "struct TF_OperationDescription \\{", + "\\};", + "struct TF_Graph \\{", + "\\};", + "struct TF_Operation \\{", + "\\};", + "// Exposed helper functions", + "// End Exposed helper functions")) + .put( + new Info("graph.h") + .linePatterns( + "class Node \\{", "// Stores debug information associated with the Node.")) + .put(new Info("Node").cppTypes("tensorflow::Node").purify()) + .put( + new Info( + "tensorflow::NodeDef", + "tensorflow::OpDef", + "tensorflow::AttrSlice", + "tensorflow::Edge", + "tensorflow::EdgeSet", + "tensorflow::WhileContext", + "tensorflow::NodeProperties", + "protobuf::RepeatedPtrField", + "gtl::iterator_range", + "tensorflow::DataType", + "tensorflow::DataTypeVector", + "tensorflow::Node::set_original_node_names", + "tensorflow::Node::AddAttr", + "tensorflow::Node::ClearAttr", + "tensorflow::Node::input_node") + .skip()) + .put( + new Info("c_api.cc") + .linePatterns( + "// Helper functions -+", + "// Shape functions -+", + "static TF_OperationDescription\\* TF_NewOperationLocked\\(TF_Graph\\* graph,", + "\\}", + "static TF_Operation\\* TF_FinishOperationLocked\\(TF_OperationDescription\\* desc,", + "\\}")) + .put(new Info("OutputTensor", "TensorId", "tensorflow::AttrValue").skip()) .put( new Info("c_api_experimental.h") .linePatterns( "typedef struct TFE_OpAttrs TFE_OpAttrs;", "#define TFE_CUSTOM_DEVICE_VERSION 4")) - .put(new Info("TF_CAPI_EXPORT", "TF_Bool").cppTypes().annotations()) + .put( + new Info("TF_CAPI_EXPORT", "TF_Bool", "TF_GUARDED_BY", "TF_MUST_USE_RESULT") + .cppTypes() + .annotations()) + .put(new Info("TF_DISALLOW_COPY_AND_ASSIGN").skip()) .put( new Info("TF_Buffer::data") .javaText( @@ -305,13 +369,24 @@ public void map(InfoMap infoMap) { .put( new Info("TF_Graph") .pointerTypes("TF_Graph") - .base("org.tensorflow.internal.c_api.AbstractTF_Graph")) + .base("org.tensorflow.internal.c_api.AbstractTF_Graph") + .purify()) + .put(new Info("tensorflow::Graph").javaNames("NativeGraphPointer")) .put( new Info("TF_Graph::graph") - .javaText("public native @MemberGetter @ByRef Graph graph();")) + .javaText("public native @MemberGetter @ByRef NativeGraphPointer graph();")) .put( - new Info("TF_Graph::refiner") - .javaText("public native @MemberGetter @ByRef ShapeRefiner refiner();")) + new Info( + "TF_Graph::refiner", + "TF_Graph::mu", + "TF_Graph::sessions", + "TF_Graph::delete_requested") + .skip()) + .put( + new Info("std::unordered_map") + .pointerTypes("NameMap") + .define() + .javaText("public native long erase(@StdString BytePointer key);")) .put( new Info("TF_Function") .pointerTypes("TF_Function") @@ -320,7 +395,8 @@ public void map(InfoMap infoMap) { new Info("TF_ImportGraphDefOptions") .pointerTypes("TF_ImportGraphDefOptions") .base("org.tensorflow.internal.c_api.AbstractTF_ImportGraphDefOptions")) - .put(new Info("TF_Operation", "TF_WhileParams").purify()) + .put(new Info("TF_WhileParams").purify()) + .put(new Info("TF_Operation").purify()) .put( new Info("TF_Operation::node") .javaText("public native @MemberGetter @ByRef Node node();")) @@ -374,6 +450,79 @@ public void map(InfoMap infoMap) { "TFE_MonitoringSampler2", "TFE_CustomDeviceTensorHandle", "TFE_CustomDevice") + .skip()) + .put(new Info("TF_OperationDescription").pointerTypes("TF_OperationDescription").purify()) + .put(new Info("tensorflow::Scope").javaNames("TF_Scope")) + .put(new Info("tensorflow::NodeBuilder").pointerTypes("NodeBuilder")) + .put( + new Info("string", "tensorflow::string") + .annotations("@StdString") + .valueTypes("BytePointer", "String") + .pointerTypes("BytePointer")) + .put(new Info("absl::Span", "tensorflow::gtl::ArraySlice").annotations("@Span")) + .put( + new Info("std::vector").pointerTypes("NativeOutputVector").define()) + .put(new Info("tensorflow::Output").javaNames("NativeOutput")) + .put(new Info("tensorflow::Operation").javaNames("NativeOperation")) + .put(new Info("tensorflow::Status").javaNames("NativeStatus").purify()) + .put( + new Info("tensorflow::int32") + .cast() + .valueTypes("int") + .pointerTypes("IntPointer", "IntBuffer", "int[]")) + .put( + new Info( + "tensorflow::CompositeOpScopes", + "tensorflow::Input", + "tensorflow::InputList", + "tensorflow::OutputHash", + "tensorflow::StackFrame", + "tensorflow::StatusGroup", + "tensorflow::internal::TF_StatusDeleter", + "tensorflow::GraphDef", + "tensorflow::Scope::graph_as_shared_ptr", + "tensorflow::Scope::ToGraphDef", + "tensorflow::Scope::ToGraph", + "tensorflow::Scope::DoShapeInference", + "tensorflow::Scope::DisabledShapeInferenceScope", + "tensorflow::Scope::control_deps", + "tensorflow::Scope::WithKernelLabel", + "tensorflow::Scope::ClearColocation", + "tensorflow::Scope::ColocateWith", + "tensorflow::Scope::ColocateWith", + "tensorflow::Scope::WithXlaCluster", + "tensorflow::Scope::WithAssignedDevice", + "tensorflow::Scope::status", + "tensorflow::Scope::UpdateStatus", + "tensorflow::Status::code", + "tensorflow::CreateOutputWithScope", + "TF_OperationDescription::colocation_constraints", + "tensorflow::Operation::num_inputs", + "tensorflow::Operation::input_type", + "tensorflow::Operation::input", + "tensorflow::Operation::num_outputs", + "tensorflow::Operation::output_type", + "tensorflow::Operation::output", + "tensorflow::Operation::hash", + "tensorflow::Output::hash", + "tensorflow::Output::type", + "tensorflow::Status::GetAllPayloads", + "tensorflow::Status::ReplaceAllPayloads", + "tensorflow::Status::ErasePayload", + "tensorflow::Status::SetPayload", + "tensorflow::Status::GetPayload", + "tensorflow::Node::SetStackTrace", + "tensorflow::Node::GetStackTrace") .skip()); } + + @Documented + @Retention(RetentionPolicy.RUNTIME) + @Target({ElementType.METHOD, ElementType.PARAMETER}) + @Cast({"absl::Span", "&"}) + @Adapter("SpanAdapter") + public @interface Span { + + String value() default ""; + } } diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/CustomGradient.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/CustomGradient.java new file mode 100644 index 00000000000..2d573808222 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/CustomGradient.java @@ -0,0 +1,62 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ +package org.tensorflow.op; + +import java.util.List; +import org.tensorflow.Operand; +import org.tensorflow.Output; +import org.tensorflow.TensorFlow; +import org.tensorflow.internal.c_api.GradFunc; + +/** + * A custom gradient for ops of type {@link T}. Should be registered using {@link + * TensorFlow#registerCustomGradient(Class, CustomGradient)}. + * + *

    Creates the gradient based off of an instance of the op inputs class, which is created using + * reflection. To operate on the {@link org.tensorflow.GraphOperation} directly use {@link + * RawCustomGradient}. + * + *

    The type of the op is not checked here, but it is required to match the class given to the + * adapter. + * + * @param the type of op this gradient is for. + */ +@SuppressWarnings("rawtypes") +@FunctionalInterface +public interface CustomGradient { + + /** + * Calculate the gradients for {@code op}. + * + * @param tf the {@link Ops} instance used to create ops + * @param op the op to calculate the gradients of. + * @param gradInputs the gradients of the op's outputs. + * @return the gradients of the op's inputs. + */ + List> call(Ops tf, T op, List> gradInputs); + + /** + * Create an adapter for the custom gradient so that it can be used by native code. + * + *

    You should not be calling this yourself, use {@link TensorFlow#registerCustomGradient(Class, + * CustomGradient)}. + */ + public static > GradFunc adapter( + CustomGradient gradient, Class opClass) { + return new TypedGradientAdapter(gradient, opClass); + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/GradientScope.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/GradientScope.java new file mode 100644 index 00000000000..5254f86701f --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/GradientScope.java @@ -0,0 +1,159 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ +package org.tensorflow.op; + +import java.util.List; +import java.util.stream.Collectors; +import java.util.stream.StreamSupport; +import org.tensorflow.DeviceSpec; +import org.tensorflow.ExecutionEnvironment; +import org.tensorflow.Graph; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.internal.c_api.NativeOperation; +import org.tensorflow.internal.c_api.TF_Scope; + +/** A {@link Scope} implementation backed by a native scope. Only used for gradient declarations. */ +public final class GradientScope implements Scope { + + @Override + public ExecutionEnvironment env() { + return graph; + } + + @Override + public GradientScope withSubScope(String childScopeName) { + return new GradientScope(nativeScope.NewSubScope(childScopeName), graph, null, device); + } + + @Override + public GradientScope withName(String opName) { + return new GradientScope(nativeScope, graph, opName, device); + } + + @Override + public GradientScope withNameAsSubScope(String defaultName) { + if (opName == null) { + return withSubScope(defaultName); + } else { + return withSubScope(opName); + } + } + + @Override + public GradientScope withDevice(DeviceSpec newDevice) { + return new GradientScope( + nativeScope.WithDevice(newDevice.toString()), graph, newDevice.toString()); + } + + @Override + public Scope withInitScope() { + throw new IllegalStateException("Can't add init operations in a gradient scope"); + } + + @Override + public String makeOpName(String defaultName) { + String name = opName != null ? opName : defaultName; + return nativeScope.GetUniqueNameForOp(name); + } + + @Override + public String makeUnique(String id) { + return nativeScope.GetUniqueNameForOp(id); + } + + @Override + public void refreshNames() {} + + @Override + public GradientScope withControlDependencies(Iterable controls) { + List controlDeps = + StreamSupport.stream(controls.spliterator(), false).collect(Collectors.toList()); + NativeOperation ops = new NativeOperation(controlDeps.size()); + + for (int i = 0; i < controlDeps.size(); i++) { + Operation op = controlDeps.get(i).op(); + if (!(op instanceof GraphOperation)) { + throw new IllegalArgumentException("Can only add graph ops as control dependencies"); + } + ops.position(i) + .put(new NativeOperation(((GraphOperation) op).getUnsafeNativeHandle().node())); + } + + return new GradientScope( + nativeScope.WithControlDependencies(new NativeOperation(ops)), graph, device); + } + + @Override + public Scope withControlDependencyOps(Iterable controls) { + List controlDeps = + StreamSupport.stream(controls.spliterator(), false).collect(Collectors.toList()); + NativeOperation ops = new NativeOperation(controlDeps.size()); + + for (int i = 0; i < controlDeps.size(); i++) { + Operation op = controlDeps.get(i); + if (!(op instanceof GraphOperation)) { + throw new IllegalArgumentException("Can only add graph ops as control dependencies"); + } + ops.position(i) + .put(new NativeOperation(((GraphOperation) op).getUnsafeNativeHandle().node())); + } + + return new GradientScope( + nativeScope.WithControlDependencies(new NativeOperation(ops)), graph, device); + } + + @Override + public OperationBuilder apply(OperationBuilder builder) { + return builder; + } + + @Override + public void onOpCreated(Operation op) {} + + @Override + public String getDeviceString() { + if (device == null) { + throw new UnsupportedOperationException( + "Can't get device string for gradient scope unless it has been explicitly set"); + } else { + return device; + } + } + + @Override + public boolean isInit() { + return false; + } + + GradientScope(TF_Scope nativeScope, Graph graph, String device) { + this(nativeScope, graph, null, device); + } + + private GradientScope(TF_Scope nativeScope, Graph graph, String opName, String device) { + this.graph = graph; + this.nativeScope = nativeScope; + this.opName = opName; + this.device = device; + } + + private final Graph graph; + private final TF_Scope nativeScope; + private final String opName; + private final String device; +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/OpScope.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/OpScope.java new file mode 100644 index 00000000000..4e9aab97082 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/OpScope.java @@ -0,0 +1,153 @@ +/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +package org.tensorflow.op; + +import java.util.ArrayList; +import java.util.List; +import org.tensorflow.DeviceSpec; +import org.tensorflow.ExecutionEnvironment; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; + +/** + * A Java implementation of {@link Scope}. This is used in all cases except custom gradient + * definitions. + */ +public final class OpScope implements Scope { + + /** + * Create a new top-level scope. + * + * @param env The execution environment used by the scope. + */ + public OpScope(ExecutionEnvironment env) { + this(env, new NameScope(env), new ArrayList<>(), DeviceSpec.newBuilder().build(), false); + } + + @Override + public ExecutionEnvironment env() { + return env; + } + + @Override + public OpScope withSubScope(String childScopeName) { + return new OpScope( + env, nameScope.withSubScope(childScopeName, env), controlDependencies, deviceSpec, isInit); + } + + @Override + public OpScope withName(String opName) { + return new OpScope(env, nameScope.withName(opName), controlDependencies, deviceSpec, isInit); + } + + @Override + public OpScope withNameAsSubScope(String defaultName) { + return new OpScope( + env, + nameScope.withSubScope(nameScope.makeOpName(defaultName), env), + controlDependencies, + deviceSpec, + isInit); + } + + @Override + public OpScope withDevice(DeviceSpec newDevice) { + return new OpScope(env, nameScope, controlDependencies, newDevice, isInit); + } + + @Override + public OpScope withInitScope() { + return new OpScope(env.initEnv(), nameScope, new ArrayList<>(), deviceSpec, true); + } + + @Override + public String makeOpName(String defaultName) { + return nameScope.makeOpName(defaultName); + } + + @Override + public String makeUnique(String id) { + return nameScope.makeUnique(id); + } + + @Override + public void refreshNames() { + nameScope.importIdsFrom(env); + } + + private OpScope( + ExecutionEnvironment env, + NameScope nameScope, + List controlDependencies, + DeviceSpec deviceSpec, + boolean isInit) { + this.env = env; + this.nameScope = nameScope; + this.controlDependencies = controlDependencies; + this.deviceSpec = deviceSpec; + this.isInit = isInit; + } + + @Override + public Scope withControlDependencyOps(Iterable controls) { + ArrayList toAdd = new ArrayList<>(); + for (Operation control : controls) { + env.checkInput(control); + if (isInit && !env.isInitOp(control)) { + throw new IllegalArgumentException("Init scope can not have non-init control dependency."); + } + if (isInit || !env.isInitOp(control)) { + toAdd.add(control); + } + } + + return new OpScope(env, nameScope, toAdd, deviceSpec, isInit); + } + + @Override + public OperationBuilder apply(OperationBuilder builder) { + builder.setDevice(deviceSpec.toString()); + for (Operation control : controlDependencies) { + if (isInit || !env.isInitOp(control)) { + builder.addControlInput(control); + } + } + return builder; + } + + @Override + public void onOpCreated(Operation op) { + if (isInit) { + env.registerInitOp(op); + } + } + + @Override + public boolean isInit() { + return isInit; + } + + @Override + public String getDeviceString() { + return deviceSpec.toString(); + } + + private final ExecutionEnvironment env; + private final List controlDependencies; + private final NameScope nameScope; + private final DeviceSpec deviceSpec; + private final boolean isInit; +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawCustomGradient.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawCustomGradient.java new file mode 100644 index 00000000000..949891b7ff3 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawCustomGradient.java @@ -0,0 +1,59 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ +package org.tensorflow.op; + +import java.util.List; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Output; +import org.tensorflow.TensorFlow; +import org.tensorflow.internal.c_api.GradFunc; + +/** + * A custom gradient for an op of unspecified type. Should be registered using {@link + * TensorFlow#registerCustomGradient(String, RawCustomGradient)}. + * + *

    Creates the gradient based off of a {@link GraphOperation}. To operate on the op input class + * instead use {@link CustomGradient}. + * + *

    The op type of {@code op} will depend on the op type string passed to the registration method. + * Note that the registration method can be called more than once, resulting this gradient function + * being used for multiple different op types. + */ +@FunctionalInterface +public interface RawCustomGradient { + + /** + * Calculate the gradients for {@code op}. + * + * @param tf the {@link Ops} instance used to create ops + * @param op the op to calculate the gradients of. + * @param gradInputs the gradients of the op's outputs. + * @return the gradients of the op's inputs. + */ + List> call(Ops tf, GraphOperation op, List> gradInputs); + + /** + * Create an adapter for the custom gradient so that it can be used by native code. + * + *

    You should not be calling this yourself, use {@link + * TensorFlow#registerCustomGradient(String, RawCustomGradient)}. + */ + public static GradFunc adapter(RawCustomGradient gradient) { + return new RawGradientAdapter(gradient); + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawGradientAdapter.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawGradientAdapter.java new file mode 100644 index 00000000000..f1968014dae --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawGradientAdapter.java @@ -0,0 +1,71 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +======================================================================= + +*/ +package org.tensorflow.op; + +import java.util.List; +import org.bytedeco.javacpp.PointerScope; +import org.tensorflow.BaseGradientAdapter; +import org.tensorflow.Graph; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Output; +import org.tensorflow.internal.c_api.NativeOperation; +import org.tensorflow.internal.c_api.NativeOutputVector; +import org.tensorflow.internal.c_api.NativeStatus; +import org.tensorflow.internal.c_api.TF_Scope; + +/** A native adapter for {@link RawCustomGradient}. */ +final class RawGradientAdapter extends BaseGradientAdapter { + + private final RawCustomGradient gradient; + + RawGradientAdapter(RawCustomGradient gradient) { + super(); + this.gradient = gradient; + } + + @Override + public NativeStatus call( + TF_Scope scope, + NativeOperation op, + NativeOutputVector grad_inputs, + NativeOutputVector grad_outputs) { + try (PointerScope pointerScope = new PointerScope()) { + Graph g = Graph.findGraphForPointer(scope.graph()); + if (g == null) { + throw new IllegalStateException("No graph found for native gradient scope."); + } + + GraphOperation operation = BaseGradientAdapter.getGraphOp(g, op.node()); + + Scope nativeScope = new GradientScope(scope, g, null).withSubScope(operation.name()); + Ops tf = new Ops(nativeScope); + + List> gradInputs = BaseGradientAdapter.fromNativeOutputs(g, grad_inputs); + + // The graph locks are not re-entrant, so attempting to add an op to a graph that has been + // locked by the gradient builder will fail without this. + BaseGradientAdapter.useDangerousLockedBuilders(g, true); + List> gradOutputs = gradient.call(tf, operation, gradInputs); + BaseGradientAdapter.useDangerousLockedBuilders(g, false); + + BaseGradientAdapter.putToNativeOutputs(gradOutputs, grad_outputs); + } + return NativeStatus.OK(); + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawOp.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawOp.java index 53137a84b54..b55a0062c44 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawOp.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/RawOp.java @@ -59,8 +59,19 @@ public final String toString() { * Constructor. * * @param operation the underlying operation + * @param requiredType the type that the underlying operation must be */ - protected RawOp(Operation operation) { + protected RawOp(Operation operation, String requiredType) { + if (!requiredType.equals(operation.type())) { + throw new IllegalArgumentException( + "Can't create a " + + this.getClass() + + " from an operation with type \"" + + operation.type() + + "\", operation must have type \"" + + requiredType + + "\"."); + } this.operation = operation; } } diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/Scope.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/Scope.java index b4705ea95a3..bb5c03844bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/Scope.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/Scope.java @@ -15,8 +15,6 @@ */ package org.tensorflow.op; -import java.util.ArrayList; -import java.util.List; import java.util.stream.Collectors; import java.util.stream.StreamSupport; import org.tensorflow.DeviceSpec; @@ -80,24 +78,10 @@ * *

    Scope objects are not thread-safe. */ -public final class Scope { - - /** - * Create a new top-level scope. - * - *

    For internal use only, use {@link ExecutionEnvironment#baseScope()} if you need a - * base level scope. - * - * @param env The execution environment used by the scope. - */ - public Scope(ExecutionEnvironment env) { - this(env, new NameScope(env), new ArrayList<>(), DeviceSpec.newBuilder().build(), false); - } +public interface Scope { /** Returns the execution environment used by this scope. */ - public ExecutionEnvironment env() { - return env; - } + ExecutionEnvironment env(); /** * Returns a new scope where added operations will have the provided name prefix. @@ -112,10 +96,7 @@ public ExecutionEnvironment env() { * @return a new subscope * @throws IllegalArgumentException if the name is invalid */ - public Scope withSubScope(String childScopeName) { - return new Scope( - env, nameScope.withSubScope(childScopeName, env), controlDependencies, deviceSpec, isInit); - } + Scope withSubScope(String childScopeName); /** * Return a new scope that uses the provided name for an op. @@ -129,9 +110,7 @@ public Scope withSubScope(String childScopeName) { * @return a new Scope that uses opName for operations. * @throws IllegalArgumentException if the name is invalid */ - public Scope withName(String opName) { - return new Scope(env, nameScope.withName(opName), controlDependencies, deviceSpec, isInit); - } + Scope withName(String opName); /** * Returns a new scope where added operations will be prefixed by this scope's op name (set by @@ -149,14 +128,7 @@ public Scope withName(String opName) { * @return a new subscope * @throws IllegalArgumentException if the name is invalid */ - public Scope withNameAsSubScope(String defaultName) { - return new Scope( - env, - nameScope.withSubScope(nameScope.makeOpName(defaultName), env), - controlDependencies, - deviceSpec, - isInit); - } + Scope withNameAsSubScope(String defaultName); /** * Return a new scope that uses the provided device specification for an op. @@ -164,19 +136,13 @@ public Scope withNameAsSubScope(String defaultName) { *

    Operations created within this scope will place the created operations on the device(s) * matching the provided spec. * - * @param deviceSpec device specification for an operator in the returned scope + * @param newDevice device specification for an operator in the returned scope * @return a new Scope that uses opName for operations. */ - public Scope withDevice(DeviceSpec deviceSpec) { - return new Scope(env, nameScope, controlDependencies, deviceSpec, isInit); - } - - // TODO stop gradient recording in init scopes (once we have gradient recording) + Scope withDevice(DeviceSpec newDevice); /** Get an extension of this scope that generates initialization ops. */ - public Scope withInitScope() { - return new Scope(env.initEnv(), nameScope, new ArrayList<>(), deviceSpec, true); - } + Scope withInitScope(); /** * Create a unique name for an operator and reserves it, using a provided default if necessary. @@ -198,55 +164,17 @@ public Scope withInitScope() { * @return unique name for the operator. * @throws IllegalArgumentException if the default name is invalid. */ - public String makeOpName(String defaultName) { - return nameScope.makeOpName(defaultName); - } + String makeOpName(String defaultName); /** Makes a unique name from {@code id} and reserves it. */ - public String makeUnique(String id) { - return nameScope.makeUnique(id); - } - - /** - * Returns a builder to create a new {@link Operation}. - * - *

    Note that {@code name} is automatically made unique. - * - * @param type of the Operation (i.e., identifies the computation to be performed) - * @param name to refer to the created Operation in this environment scope. Is uniquified. - * @return an {@link OperationBuilder} to create an Operation when {@link - * OperationBuilder#build()} is invoked. If {@link OperationBuilder#build()} is not invoked, - * then some resources may leak. - */ - public OperationBuilder opBuilder(String type, String name) { - return env.opBuilder(type, makeOpName(name), this); - } - - public static boolean isValidOpName(String name) { - return NameScope.isValidName(name); - } + String makeUnique(String id); /** * Refresh the used name list (used for uniquifying names) from the underlying graph. * *

    Should be used if you made changes to the graph from non-{@code Scope} APIs. */ - public void refreshNames() { - nameScope.importIdsFrom(env); - } - - private Scope( - ExecutionEnvironment env, - NameScope nameScope, - List controlDependencies, - DeviceSpec deviceSpec, - boolean isInit) { - this.env = env; - this.nameScope = nameScope; - this.controlDependencies = controlDependencies; - this.deviceSpec = deviceSpec; - this.isInit = isInit; - } + void refreshNames(); /** * Returns a new scope where added operations will have the provided control dependencies. @@ -260,7 +188,7 @@ private Scope( * @param controls control dependencies for ops created with the returned scope * @return a new scope with the provided control dependencies */ - public Scope withControlDependencies(Iterable controls) { + default Scope withControlDependencies(Iterable controls) { return withControlDependencyOps( StreamSupport.stream(controls.spliterator(), false) .map(Op::op) @@ -279,19 +207,26 @@ public Scope withControlDependencies(Iterable controls) { * @param controls control dependencies for ops created with the returned scope * @return a new scope with the provided control dependencies */ - public Scope withControlDependencyOps(Iterable controls) { - ArrayList toAdd = new ArrayList<>(); - for (Operation control : controls) { - env.checkInput(control); - if (isInit && !env.isInitOp(control)) { - throw new IllegalArgumentException("Init scope can not have non-init control dependency."); - } - if (isInit || !env.isInitOp(control)) { - toAdd.add(control); - } - } + Scope withControlDependencyOps(Iterable controls); + + /** + * Returns a builder to create a new {@link Operation}. + * + *

    Note that {@code name} is automatically made unique. + * + * @param type of the Operation (i.e., identifies the computation to be performed) + * @param name to refer to the created Operation in this environment scope. Is uniquified. + * @return an {@link OperationBuilder} to create an Operation when {@link + * OperationBuilder#build()} is invoked. If {@link OperationBuilder#build()} is not invoked, + * then some resources may leak. + */ + default OperationBuilder opBuilder(String type, String name) { + return env().opBuilder(type, makeOpName(name), this); + } - return new Scope(env, nameScope, toAdd, deviceSpec, isInit); + /** Check whether {@code name} is a valid name for an operation. */ + static boolean isValidOpName(String name) { + return NameScope.isValidName(name); } /** @@ -302,40 +237,18 @@ public Scope withControlDependencyOps(Iterable controls) { * * @param builder OperationBuilder to add control inputs and device specification to */ - public OperationBuilder apply(OperationBuilder builder) { - builder.setDevice(deviceSpec.toString()); - for (Operation control : controlDependencies) { - if (isInit || !env.isInitOp(control)) { - builder.addControlInput(control); - } - } - return builder; - } + OperationBuilder apply(OperationBuilder builder); /** * Handle op creation, like registering it as an init op if the scope is init. * *

    FOR INTERNAL USE ONLY */ - public void onOpCreated(Operation op) { - if (isInit) { - env.registerInitOp(op); - } - } + void onOpCreated(Operation op); /** Returns device string from the scope. */ - public String getDeviceString() { - return deviceSpec.toString(); - } + String getDeviceString(); /** Get whether this scope is building init ops. */ - public boolean isInit() { - return isInit; - } - - private final ExecutionEnvironment env; - private final List controlDependencies; - private final NameScope nameScope; - private final DeviceSpec deviceSpec; - private final boolean isInit; + boolean isInit(); } diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/TypedGradientAdapter.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/TypedGradientAdapter.java new file mode 100644 index 00000000000..085174fcb4c --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/TypedGradientAdapter.java @@ -0,0 +1,82 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +======================================================================= + +*/ +package org.tensorflow.op; + +import java.lang.reflect.Constructor; +import java.lang.reflect.InvocationTargetException; +import java.util.List; +import org.bytedeco.javacpp.PointerScope; +import org.tensorflow.BaseGradientAdapter; +import org.tensorflow.Graph; +import org.tensorflow.Operand; +import org.tensorflow.Output; +import org.tensorflow.internal.c_api.NativeOperation; +import org.tensorflow.internal.c_api.NativeOutputVector; +import org.tensorflow.internal.c_api.NativeStatus; +import org.tensorflow.internal.c_api.TF_Scope; + +/** A native adapter for {@link CustomGradient}. */ +final class TypedGradientAdapter> extends BaseGradientAdapter { + + private final CustomGradient gradient; + private final Class opInputClass; + private final Constructor ctor; + + TypedGradientAdapter(CustomGradient gradient, Class opInputClass) { + super(); + this.gradient = gradient; + this.opInputClass = opInputClass; + //noinspection unchecked + this.ctor = (Constructor) this.opInputClass.getDeclaredConstructors()[0]; + } + + @Override + public NativeStatus call( + TF_Scope scope, + NativeOperation op, + NativeOutputVector grad_inputs, + NativeOutputVector grad_outputs) { + try (PointerScope pointerScope = new PointerScope()) { + Graph g = Graph.findGraphForPointer(scope.graph()); + if (g == null) { + throw new IllegalStateException("No graph found for native gradient scope."); + } + + T rawOp = ctor.newInstance(BaseGradientAdapter.getGraphOp(g, op.node())); + + Scope nativeScope = + new GradientScope(scope, g, null).withSubScope(rawOp.getOutputs().op().name()); + + Ops tf = new Ops(nativeScope); + + List> gradInputs = BaseGradientAdapter.fromNativeOutputs(g, grad_inputs); + + // The graph locks are not re-entrant, so attempting to add an op to a graph that has been + // locked by the gradient builder will fail without this. + BaseGradientAdapter.useDangerousLockedBuilders(g, true); + List> gradOutputs = gradient.call(tf, rawOp, gradInputs); + BaseGradientAdapter.useDangerousLockedBuilders(g, false); + + BaseGradientAdapter.putToNativeOutputs(gradOutputs, grad_outputs); + + } catch (InvocationTargetException | InstantiationException | IllegalAccessException e) { + throw new RuntimeException("Could not instantiate Op class " + opInputClass, e); + } + return NativeStatus.OK(); + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/annotation/OpInputsMetadata.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/annotation/OpInputsMetadata.java new file mode 100644 index 00000000000..e15f60e505a --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/annotation/OpInputsMetadata.java @@ -0,0 +1,36 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +======================================================================= + +*/ +package org.tensorflow.op.annotation; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; +import org.tensorflow.op.RawOp; + +/** + * An annotation to provide metadata about an op inputs accessor class. Should only be used by users + * on custom ops, will be generated for non-custom ops. + */ +@Target(ElementType.TYPE) +@Retention(RetentionPolicy.RUNTIME) +public @interface OpInputsMetadata { + + /** The main op class. */ + Class outputsClass(); +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/annotation/OpMetadata.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/annotation/OpMetadata.java new file mode 100644 index 00000000000..7f3bc929f73 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/annotation/OpMetadata.java @@ -0,0 +1,40 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +======================================================================= + +*/ +package org.tensorflow.op.annotation; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; +import org.tensorflow.op.RawOpInputs; + +/** + * An annotation to provide metadata about an op. Should only be used by users on custom ops, will + * be generated for non-custom ops. + */ +@Target(ElementType.TYPE) +@Retention(RetentionPolicy.RUNTIME) +public @interface OpMetadata { + + /** The type of the op in the TF runtime. */ + String opType(); + + /** The typesafe inputs class (which should be annotated with {@link OpInputsMetadata}). */ + @SuppressWarnings("rawtypes") + Class inputsClass(); +} diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Constant.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Constant.java index f9f6e00f0f6..246d222f303 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Constant.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Constant.java @@ -1356,7 +1356,7 @@ public Output asOutput() { } private Constant(Operation operation) { - super(operation); + super(operation, OP_NAME); output = operation.output(0); } diff --git a/tensorflow-core/tensorflow-core-api/src/main/resources/org/tensorflow/internal/c_api/include/tensorflow_adapters.h b/tensorflow-core/tensorflow-core-api/src/main/resources/org/tensorflow/internal/c_api/include/tensorflow_adapters.h new file mode 100644 index 00000000000..c3ad3ae672e --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/main/resources/org/tensorflow/internal/c_api/include/tensorflow_adapters.h @@ -0,0 +1,42 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + http://www.apache.org/licenses/LICENSE-2.0 + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. + ======================================================================= + */ + +#include "absl/types/span.h" + +using namespace absl; + +template class SpanAdapter { +public: + SpanAdapter(T const * ptr, typename Span::size_type size, void* owner) : ptr((T*)ptr), size(size), owner(owner), + arr2(ptr ? Span((T*)ptr, size) : Span()), arr(arr2) { } + SpanAdapter(const Span& arr) : ptr(0), size(0), owner(0), arr2(arr), arr(arr2) { } + SpanAdapter( Span& arr) : ptr(0), size(0), owner(0), arr(arr) { } + SpanAdapter(const Span* arr) : ptr(0), size(0), owner(0), arr(*(Span*)arr) { } + void assign(T* ptr, typename Span::size_type size, void* owner) { + this->ptr = ptr; + this->size = size; + this->owner = owner; + arr.set(ptr, size); + } + static void deallocate(void* owner) { free(owner); } + operator T*() { size = arr.size(); return (T*)arr.data(); } + operator Span&() { return arr; } + operator Span*() { return ptr ? &arr : 0; } + T* ptr; + typename Span::size_type size; + void* owner; + Span arr2; + Span& arr; +}; + diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/CustomGradientTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/CustomGradientTest.java new file mode 100644 index 00000000000..62626c35641 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/CustomGradientTest.java @@ -0,0 +1,81 @@ +/* + Copyright 2021 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ +package org.tensorflow; + +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertFalse; +import static org.junit.jupiter.api.Assertions.assertNotNull; +import static org.junit.jupiter.api.Assertions.assertTrue; + +import java.util.Arrays; +import org.junit.jupiter.api.Test; +import org.tensorflow.ndarray.index.Indices; +import org.tensorflow.op.Ops; +import org.tensorflow.op.dtypes.Cast; +import org.tensorflow.op.nn.NthElement; +import org.tensorflow.proto.framework.DataType; +import org.tensorflow.types.TFloat32; + +public class CustomGradientTest { + + @Test + public void testAlreadyExisting() { + assertFalse( + TensorFlow.registerCustomGradient( + Cast.Inputs.class, + (tf, op, gradInputs) -> { + Operand out = gradInputs.get(0); + Operand a = tf.stridedSlice(out, Indices.slice(0, 1)); + Operand b = tf.stridedSlice(out, Indices.slice(1, 2)); + return Arrays.asList(a, b, tf.constant(0f)); + })); + } + + @Test + public void testCustomGradient() { + try (Graph g = new Graph(); + Session s = new Session(g)) { + assertTrue( + TensorFlow.registerCustomGradient( + NthElement.Inputs.class, + (tf, op, gradInputs) -> + Arrays.asList(tf.withName("inAGrad").constant(0f), tf.constant(0f)))); + + Ops tf = Ops.create(g); + Output x = tf.placeholder(TFloat32.class).output(); + Output y = + tf.math.add(tf.nn.nthElement(x, tf.constant(2)), tf.constant(4f)).asOutput(); + + Output[] grads0 = g.addGradients(y, toArray(x)); + assertNotNull(grads0); + assertEquals(1, grads0.length); + assertEquals(DataType.DT_FLOAT, grads0[0].dataType()); + + try (TFloat32 c1 = TFloat32.vectorOf(3.0f, 2.0f, 1.0f, 0.0f); + AutoCloseableList outputs = + new AutoCloseableList<>(s.runner().feed(x, c1).fetch(grads0[0]).run())) { + + assertEquals(1, outputs.size()); + assertEquals(0.0f, ((TFloat32) outputs.get(0)).getFloat(), 0.0f); + } + } + } + + private static Output[] toArray(Output... outputs) { + return outputs; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphOperationTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphOperationTest.java index f52166aaac3..ab5d9232593 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphOperationTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphOperationTest.java @@ -45,6 +45,7 @@ public class GraphOperationTest { public void outputListLengthFailsOnInvalidName() { try (Graph g = new Graph()) { Ops tf = Ops.create(g); + Operation op = tf.math.add(tf.constant(1), tf.constant(2)).op(); assertEquals(1, op.outputListLength("z")); diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphTest.java index 464701306f8..154d3903dcd 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/GraphTest.java @@ -147,7 +147,7 @@ private static void validateImportedGraph(Graph g, String prefix) { public void iterateOverOperations() { try (Graph g = new Graph()) { Ops tf = Ops.create(g); - Iterator iterator = g.operations(); + Iterator iterator = g.operations(); HashSet operations; assertFalse(iterator.hasNext()); @@ -275,7 +275,6 @@ public void addGradientsToGraph() { .fetch(grads1[0]) .fetch(grads1[1]) .run())) { - assertEquals(3, outputs.size()); assertEquals(108.0f, ((TFloat32) outputs.get(0)).getFloat(), 0.0f); assertEquals(6.0f, ((TFloat32) outputs.get(1)).getFloat(), 0.0f); diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SessionTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SessionTest.java index 3575da6c8c2..95da0520f7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SessionTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/SessionTest.java @@ -264,7 +264,7 @@ public static void testFetchVariable() { private static int numOperations(Graph g) { int numOperations = 0; - for (Iterator it = g.operations(); it.hasNext(); ) { + for (Iterator it = g.operations(); it.hasNext(); ) { Operation o = it.next(); numOperations++; } diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/RawOpTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/RawOpTest.java index 5d523a986ad..d58e349b7e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/RawOpTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/RawOpTest.java @@ -18,18 +18,30 @@ import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertFalse; import static org.junit.jupiter.api.Assertions.assertNotEquals; +import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertTrue; import java.util.HashSet; import java.util.Set; import org.junit.jupiter.api.Test; import org.tensorflow.Graph; +import org.tensorflow.Operand; import org.tensorflow.Output; +import org.tensorflow.op.math.Add; import org.tensorflow.types.TInt32; /** Unit tests for {@link RawOp} */ public class RawOpTest { + @Test + public void wrongOpType() { + try (Graph g = new Graph()) { + Ops tf = Ops.create(g); + Operand a = tf.constant(10); + assertThrows(IllegalArgumentException.class, () -> new Add(a.op())); + } + } + @Test public void equalsHashcode() { try (Graph g = new Graph()) { @@ -38,10 +50,10 @@ public void equalsHashcode() { Output array = tf.constant(new int[2]).asOutput(); RawOp test1 = - new RawOp(g.baseScope().opBuilder("Shape", "shape1").addInput(array).build()) {}; + new RawOp(g.baseScope().opBuilder("Shape", "shape1").addInput(array).build(), "Shape") {}; RawOp test2 = - new RawOp(g.baseScope().opBuilder("Shape", "shape2").addInput(array).build()) {}; - RawOp test3 = new RawOp(test1.operation) {}; + new RawOp(g.baseScope().opBuilder("Shape", "shape2").addInput(array).build(), "Shape") {}; + RawOp test3 = new RawOp(test1.operation, test1.operation.type()) {}; // equals() tests assertNotEquals(test1, test2); diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/ScopeTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/ScopeTest.java index 6b37a908f8e..e2213c7ab1c 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/ScopeTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/ScopeTest.java @@ -50,7 +50,7 @@ public void testSeparateOps() { @Test public void basicNames() { try (Graph g = new Graph()) { - Scope root = new Scope(g); + Scope root = new OpScope(g); assertEquals("add", root.makeOpName("add")); assertEquals("add_1", root.makeOpName("add")); assertEquals("add_2", root.makeOpName("add")); @@ -61,7 +61,7 @@ public void basicNames() { @Test public void hierarchicalNames() { try (Graph g = new Graph()) { - Scope root = new Scope(g); + Scope root = new OpScope(g); Scope child = root.withSubScope("child"); assertEquals("child/add", child.makeOpName("add")); assertEquals("child/add_1", child.makeOpName("add")); @@ -87,7 +87,7 @@ public void hierarchicalNames() { @Test public void scopeAndOpNames() { try (Graph g = new Graph()) { - Scope root = new Scope(g); + Scope root = new OpScope(g); Scope child = root.withSubScope("child"); @@ -100,7 +100,7 @@ public void scopeAndOpNames() { @Test public void validateNames() { try (Graph g = new Graph()) { - Scope root = new Scope(g); + Scope root = new OpScope(g); final String[] invalid_names = { "_", "-", "-x", // Names are constrained to start with [A-Za-z0-9.] @@ -137,7 +137,7 @@ public void validateNames() { @Test public void basic() { try (Graph g = new Graph()) { - Scope s = new Scope(g); + Scope s = new OpScope(g); Const c1 = Const.create(s, 42); assertEquals("Const", c1.output().op().name()); Const c2 = Const.create(s, 7); @@ -152,7 +152,7 @@ public void basic() { @Test public void hierarchy() { try (Graph g = new Graph()) { - Scope root = new Scope(g); + Scope root = new OpScope(g); Scope child = root.withSubScope("child"); assertEquals("child/Const", Const.create(child, 42).output().op().name()); assertEquals("child/four", Const.create(child.withName("four"), 4).output().op().name()); @@ -163,7 +163,7 @@ public void hierarchy() { public void composite() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope s = new Scope(g); + Scope s = new OpScope(g); Output data = Const.create(s.withName("data"), new int[] {600, 470, 170, 430, 300}).output(); @@ -195,6 +195,7 @@ public void composite() { // "handwritten" sample operator classes private static final class Const { + private final Output output; static Const create(Scope s, int v) { @@ -224,6 +225,7 @@ Output output() { } private static final class Mean { + private final Output output; static Mean create(Scope s, Output input, Output reductionIndices) { @@ -241,6 +243,7 @@ Output output() { } private static final class SquaredDifference { + private final Output output; static SquaredDifference create(Scope s, Output x, Output y) { @@ -262,6 +265,7 @@ Output output() { } private static final class Variance { + private final Output output; static Variance create(Scope base, Output x) { diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskTest.java index 7c5210c0f2d..246b44b8077 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskTest.java @@ -1,19 +1,19 @@ /* - Copyright 2021 The TensorFlow Authors. All Rights Reserved. + Copyright 2021 The TensorFlow Authors. All Rights Reserved. - Licensed under the Apache License, Version 2.0 (the "License"); - you may not use this file except in compliance with the License. - You may obtain a copy of the License at +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at - http://www.apache.org/licenses/LICENSE-2.0 + http://www.apache.org/licenses/LICENSE-2.0 - Unless required by applicable law or agreed to in writing, software - distributed under the License is distributed on an "AS IS" BASIS, - WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - See the License for the specific language governing permissions and - limitations under the License. - ============================================================================== - */ +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ package org.tensorflow.op.core; import static org.junit.jupiter.api.Assertions.assertEquals; @@ -23,23 +23,24 @@ import org.tensorflow.Operand; import org.tensorflow.Session; import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Scope; import org.tensorflow.types.TBool; -import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; public class BooleanMaskTest { @Test - public void testBooleanMask(){ + public void testBooleanMask() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand input = Constant.arrayOf(scope, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9); Operand input2 = ExpandDims.create(scope, input, Constant.scalarOf(scope, 0)); - Operand mask = Constant.arrayOf(scope, true, true, false, false, true, true, true, false, false, false); + Operand mask = + Constant.arrayOf(scope, true, true, false, false, true, true, true, false, false, false); Operand output1 = BooleanMask.create(scope, input, mask); Operand output2 = BooleanMask.create(scope, input2, mask, BooleanMask.axis(1)); diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskUpdateTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskUpdateTest.java index c2b514bfdb6..16c14f7a9a3 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskUpdateTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/BooleanMaskUpdateTest.java @@ -1,31 +1,31 @@ /* - Copyright 2021 The TensorFlow Authors. All Rights Reserved. + Copyright 2021 The TensorFlow Authors. All Rights Reserved. - Licensed under the Apache License, Version 2.0 (the "License"); - you may not use this file except in compliance with the License. - You may obtain a copy of the License at +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at - http://www.apache.org/licenses/LICENSE-2.0 + http://www.apache.org/licenses/LICENSE-2.0 - Unless required by applicable law or agreed to in writing, software - distributed under the License is distributed on an "AS IS" BASIS, - WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - See the License for the specific language governing permissions and - limitations under the License. - ============================================================================== - */ +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +============================================================================== +*/ package org.tensorflow.op.core; import static org.junit.jupiter.api.Assertions.assertEquals; import java.util.List; - import org.junit.jupiter.api.Test; import org.tensorflow.Graph; import org.tensorflow.Operand; import org.tensorflow.Session; import org.tensorflow.Tensor; import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Scope; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt32; @@ -36,17 +36,19 @@ public class BooleanMaskUpdateTest { public void testBooleanMaskUpdateSlice() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); - Operand input = Constant.tensorOf(scope, new int[][]{{0, 0, 0}, {1, 1, 1}, {2, 2, 2}}); + Operand input = + Constant.tensorOf(scope, new int[][] {{0, 0, 0}, {1, 1, 1}, {2, 2, 2}}); Operand mask = Constant.arrayOf(scope, true, false, false); - Operand value = Constant.tensorOf(scope, new int[][]{{-1, -1, -1}}); + Operand value = Constant.tensorOf(scope, new int[][] {{-1, -1, -1}}); Operand output = BooleanMaskUpdate.create(scope, input, mask, value); - Operand bcastOutput = BooleanMaskUpdate.create(scope, input, mask, Constant.scalarOf(scope, -1)); + Operand bcastOutput = + BooleanMaskUpdate.create(scope, input, mask, Constant.scalarOf(scope, -1)); List results = sess.runner().fetch(output).fetch(bcastOutput).run(); try (TInt32 result = (TInt32) results.get(0); @@ -73,17 +75,19 @@ public void testBooleanMaskUpdateSlice() { public void testBooleanMaskUpdateSliceWithBroadcast() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); - Operand input = Constant.tensorOf(scope, new int[][]{{0, 0, 0}, {1, 1, 1}, {2, 2, 2}}); + Operand input = + Constant.tensorOf(scope, new int[][] {{0, 0, 0}, {1, 1, 1}, {2, 2, 2}}); Operand mask = Constant.arrayOf(scope, true, false, false); - Operand value = Constant.vectorOf(scope, new int[]{-1, -1, -1}); + Operand value = Constant.vectorOf(scope, new int[] {-1, -1, -1}); Operand output = BooleanMaskUpdate.create(scope, input, mask, value); - Operand bcastOutput = BooleanMaskUpdate.create(scope, input, mask, Constant.scalarOf(scope, -1)); + Operand bcastOutput = + BooleanMaskUpdate.create(scope, input, mask, Constant.scalarOf(scope, -1)); List results = sess.runner().fetch(output).fetch(bcastOutput).run(); try (TInt32 result = (TInt32) results.get(0); @@ -110,18 +114,22 @@ public void testBooleanMaskUpdateSliceWithBroadcast() { public void testBooleanMaskUpdateAxis() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); - Operand input = Constant.tensorOf(scope, new int[][][]{{{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}}}); + Operand input = + Constant.tensorOf(scope, new int[][][] {{{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}}}); - Operand mask = Constant.arrayOf(scope, true, true, false, false, true, true, true, false, false, false); + Operand mask = + Constant.arrayOf(scope, true, true, false, false, true, true, true, false, false, false); Operand value = Constant.arrayOf(scope, -1, -1, -1, -1, -1); - Operand output = BooleanMaskUpdate.create(scope, input, mask, value, BooleanMaskUpdate.axis(2)); + Operand output = + BooleanMaskUpdate.create(scope, input, mask, value, BooleanMaskUpdate.axis(2)); - Operand bcastOutput = BooleanMaskUpdate - .create(scope, input, mask, Constant.scalarOf(scope, -1), BooleanMaskUpdate.axis(2)); + Operand bcastOutput = + BooleanMaskUpdate.create( + scope, input, mask, Constant.scalarOf(scope, -1), BooleanMaskUpdate.axis(2)); List results = sess.runner().fetch(output).fetch(bcastOutput).run(); try (TInt32 result = (TInt32) results.get(0); diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ConstantTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ConstantTest.java index 6df73261867..5c413b3abeb 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ConstantTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ConstantTest.java @@ -38,6 +38,7 @@ import org.tensorflow.ndarray.buffer.FloatDataBuffer; import org.tensorflow.ndarray.buffer.IntDataBuffer; import org.tensorflow.ndarray.buffer.LongDataBuffer; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Ops; import org.tensorflow.op.Scope; import org.tensorflow.types.TBfloat16; @@ -62,7 +63,7 @@ public void createInts() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Constant op1 = Constant.tensorOf(scope, shape, buffer); Constant op2 = Constant.tensorOf(scope, array); try (AutoCloseableList t = @@ -81,7 +82,7 @@ public void createFloats() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Constant op1 = Constant.tensorOf(scope, shape, buffer); Constant op2 = Constant.tensorOf(scope, array); try (AutoCloseableList t = @@ -100,7 +101,7 @@ public void createDoubles() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Constant op1 = Constant.tensorOf(scope, shape, buffer); Constant op2 = Constant.tensorOf(scope, array); try (AutoCloseableList t = @@ -119,7 +120,7 @@ public void createLongs() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Constant op1 = Constant.tensorOf(scope, shape, buffer); Constant op2 = Constant.tensorOf(scope, array); try (AutoCloseableList t = @@ -138,7 +139,7 @@ public void createStrings() throws IOException { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Constant op1 = Constant.tensorOf(scope, shape, buffer); Constant op2 = Constant.tensorOf(scope, array); try (AutoCloseableList t = diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/IndexingTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/IndexingTest.java index 9a66d2445d2..7fd64957700 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/IndexingTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/IndexingTest.java @@ -21,52 +21,55 @@ import org.tensorflow.Graph; import org.tensorflow.Session; import org.tensorflow.ndarray.Shape; -import org.tensorflow.ndarray.index.Indices; import org.tensorflow.ndarray.index.Index; +import org.tensorflow.ndarray.index.Indices; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Scope; import org.tensorflow.types.TFloat32; public class IndexingTest { // [2, 1:2, :, tf.newaxis, ..., :4, 4::2] - private static final Index[] slice = new Index[]{ - Indices.at(2), - Indices.at(1, true), - Indices.all(), - Indices.newAxis(), - Indices.ellipsis(), - Indices.sliceTo( 4), - Indices.sliceFrom(4, 2) - }; + private static final Index[] slice = + new Index[] { + Indices.at(2), + Indices.at(1, true), + Indices.all(), + Indices.newAxis(), + Indices.ellipsis(), + Indices.sliceTo(4), + Indices.sliceFrom(4, 2) + }; @Test public void testIndexMerge() { StridedSliceHelper.StridedSliceArgs args = StridedSliceHelper.mergeIndexes(slice); - assertArrayEquals(new int[]{2, 1, 0, 0, 0, 0, 4}, args.begin); - assertArrayEquals(new int[]{3, 2, 0, 0, 0, 4, 0}, args.end); - assertArrayEquals(new int[]{1, 1, 1, 1, 1, 1, 2}, args.strides); + assertArrayEquals(new int[] {2, 1, 0, 0, 0, 0, 4}, args.begin); + assertArrayEquals(new int[] {3, 2, 0, 0, 0, 4, 0}, args.end); + assertArrayEquals(new int[] {1, 1, 1, 1, 1, 1, 2}, args.strides); assertEquals(0b0100100, args.beginMask); assertEquals(0b1000100, args.endMask); assertEquals(0b0010000, args.ellipsisMask); assertEquals(0b0001000, args.newAxisMask); assertEquals(0b0000001, args.shrinkAxisMask); - } @Test - public void testStridedSliceIndex(){ + public void testStridedSliceIndex() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {10, 10, 10, 10, 10, 10, 10, 10}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TFloat32.class); StridedSlice output = StridedSliceHelper.stridedSlice(scope, op, slice); try (TFloat32 result = (TFloat32) sess.runner().fetch(output.asOutput()).run().get(0)) { // expected shape from Python tensorflow - assertEquals(Shape.of(1, 10, 1, 10, 10, 10, 4, 3), result.shape(), "Slice index didn't match expected (Python)"); + assertEquals( + Shape.of(1, 10, 1, 10, 10, 10, 4, 3), + result.shape(), + "Slice index didn't match expected (Python)"); } } } - } diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ShapesTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ShapesTest.java index 39c04c942af..27bfa5fffb6 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ShapesTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ShapesTest.java @@ -22,6 +22,7 @@ import org.tensorflow.Graph; import org.tensorflow.Operand; import org.tensorflow.Session; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Scope; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -34,7 +35,7 @@ public class ShapesTest { public void testFlatten_Operand() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Shape expResult = Shape.create(scope, operand, TInt64.class); Operand reshaped = @@ -43,12 +44,11 @@ public void testFlatten_Operand() { Shape tfshape = Shape.create(scope, actual, TInt64.class); AtomicInteger index = new AtomicInteger(); - try (TInt64 result1 = (TInt64)session.runner().fetch(tfshape.asOutput()).run().get(0); - TInt64 result2 = (TInt64)session.runner().fetch(expResult.asOutput()).run().get(0)) { + try (TInt64 result1 = (TInt64) session.runner().fetch(tfshape.asOutput()).run().get(0); + TInt64 result2 = (TInt64) session.runner().fetch(expResult.asOutput()).run().get(0)) { result1 .scalars() - .forEach( - s -> assertEquals(result2.getLong(index.getAndIncrement()), s.getLong())); + .forEach(s -> assertEquals(result2.getLong(index.getAndIncrement()), s.getLong())); } } } @@ -57,7 +57,7 @@ public void testFlatten_Operand() { @Test public void testFlatten_Shape() { try (EagerSession session = EagerSession.create()) { - Scope scope = new Scope(session); + Scope scope = new OpScope(session); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Shape expShape = Shape.create(scope, operand, TInt64.class); Operand actual = @@ -70,9 +70,7 @@ public void testFlatten_Shape() { .asTensor() .scalars() .forEach( - s -> - assertEquals( - expShape.asTensor().getLong(index.getAndIncrement()), s.getLong())); + s -> assertEquals(expShape.asTensor().getLong(index.getAndIncrement()), s.getLong())); } } @@ -81,7 +79,7 @@ public void testFlatten_Shape() { public void testSize_Shape() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2, 1})); @@ -89,7 +87,7 @@ public void testSize_Shape() { Operand size = Shapes.size(scope, tfshape, TInt64.class); AtomicInteger index = new AtomicInteger(); - try (TInt64 result1 = (TInt64)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt64 result1 = (TInt64) session.runner().fetch(size.asOutput()).run().get(0)) { result1.scalars().forEach(s -> assertEquals(8, s.getLong())); } } @@ -100,24 +98,24 @@ public void testSize_Shape() { public void testSize_Shape_Operand() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2, 1})); Shape tfshape = Shape.create(scope, actual); Operand size = Shapes.size(scope, tfshape, Constant.scalarOf(scope, 0)); - try (TInt32 result = (TInt32)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(size.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(4, s.getInt())); } size = Shapes.size(scope, tfshape, Constant.scalarOf(scope, 1)); - try (TInt32 result = (TInt32)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(size.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(2, s.getInt())); } size = Shapes.size(scope, tfshape, Constant.scalarOf(scope, 2)); - try (TInt32 result = (TInt32)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(size.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(1, s.getInt())); } } @@ -128,23 +126,23 @@ public void testSize_Shape_Operand() { public void testSize_Operand_Operand() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2, 1})); Operand size = Shapes.size(scope, actual, Constant.scalarOf(scope, 0)); - try (TInt32 result = (TInt32)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(size.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(4, s.getInt())); } size = Shapes.size(scope, actual, Constant.scalarOf(scope, 1)); - try (TInt32 result = (TInt32)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(size.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(2, s.getInt())); } size = Shapes.size(scope, actual, Constant.scalarOf(scope, 2)); - try (TInt32 result = (TInt32)session.runner().fetch(size.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(size.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(1, s.getInt())); } } @@ -155,14 +153,14 @@ public void testSize_Operand_Operand() { public void testNumDimensions() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2, 1})); Shape tfshape = Shape.create(scope, actual); Operand nDims = Shapes.numDimensions(scope, tfshape); - try (TInt32 result = (TInt32)session.runner().fetch(nDims.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(nDims.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(3, s.getInt())); } } @@ -172,7 +170,7 @@ public void testNumDimensions() { @Test public void testReduceDims_Operand_Operand() { try (EagerSession session = EagerSession.create()) { - Scope scope = new Scope(session); + Scope scope = new OpScope(session); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {2, 2, 2})); @@ -197,7 +195,7 @@ public void testReduceDims_Operand_Operand() { @Test public void testReduceDims_Shape_Operand() { try (EagerSession session = EagerSession.create()) { - Scope scope = new Scope(session); + Scope scope = new OpScope(session); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {2, 2, 2})); @@ -249,7 +247,7 @@ public void testReduceDims_Shape_Operand() { public void testSqueeze() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 1, 2, 1})); @@ -258,7 +256,7 @@ public void testSqueeze() { Operand squeezed = Shapes.squeeze(scope, tfshape); AtomicInteger index = new AtomicInteger(); int[] expected = {4, 2}; - try (TInt32 result = (TInt32)session.runner().fetch(squeezed.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(squeezed.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -274,7 +272,7 @@ public void testSqueeze() { public void testHead() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 1, 2, 1})); @@ -283,7 +281,7 @@ public void testHead() { Operand head = Shapes.head(scope, tfshape); AtomicInteger index = new AtomicInteger(); int[] expected = {4}; - try (TInt32 result = (TInt32)session.runner().fetch(head.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(head.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -299,7 +297,7 @@ public void testHead() { public void testTake() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 1, 2, 1})); @@ -308,7 +306,7 @@ public void testTake() { Operand take = Shapes.take(scope, tfshape, Constant.scalarOf(scope, 2)); AtomicInteger index = new AtomicInteger(); int[] expected = {4, 1}; - try (TInt32 result = (TInt32)session.runner().fetch(take.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(take.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -324,7 +322,7 @@ public void testTake() { public void testTail() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 1, 2, 1})); @@ -333,7 +331,7 @@ public void testTail() { Operand tail = Shapes.tail(scope, tfshape); AtomicInteger index = new AtomicInteger(); int[] expected = {1}; - try (TInt32 result = (TInt32)session.runner().fetch(tail.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(tail.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -349,7 +347,7 @@ public void testTail() { public void testTakeLast() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 1, 2, 1})); @@ -358,7 +356,7 @@ public void testTakeLast() { Operand takeLast = Shapes.takeLast(scope, tfshape, Constant.scalarOf(scope, 3)); AtomicInteger index = new AtomicInteger(); int[] expected = {1, 2, 1}; - try (TInt32 result = (TInt32)session.runner().fetch(takeLast.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(takeLast.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -374,15 +372,16 @@ public void testTakeLast() { public void testPrependInt() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); - Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); + Operand actual = + Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); Shape tfshape = Shape.create(scope, actual); Operand prepend = Shapes.prepend(scope, tfshape, 3); AtomicInteger index = new AtomicInteger(); int[] expected = {3, 4, 2}; - try (TInt32 result = (TInt32)session.runner().fetch(prepend.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(prepend.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -398,15 +397,16 @@ public void testPrependInt() { public void testPrependLong() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); - Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); + Operand actual = + Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); Shape tfshape = Shape.create(scope, actual, TInt64.class); Operand prepend = Shapes.prepend(scope, tfshape, 1L); AtomicInteger index = new AtomicInteger(); long[] expected = {1, 4, 2}; - try (TInt64 result = (TInt64)session.runner().fetch(prepend.asOutput()).run().get(0)) { + try (TInt64 result = (TInt64) session.runner().fetch(prepend.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -422,7 +422,7 @@ public void testPrependLong() { public void testPrependShapeTInt32() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand1 = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual1 = Reshape.create(scope, operand1, Constant.vectorOf(scope, new long[] {4, 2})); @@ -435,7 +435,7 @@ public void testPrependShapeTInt32() { Operand prepend = Shapes.prepend(scope, tfshape1, tfshape2); AtomicInteger index = new AtomicInteger(); int[] expected = {2, 4, 4, 2}; - try (TInt32 result = (TInt32)session.runner().fetch(prepend.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(prepend.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -451,7 +451,7 @@ public void testPrependShapeTInt32() { public void testPrependShapeTInt64() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand1 = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual1 = Reshape.create(scope, operand1, Constant.vectorOf(scope, new long[] {4, 2})); @@ -464,7 +464,7 @@ public void testPrependShapeTInt64() { Operand prepend = Shapes.prepend(scope, tfshape1, tfshape2); AtomicInteger index = new AtomicInteger(); long[] expected = {2, 4, 4, 2}; - try (TInt64 result = (TInt64)session.runner().fetch(prepend.asOutput()).run().get(0)) { + try (TInt64 result = (TInt64) session.runner().fetch(prepend.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -480,15 +480,16 @@ public void testPrependShapeTInt64() { public void testAppendLong() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); - Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); + Operand actual = + Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); Shape tfshape = Shape.create(scope, actual, TInt64.class); Operand append = Shapes.append(scope, tfshape, 2L); AtomicInteger index = new AtomicInteger(); long[] expected = {4L, 2L, 2L}; - try (TInt64 result = (TInt64)session.runner().fetch(append.asOutput()).run().get(0)) { + try (TInt64 result = (TInt64) session.runner().fetch(append.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -504,15 +505,16 @@ public void testAppendLong() { public void testAppendInt() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); - Operand actual = Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); + Operand actual = + Reshape.create(scope, operand, Constant.vectorOf(scope, new long[] {4, 2})); Shape tfshape = Shape.create(scope, actual); Operand append = Shapes.append(scope, tfshape, 2); AtomicInteger index = new AtomicInteger(); int[] expected = {4, 2, 2}; - try (TInt32 result = (TInt32)session.runner().fetch(append.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(append.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -528,7 +530,7 @@ public void testAppendInt() { public void testAppendShapeTInt32() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand1 = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual1 = Reshape.create(scope, operand1, Constant.vectorOf(scope, new long[] {4, 2})); @@ -541,7 +543,7 @@ public void testAppendShapeTInt32() { Operand append = Shapes.append(scope, tfshape1, tfshape2); AtomicInteger index = new AtomicInteger(); int[] expected = {4, 2, 2, 4}; - try (TInt32 result = (TInt32)session.runner().fetch(append.asOutput()).run().get(0)) { + try (TInt32 result = (TInt32) session.runner().fetch(append.asOutput()).run().get(0)) { result .scalars() .forEach( @@ -557,7 +559,7 @@ public void testAppendShapeTInt32() { public void testAppendShapeTInt64() { try (Graph g = new Graph(); Session session = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); Operand operand1 = Constant.arrayOf(scope, new float[] {1, 2, 3, 4, 5, 6, 7, 8}); Operand actual1 = Reshape.create(scope, operand1, Constant.vectorOf(scope, new long[] {4, 2})); @@ -570,7 +572,7 @@ public void testAppendShapeTInt64() { Operand append = Shapes.append(scope, tfshape1, tfshape2); AtomicInteger index = new AtomicInteger(); long[] expected = {4, 2, 2, 4}; - try (TInt64 result = (TInt64)session.runner().fetch(append.asOutput()).run().get(0)) { + try (TInt64 result = (TInt64) session.runner().fetch(append.asOutput()).run().get(0)) { result .scalars() .forEach( diff --git a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ZerosTest.java b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ZerosTest.java index 4121baf3af1..b4d36702c93 100644 --- a/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ZerosTest.java +++ b/tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/op/core/ZerosTest.java @@ -23,6 +23,7 @@ import org.junit.jupiter.api.Test; import org.tensorflow.Graph; import org.tensorflow.Session; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Scope; import org.tensorflow.types.TBool; import org.tensorflow.types.TFloat32; @@ -38,10 +39,10 @@ public class ZerosTest { public void createIntZeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TInt32.class); - try (TInt32 result = (TInt32)sess.runner().fetch(op).run().get(0)) { + try (TInt32 result = (TInt32) sess.runner().fetch(op).run().get(0)) { result.scalars().forEach(s -> assertEquals(0, s.getInt())); } } @@ -51,10 +52,10 @@ public void createIntZeros() { public void createFloatZeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TFloat32.class); - try (TFloat32 result = (TFloat32)sess.runner().fetch(op.asOutput()).run().get(0)) { + try (TFloat32 result = (TFloat32) sess.runner().fetch(op.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(0.0f, s.getFloat(), 0)); } } @@ -64,10 +65,10 @@ public void createFloatZeros() { public void createDoubleZeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TFloat64.class); - try (TFloat64 result = (TFloat64)sess.runner().fetch(op.asOutput()).run().get(0)) { + try (TFloat64 result = (TFloat64) sess.runner().fetch(op.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(0.0f, s.getDouble(), 0)); } } @@ -77,10 +78,10 @@ public void createDoubleZeros() { public void createLongZeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TInt64.class); - try (TInt64 result = (TInt64)sess.runner().fetch(op.asOutput()).run().get(0)) { + try (TInt64 result = (TInt64) sess.runner().fetch(op.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(0L, s.getLong())); } } @@ -90,23 +91,23 @@ public void createLongZeros() { public void createBooleanZeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TBool.class); - try (TBool result = (TBool)sess.runner().fetch(op.asOutput()).run().get(0)) { + try (TBool result = (TBool) sess.runner().fetch(op.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertFalse(s.getBoolean())); } - } + } } @Test public void createUint8Zeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TUint8.class); - try (TUint8 result = (TUint8)sess.runner().fetch(op.asOutput()).run().get(0)) { + try (TUint8 result = (TUint8) sess.runner().fetch(op.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertEquals(0, s.getByte())); } } @@ -116,10 +117,10 @@ public void createUint8Zeros() { public void createStringZeros() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; Zeros op = Zeros.create(scope, Constant.vectorOf(scope, shape), TString.class); - try (TString result = (TString)sess.runner().fetch(op.asOutput()).run().get(0)) { + try (TString result = (TString) sess.runner().fetch(op.asOutput()).run().get(0)) { result.scalars().forEach(s -> assertTrue(s.getObject().isEmpty())); } } @@ -129,10 +130,12 @@ public void createStringZeros() { public void operationsComposingZerosAreCorrectlyNamed() { try (Graph g = new Graph(); Session sess = new Session(g)) { - Scope scope = new Scope(g); + Scope scope = new OpScope(g); long[] shape = {2, 2}; - Zeros zeros = Zeros.create(scope.withSubScope("test"), Constant.vectorOf(scope, shape), TFloat32.class); - List results = sess.runner().addTarget("test/Zeros/Zero").addTarget("test/Zeros/Fill").run(); + Zeros zeros = + Zeros.create(scope.withSubScope("test"), Constant.vectorOf(scope, shape), TFloat32.class); + List results = + sess.runner().addTarget("test/Zeros/Zero").addTarget("test/Zeros/Fill").run(); } } } diff --git a/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/Names.java b/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/Names.java index 4176f517022..7252d258814 100644 --- a/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/Names.java +++ b/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/Names.java @@ -29,6 +29,9 @@ public class Names { public static final ClassName Operator = ClassName.get(OpPackage + ".annotation", "Operator"); public static final ClassName Endpoint = ClassName.get(OpPackage + ".annotation", "Endpoint"); + public static final ClassName OpMetadata = ClassName.get(OpPackage + ".annotation", "OpMetadata"); + public static final ClassName OpInputsMetadata = + ClassName.get(OpPackage + ".annotation", "OpInputsMetadata"); public static final ClassName TType = ClassName.get(TypesPackage + ".family", "TType"); public static final ClassName TString = ClassName.get(TypesPackage, "TString"); @@ -71,6 +74,7 @@ public class Names { ClassName.get(TensorflowPackage, "ConcreteFunction"); public static final ClassName Scope = ClassName.get(OpPackage, "Scope"); + public static final ClassName OpScope = ClassName.get(OpPackage, "OpScope"); public static final TypeName DeviceSpec = ClassName.get(TensorflowPackage, "DeviceSpec"); public static final ClassName Ops = ClassName.get(OpPackage, "Ops"); diff --git a/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/op/generator/ClassGenerator.java b/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/op/generator/ClassGenerator.java index 14d3f31a977..ff964525006 100644 --- a/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/op/generator/ClassGenerator.java +++ b/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/op/generator/ClassGenerator.java @@ -34,6 +34,7 @@ import com.squareup.javapoet.WildcardTypeName; import java.util.ArrayList; import java.util.Arrays; +import java.util.Collection; import java.util.HashMap; import java.util.HashSet; import java.util.Iterator; @@ -70,6 +71,7 @@ enum RenderMode { } private static final String OP_NAME_FIELD = "OP_NAME"; + private static final String INPUTS_CLASS_NAME = "Inputs"; /** The in-progress class builder for the top level op class. */ private final TypeSpec.Builder builder; @@ -214,12 +216,29 @@ private String fullClassName() { return fullPackage + "." + className; } + private ClassName className() { + return ClassName.get(fullPackage, className); + } + + private ClassName inputsClassName() { + return ClassName.get(fullPackage, className, INPUTS_CLASS_NAME); + } + + private TypeName maybeParameterize( + ClassName baseType, Collection parameters) { + if (parameters.isEmpty()) { + return baseType; + } else { + return ParameterizedTypeName.get(baseType, parameters.toArray(new TypeName[0])); + } + } + /** Build the class. */ void buildClass() { builder.addModifiers(Modifier.PUBLIC); if (!isStateSelector) { builder.addModifiers(Modifier.FINAL); - builder.superclass(Names.RawOp); + addInputsMetadataAnnotation(); } if (isStateSubclass) { @@ -341,6 +360,7 @@ void buildClass() { buildConstructor(); buildInputsClass(); + builder.superclass(Names.RawOp); } } @@ -866,7 +886,7 @@ private void buildInterfaceImpl() { /** Add a constructor to get the outputs from an operation */ private void buildConstructor() { - MethodSpec.Builder ctor = MethodSpec.constructorBuilder().addModifiers(Modifier.PRIVATE); + MethodSpec.Builder ctor = MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC); ctor.addParameter(Names.Operation, "operation"); @@ -881,7 +901,7 @@ private void buildConstructor() { } } CodeBlock.Builder body = CodeBlock.builder(); - body.addStatement("super(operation)"); + body.addStatement("super(operation, $L)", OP_NAME_FIELD); if (op.getOutputArgCount() > 0) { body.addStatement("int outputIdx = 0"); @@ -919,9 +939,9 @@ private void buildConstructor() { builder.addMethod(ctor.build()); } - private void buildInputsClass() { + private Set buildInputsClass() { TypeSpec.Builder inputsBuilder = - TypeSpec.classBuilder("Inputs").addModifiers(Modifier.PUBLIC, Modifier.STATIC); + TypeSpec.classBuilder(INPUTS_CLASS_NAME).addModifiers(Modifier.PUBLIC, Modifier.STATIC); MethodSpec.Builder ctor = MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC); ctor.addParameter(Names.GraphOperation, "op"); @@ -1008,7 +1028,7 @@ private void buildInputsClass() { } } - TypeName outputClass = ClassName.get(fullPackage, className); + TypeName outputClass = className(); if (!this.builder.typeVariables.isEmpty()) { outputClass = ParameterizedTypeName.get( @@ -1029,6 +1049,25 @@ private void buildInputsClass() { inputsBuilder.addMethod(ctor.build()); inputsBuilder.addTypeVariables(typeVars); + addInputsMetadataAnnotation(inputsBuilder); this.builder.addType(inputsBuilder.build()); + return typeVars; + } + + /** Adds the GeneratedOpMetadata annotation to the op class. */ + private void addInputsMetadataAnnotation() { + builder.addAnnotation( + AnnotationSpec.builder(Names.OpMetadata) + .addMember("opType", "$L", className + ".OP_NAME") + .addMember("inputsClass", "$T.class", inputsClassName()) + .build()); + } + + /** Adds the GeneratedOpInputsMetadata annotation to the op input class. */ + private void addInputsMetadataAnnotation(TypeSpec.Builder inputsBuilder) { + inputsBuilder.addAnnotation( + AnnotationSpec.builder(Names.OpInputsMetadata) + .addMember("outputsClass", "$T.class", className()) + .build()); } } diff --git a/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/processor/operator/OperatorProcessor.java b/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/processor/operator/OperatorProcessor.java index 70c7bb0a7de..99277e8fe24 100644 --- a/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/processor/operator/OperatorProcessor.java +++ b/tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/processor/operator/OperatorProcessor.java @@ -155,6 +155,7 @@ public Set getSupportedAnnotationTypes() { } private static class OpsSpec { + private static final Comparator PARAMETER_SPEC_COMPARATOR = (o1, o2) -> { if (o1.parameters.size() > o2.parameters.size()) { @@ -495,7 +496,6 @@ private static TypeSpec buildTopClass(OpsSpec spec) { MethodSpec.Builder ctorBuilder = MethodSpec.constructorBuilder() .addParameter(Names.Scope, "scope") - .addModifiers(Modifier.PRIVATE) .addStatement("this.scope = scope", Names.Scope); TypeSpec.Builder opsBuilder = @@ -675,7 +675,7 @@ private static TypeSpec buildTopClass(OpsSpec spec) { .addModifiers(Modifier.PUBLIC, Modifier.STATIC) .addParameter(Names.ExecutionEnvironment, "env") .returns(Names.Ops) - .addStatement("return new Ops(env.baseScope())", Names.Scope) + .addStatement("return new Ops(env.baseScope())") .addJavadoc( "Creates an API for building operations in the provided execution environment\n") .build()); diff --git a/tensorflow-framework/src/main/java/org/tensorflow/framework/op/FrameworkOps.java b/tensorflow-framework/src/main/java/org/tensorflow/framework/op/FrameworkOps.java index 7b6322d0f0d..6d75d205133 100644 --- a/tensorflow-framework/src/main/java/org/tensorflow/framework/op/FrameworkOps.java +++ b/tensorflow-framework/src/main/java/org/tensorflow/framework/op/FrameworkOps.java @@ -18,6 +18,7 @@ import org.tensorflow.EagerSession; import org.tensorflow.ExecutionEnvironment; import org.tensorflow.op.Op; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Ops; import org.tensorflow.op.Scope; @@ -71,7 +72,7 @@ private FrameworkOps(Ops core) { * @return the FrameworkOps */ public static FrameworkOps create(ExecutionEnvironment env) { - return new FrameworkOps(new Scope(env)); + return new FrameworkOps(new OpScope(env)); } /** @@ -83,7 +84,7 @@ public static FrameworkOps create(ExecutionEnvironment env) { * @return the FrameworkOps */ public static FrameworkOps create() { - return new FrameworkOps(new Scope(EagerSession.getDefault())); + return new FrameworkOps(new OpScope(EagerSession.getDefault())); } /** diff --git a/tensorflow-framework/src/main/java/org/tensorflow/framework/optimizers/Optimizer.java b/tensorflow-framework/src/main/java/org/tensorflow/framework/optimizers/Optimizer.java index b1366146836..a12e46f82e5 100644 --- a/tensorflow-framework/src/main/java/org/tensorflow/framework/optimizers/Optimizer.java +++ b/tensorflow-framework/src/main/java/org/tensorflow/framework/optimizers/Optimizer.java @@ -26,6 +26,7 @@ import org.tensorflow.Operation; import org.tensorflow.Output; import org.tensorflow.op.Op; +import org.tensorflow.op.OpScope; import org.tensorflow.op.Ops; import org.tensorflow.op.Scope; import org.tensorflow.op.core.NoOp; @@ -280,7 +281,7 @@ private Op applyDense(GradAndVar gradVarPair) { * @return A NoOp with a control dependency on each update operation. */ protected Op finish(List updateOperations, String name) { - Scope scope = new Scope(graph); + Scope scope = new OpScope(graph); scope = scope.withName(name); scope = scope.withControlDependencies(updateOperations); return NoOp.create(scope);