Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
[MXNET-918] [Introduce Random module / Refact code generation (#13038)…
Browse files Browse the repository at this point in the history
…][Cherry pick] (#13242)

* [MXNET-918] Introduce Random module / Refact code generation (#13038)

* refactor code gen

* remove xxxAPIMacroBase (overkill)

* CI errors / scala-style

* PR review comments

* clean up the duplicated code

* add comments
  • Loading branch information
lanking520 authored and yzhliu committed Nov 13, 2018
1 parent 1bb5b7f commit fb4cad9
Show file tree
Hide file tree
Showing 6 changed files with 454 additions and 634 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ public void runBatchInference() {
List<NDArray> nd = new ArrayList<>();
NDArray[] temp = new NDArray[batchSize];
for (int i = 0; i < batchSize; i++) temp[i] = img.copy();
NDArray batched = NDArray.concat(temp, batchSize).setdim(0).invoke().get();
NDArray batched = NDArray.concat(temp, batchSize, 0, null)[0];
nd.add(batched);
objDet.objectDetectWithNDArray(nd, 3);
}
Expand Down
Loading

0 comments on commit fb4cad9

Please sign in to comment.