diff options
author | Jon Bratseth <bratseth@yahoo-inc.com> | 2017-12-13 16:08:25 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@yahoo-inc.com> | 2017-12-13 16:08:25 +0100 |
commit | 94cf83e79abc75f4498d18ec3799ed0171e5b6df (patch) | |
tree | f39a96be167db90a126ae38db551d933cca1e679 /searchlib | |
parent | 91f1c2486b3f60423f43b845d959badb54ffd3fd (diff) |
Remove warning
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/Mnist_SoftmaxTestCase.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/Mnist_SoftmaxTestCase.java b/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/Mnist_SoftmaxTestCase.java index dab42801d70..863479f3531 100644 --- a/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/Mnist_SoftmaxTestCase.java +++ b/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/tensorflow/Mnist_SoftmaxTestCase.java @@ -85,7 +85,7 @@ public class Mnist_SoftmaxTestCase { private Tensor tensorFlowExecute(String modelDir, String operationName) { SavedModelBundle model = SavedModelBundle.load(modelDir, "serve"); Session.Runner runner = model.session().runner(); - org.tensorflow.Tensor placeholder = org.tensorflow.Tensor.create(new long[]{ 1, 784 }, FloatBuffer.allocate(784)); + org.tensorflow.Tensor<?> placeholder = org.tensorflow.Tensor.create(new long[]{ 1, 784 }, FloatBuffer.allocate(784)); runner.feed("Placeholder", placeholder); List<org.tensorflow.Tensor<?>> results = runner.fetch(operationName).run(); assertEquals(1, results.size()); |