From 300c74f25cfec1b1c9c668db973b0c90a0a5a7a0 Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Sun, 29 May 2022 18:10:12 +0200 Subject: [PATCH] refactor methods --- .../icy/tensorflow/v1/tensor/TensorBuilder.java | 2 +- src/main/java/tensorflow/v1/Tensorflow1Interface.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java b/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java index a8b9564..3332be6 100644 --- a/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java +++ b/src/main/java/org/bioimageanalysis/icy/tensorflow/v1/tensor/TensorBuilder.java @@ -31,7 +31,7 @@ public final class TensorBuilder */ public static Tensor<?> build(org.bioimageanalysis.icy.deeplearning.tensor.Tensor tensor) { - return build(tensor.getDataAsNDArray()); + return build(tensor.getData()); } /** diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java index c9ae2ab..ec43158 100644 --- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java +++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java @@ -120,7 +120,7 @@ public class Tensorflow1Interface implements DeepLearningInterface if (outputNDArrays.size() != outputTensors.size()) throw new RunModelException(outputNDArrays.size(), outputTensors.size()); for (int i = 0; i < outputNDArrays.size(); i ++) { - outputTensors.get(i).setNDArrayData(Nd4fBuilder.build(outputNDArrays.get(i))); + outputTensors.get(i).setData(Nd4fBuilder.build(outputNDArrays.get(i))); } return outputTensors; } -- GitLab