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

Class ShapeFunctionOptimizations.FuseChainedConcatOps

java.lang.Object

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

All Implemented Interfaces:

Optimizer

Enclosing class:

ShapeFunctionOptimizations

public static class ShapeFunctionOptimizations.FuseChainedConcatOps
extends Object
implements Optimizer

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

Constructor Summary

Constructors

Constructor	Description
FuseChainedConcatOps()

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

FuseChainedConcatOps

public FuseChainedConcatOps()

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