Skip to content

Commit 138a9e1

Browse files
committed
correct javadocs
1 parent 306376f commit 138a9e1

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -290,6 +290,10 @@ protected void runFromShmas(List<String> inputs, List<String> outputs) throws IO
290290
* MEthod only used in MacOS Intel and Windows systems that makes all the arrangements
291291
* to create another process, communicate the model info and tensors to the other
292292
* process and then retrieve the results of the other process
293+
* @param <T>
294+
* ImgLib2 data type of the input tensors
295+
* @param <R>
296+
* ImgLib2 data type of the output tensors, it can be the same as the input tensors' data type
293297
* @param inputTensors
294298
* tensors that are going to be run on the model
295299
* @param outputTensors
@@ -339,6 +343,8 @@ else if (task.status == TaskStatus.CRASHED)
339343
* Create the list a list of output tensors agnostic to the Deep Learning
340344
* engine that can be readable by the model-runner
341345
*
346+
* @param <T>
347+
* ImgLib2 data type of the output tensors
342348
* @param outputNDArrays
343349
* an NDList containing NDArrays (tensors)
344350
* @param outputTensors

src/main/java/io/bioimage/modelrunner/pytorch/tensor/NDArrayBuilder.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ public class NDArrayBuilder {
5555
* Creates a {@link NDArray} from a given {@link Tensor}. The {@link Tensor}
5656
* contains the data and info(dimensions, dataype) necessary to build a {@link NDArray}
5757
*
58+
* @param <T>
59+
* ImgLib2 data type of the output tensors
5860
* @param tensor
5961
* The {@link Tensor} that will be copied into a {@link NDArray}
6062
* @param manager

0 commit comments

Comments
 (0)