Package org.nd4j.autodiff.samediff.optimize

Interface Optimizer

All Known Implementing Classes:

ConstantFunctionOptimizations.FoldConstantFunctions, CuDNNFunctionOptimizations.CudnnConv2dNCHWtoNHWCConversion, IdentityFunctionOptimizations.RemoveIdentityOps, IdentityFunctionOptimizations.RemoveIdentityPermute, ShapeFunctionOptimizations.FuseChainedConcatOps, ShapeFunctionOptimizations.FuseChainedPermutes, ShapeFunctionOptimizations.FuseChainedReshapes, UnusedFunctionOptimizations.RemoveUnusedConstants

public interface Optimizer

Author:

Alex Black

Method Summary

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

Method Detail

checkAndApply

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

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