Data/Model: Fix wrong format type

This patch fixes wrong format type when get tensors. Signed-off-by: Yelin Jeong <[email protected]>
nnstreamer · Sep 24, 2024 · 573d5fb · 573d5fb
1 parent c125b4d
commit 573d5fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/Model.kt b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/Model.kt
@@ -169,7 +169,7 @@ data class Model(
         val outFormat = outputInfo["format"]?.let { getFormat(it) } ?: "static"
 
         val inTensors = getTensors(inputInfo["type"], inFormat)
-        val outTensors = getTensors(outputInfo["type"], inFormat)
+        val outTensors = getTensors(outputInfo["type"], outFormat)
 
         val inTypes = inputInfo["type"]?.let { getType(it) } ?: ""
         val outTypes = outputInfo["type"]?.let { getType(it) } ?: ""