org.nd4j.autodiff.samediff.optimize.optimizations

Class OptimizationUtils

java.lang.Object

org.nd4j.autodiff.samediff.optimize.optimizations.OptimizationUtils

public class OptimizationUtils
extends Object

Method Summary

Modifier and Type	Method and Description
static void	removeOp(@NonNull SameDiff sd, @NonNull String opToRemove)
static void	removeVariable(@NonNull SameDiff sd, @NonNull String varToRemove)
static void	replaceOpInputsWith(SameDiff sd, @NonNull String replaceInput, @NonNull String newInput)

Methods inherited from class java.lang.Object

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

Method Detail

replaceOpInputsWith

public static void replaceOpInputsWith(SameDiff sd,
                                       @NonNull
                                       @NonNull String replaceInput,
                                       @NonNull
                                       @NonNull String newInput)

removeOp

public static void removeOp(@NonNull
                            @NonNull SameDiff sd,
                            @NonNull
                            @NonNull String opToRemove)

removeVariable

public static void removeVariable(@NonNull
                                  @NonNull SameDiff sd,
                                  @NonNull
                                  @NonNull String varToRemove)