Skip to content
Snippets Groups Projects
Commit 2fcae999 authored by carlosuc3m's avatar carlosuc3m
Browse files

adapt to new model runner

parent 422a98d7
No related branches found
No related tags found
No related merge requests found
......@@ -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 ++) {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment