From 95b2686ce5f43174fa864c9011f426b70c7b7348 Mon Sep 17 00:00:00 2001 From: Alex Black Date: Sat, 24 Aug 2019 14:01:56 +1000 Subject: [PATCH] Small ND4S fix (#8143) Signed-off-by: Alex Black --- nd4s/src/test/scala/org/nd4s/samediff/SameDiffTest.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nd4s/src/test/scala/org/nd4s/samediff/SameDiffTest.scala b/nd4s/src/test/scala/org/nd4s/samediff/SameDiffTest.scala index a99b78214..c9ee41bfb 100644 --- a/nd4s/src/test/scala/org/nd4s/samediff/SameDiffTest.scala +++ b/nd4s/src/test/scala/org/nd4s/samediff/SameDiffTest.scala @@ -20,7 +20,7 @@ import org.nd4j.linalg.api.ops.impl.layers.convolution.config.{ Conv2DConfig, Lo import org.nd4j.linalg.api.ops.impl.reduce3.ManhattanDistance import org.nd4j.linalg.api.ops.impl.shape.tensorops.TensorArray import org.nd4j.linalg.api.ops.impl.transforms.any.IsMax -import org.nd4j.linalg.api.ops.impl.transforms.comparison.{ OldMax, OldMin } +import org.nd4j.linalg.api.ops.impl.transforms.custom.{ Max, Min } import org.nd4j.linalg.api.ops.impl.transforms.custom._ import org.nd4j.linalg.api.ops.random.impl.BernoulliDistribution import org.nd4j.linalg.api.shape.LongShapeDescriptor