Package org.nd4j.autodiff.samediff.optimize.optimizations

Class ShapeFunctionOptimizations.FuseChainedReshapes

java.lang.Object

org.nd4j.autodiff.samediff.optimize.optimizations.ShapeFunctionOptimizations.FuseChainedReshapes

All Implemented Interfaces:

Optimizer

Enclosing class:

ShapeFunctionOptimizations

public static class ShapeFunctionOptimizations.FuseChainedReshapes
extends Object
implements Optimizer

Fuse [reshape1 -> reshape2 -> ... -> reshapeN] into a single reshape op, as long as the intermediate reshape ops aren't needed for another op

Constructor Summary

Constructors

Constructor	Description
FuseChainedReshapes()

Method Summary

Modifier and Type	Method	Description
boolean	checkAndApply(SameDiff sd, OptimizationHelper helper, SameDiffOp op, ArrayHolder constantArrays, ArrayHolder variablesArrays)

Methods inherited from class java.lang.Object

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

Constructor Detail

FuseChainedReshapes

public FuseChainedReshapes()

Method Detail

checkAndApply

public boolean checkAndApply(SameDiff sd,
                             OptimizationHelper helper,
                             SameDiffOp op,
                             ArrayHolder constantArrays,
                             ArrayHolder variablesArrays)

Specified by:

checkAndApply in interface Optimizer

Parameters:

sd - Current SameDiff instance to optimize

helper - Helper class for optimization

op - Operation to check for optimization

constantArrays - Array holder for constant arrays

variablesArrays - Array holder for variable arrays

Returns:

True if the optimization was applied