Class ShapeFunctionOptimizations
- java.lang.Object
-
- org.nd4j.autodiff.samediff.optimize.optimizations.BaseOptimizerSet
-
- org.nd4j.autodiff.samediff.optimize.optimizations.ShapeFunctionOptimizations
-
- All Implemented Interfaces:
OptimizerSet
public class ShapeFunctionOptimizations extends BaseOptimizerSet
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
ShapeFunctionOptimizations.FuseChainedConcatOps
Fuse [concat(concat(concat(x,y,dim=D), z, dim=D), a, dim=D)] into a single concat op, concat(x,y,z,a, dim=D) As long as the intermediate outputs aren't needed elsewherestatic class
ShapeFunctionOptimizations.FuseChainedPermutes
Fuse [permute1 -> permute2 -> ...static class
ShapeFunctionOptimizations.FuseChainedReshapes
Fuse [reshape1 -> reshape2 -> ...
-
Constructor Summary
Constructors Constructor Description ShapeFunctionOptimizations()
-