Adding ineritDoc and @OverRide

deepjavalibrary · Apr 21, 2021 · 46c4b8b · 46c4b8b
1 parent 728725c
commit 46c4b8b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxNDArray.java b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxNDArray.java
@@ -1527,6 +1527,8 @@ public NDArray oneHot(int depth, float onValue, float offValue, DataType dataTyp
         return manager.invoke("_npx_one_hot", this, params).toType(dataType, false);
     }
 
+    /** {@inheritDoc} */
+    @Override
     public NDArray batchDot(NDArray other) {
         return manager.invoke("_npx_batch_dot", new NDArray[] {this, other}, null);
     }

diff --git a/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java b/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java
@@ -1413,6 +1413,7 @@ public NDArray oneHot(int depth, float onValue, float offValue, DataType dataTyp
         throw new UnsupportedOperationException("Not implemented");
     }
 
+    /** {@inheritDoc} */
     @Override
     public NDArray batchDot(NDArray other) {
         throw new UnsupportedOperationException("Not implemented");

diff --git a/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfNDArray.java b/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfNDArray.java
@@ -485,6 +485,7 @@ public NDArray oneHot(int depth, float onValue, float offValue, DataType dataTyp
         }
     }
 
+    /** {@inheritDoc} */
     @Override
     public NDArray batchDot(NDArray other) {
         throw new UnsupportedOperationException("Not implemented");