Skip to content

Commit 78a993c

Browse files
committedMar 4, 2025
Tweak order of arena parameter in OnnxRuntime::execute
1 parent 0380011 commit 78a993c

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed
 

‎cr-examples/onnx/src/main/java/oracle/code/onnx/OnnxRuntime.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -132,10 +132,10 @@ protected CachedModel computeValue(Class<?> type) {
132132
private static final CachedModelClassValue MODEL_CACHE = new CachedModelClassValue();
133133

134134
public static <T> Tensor<T> execute(MethodHandles.Lookup l, OnnxFunction<Tensor<T>> codeLambda) {
135-
return execute(l, codeLambda, Arena.ofAuto());
135+
return execute(l, Arena.ofAuto(), codeLambda);
136136
}
137137

138-
public static <T> Tensor<T> execute(MethodHandles.Lookup l, OnnxFunction<Tensor<T>> codeLambda, Arena sessionArena) {
138+
public static <T> Tensor<T> execute(MethodHandles.Lookup l, Arena sessionArena, OnnxFunction<Tensor<T>> codeLambda) {
139139
var q = Op.ofQuotable(codeLambda).orElseThrow();
140140

141141
var model = MODEL_CACHE.computeIfAbsent(codeLambda.getClass(), l, q);

‎cr-examples/onnx/src/test/java/oracle/code/onnx/mnist/MNISTDemo.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,8 @@ public static float[] classify(float[] imageData) {
110110
try (Arena arena = Arena.ofConfined()) {
111111
var imageTensor = Tensor.ofShape(new long[]{1, 1, IMAGE_SIZE, IMAGE_SIZE}, imageData);
112112

113-
var predictionTensor = OnnxRuntime.execute(MethodHandles.lookup(),
114-
() -> cnn(imageTensor), arena);
113+
var predictionTensor = OnnxRuntime.execute(MethodHandles.lookup(), arena,
114+
() -> cnn(imageTensor));
115115

116116
return predictionTensor.data().toArray(ValueLayout.JAVA_FLOAT);
117117
}

0 commit comments

Comments
 (0)