org.nd4j.autodiff.samediff.optimize.optimizations

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 and 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 elsewhere
static class	ShapeFunctionOptimizations.FuseChainedPermutes Fuse [permute1 -> permute2 -> ...
static class	ShapeFunctionOptimizations.FuseChainedReshapes Fuse [reshape1 -> reshape2 -> ...

Constructor Summary

Constructors

Constructor and Description
ShapeFunctionOptimizations()

Method Summary

Methods inherited from class org.nd4j.autodiff.samediff.optimize.optimizations.BaseOptimizerSet

getOptimizers

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ShapeFunctionOptimizations

public ShapeFunctionOptimizations()