From 96f4dab6622a5aa6bf66e18c3017094c0d88e5cd Mon Sep 17 00:00:00 2001
From: Ryan Nett <rnett@calpoly.edu>
Date: Mon, 25 Jan 2021 18:40:56 -0800
Subject: [PATCH] Update manual ops for new codegen

Signed-off-by: Ryan Nett <rnett@calpoly.edu>
---
 .../src/gen/annotations/org/tensorflow/op/Ops.java            | 4 ++--
 .../src/main/java/org/tensorflow/op/core/Ones.java            | 2 +-
 .../src/main/java/org/tensorflow/op/core/Zeros.java           | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

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 810acfcbc25..84736ada6a5 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
@@ -3433,7 +3433,7 @@ public <U extends TType> OneHot<U> oneHot(Operand<? extends TNumber> indices,
    * @return a constant tensor initialized with ones
    * @throws IllegalArgumentException if the tensor type or shape cannot be initialized with ones.
    */
-  public <T extends TType, U extends TNumber> Ones<T> ones(Operand<U> dims, Class<T> type) {
+  public <T extends TType> Ones<T> ones(Operand<? extends TNumber> dims, Class<T> type) {
     return Ones.create(scope, dims, type);
   }
 
@@ -7674,7 +7674,7 @@ public <T extends TType> XlaSpmdShardToFullShape<T> xlaSpmdShardToFullShape(Oper
    * @return a constant tensor initialized with zeros
    * @throws IllegalArgumentException if the tensor type or shape cannot be initialized with zeros.
    */
-  public <T extends TType, U extends TNumber> Zeros<T> zeros(Operand<U> dims, Class<T> type) {
+  public <T extends TType> Zeros<T> zeros(Operand<? extends TNumber> dims, Class<T> type) {
     return Zeros.create(scope, dims, type);
   }
 
diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Ones.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Ones.java
index a57c05f1940..eff062c3de7 100644
--- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Ones.java
+++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Ones.java
@@ -49,7 +49,7 @@ public final class Ones<T extends TType> implements Op, Operand<T> {
    * @throws IllegalArgumentException if the tensor type or shape cannot be initialized with ones.
    */
   @Endpoint
-  public static <T extends TType, U extends TNumber> Ones<T> create(Scope scope, Operand<U> dims, Class<T> type) {
+  public static <T extends TType> Ones<T> create(Scope scope, Operand<? extends TNumber> dims, Class<T> type) {
     Scope onesScope = scope.withSubScope("Ones");
     if (type == TString.class) {
       throw new IllegalArgumentException("Can't create Ones of String DataType");
diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Zeros.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Zeros.java
index a5b5bb137c2..5fdfc948381 100644
--- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Zeros.java
+++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Zeros.java
@@ -50,7 +50,7 @@ public final class Zeros<T extends TType> implements Op, Operand<T> {
    */
   @Endpoint
   @SuppressWarnings("unchecked")
-  public static <T extends TType, U extends TNumber> Zeros<T> create(Scope scope, Operand<U> dims, Class<T> type) {
+  public static <T extends TType> Zeros<T> create(Scope scope, Operand<? extends TNumber> dims, Class<T> type) {
     Scope zerosScope = scope.withSubScope("Zeros");
     Operand<T> zero;
     if (type == TString.class) {