File tree Expand file tree Collapse file tree
tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/generator/op Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -105,7 +105,7 @@ enum RenderMode {
105105
106106 /**
107107 * The generated options class, or null if it doesn't have one or {@link #buildOptionsClass()} has
108- * not been ran .
108+ * not been run .
109109 */
110110 private TypeSpec optionsClass = null ;
111111
@@ -748,6 +748,10 @@ private void buildSecondaryFactory(
748748 body .add ("$T.class" , defaultTypes .get (attr ));
749749 } else {
750750 factoryBuilder .addParameter (param );
751+ // Checking if the parameter being added is the variadic options or not
752+ if (param .name .equals ("options" )) {
753+ factoryBuilder .varargs ();
754+ }
751755 factoryBuilder .addJavadoc ("\n @param $L $L" , param .name , paramTags .get (param .name ));
752756 typeVars .addAll (new ResolvedType (param .type ).findGenerics ());
753757 body .add ("$L" , param .name );
You can’t perform that action at this time.
0 commit comments