Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small ND4S fix (#8143) 

Signed-off-by: Alex Black <blacka101@gmail.com>
master
Alex Black 2019-08-24 14:01:56 +10:00 committed by GitHub
parent f03b0ee78f
commit 95b2686ce5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nd4s/src/test/scala/org/nd4s/samediff/SameDiffTest.scala
View File

@ -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.reduce3.ManhattanDistance
import org.nd4j.linalg.api.ops.impl.shape.tensorops.TensorArray 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.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.impl.transforms.custom._
import org.nd4j.linalg.api.ops.random.impl.BernoulliDistribution import org.nd4j.linalg.api.ops.random.impl.BernoulliDistribution
import org.nd4j.linalg.api.shape.LongShapeDescriptor import org.nd4j.linalg.api.shape.LongShapeDescriptor