From 2fcae999c75ae30656180ef5c270c7d097a2228c Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Fri, 30 Sep 2022 19:49:18 +0200
Subject: [PATCH] adapt to new model runner
---
src/main/java/tensorflow/v1/Tensorflow1Interface.java | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index bb68bb7..47300d5 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -78,7 +78,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
}
@Override
- public List<Tensor> run(List<Tensor> inputTensors, List<Tensor> outputTensors) throws RunModelException {
+ public List<Tensor<?>> run(List<Tensor<?>> inputTensors, List<Tensor<?>> outputTensors) throws RunModelException {
Session session = model.session();
Session.Runner runner = session.runner();
List<String> inputListNames = new ArrayList<String>();
@@ -117,7 +117,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
* @throws RunModelException If the number of tensors expected is not the same as the number of
* Tensors outputed by the model
*/
- public static List<Tensor> fillOutputTensors(List<org.tensorflow.Tensor<?>> outputNDArrays, List<Tensor> outputTensors) throws RunModelException{
+ public static List<Tensor<?>> fillOutputTensors(List<org.tensorflow.Tensor<?>> outputNDArrays, List<Tensor<?>> outputTensors) throws RunModelException{
if (outputNDArrays.size() != outputTensors.size())
throw new RunModelException(outputNDArrays.size(), outputTensors.size());
for (int i = 0; i < outputNDArrays.size(); i ++) {
--
GitLab