Package org.nd4j.autodiff.samediff.optimize

Class GraphOptimizer

java.lang.Object

org.nd4j.autodiff.samediff.optimize.GraphOptimizer

public class GraphOptimizer
extends Object

Author:

Alex Black

Constructor Summary

Constructors

Constructor	Description
GraphOptimizer()

Method Summary

Modifier and Type	Method	Description
static List<OptimizerSet>	defaultOptimizations()
static SameDiff	optimize(SameDiff graph, String... requiredOutputs)
static SameDiff	optimize(SameDiff graph, List<String> requiredOutputs)
static SameDiff	optimize(SameDiff graph, List<String> requiredOutputs, List<OptimizerSet> optimizations)
static SameDiff	optimize(SameDiff graph, List<String> requiredOutputs, List<OptimizerSet> optimizations, OptimizationDebugger debugger)

Methods inherited from class java.lang.Object

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

Constructor Detail

GraphOptimizer

public GraphOptimizer()

Method Detail

defaultOptimizations

public static List<OptimizerSet> defaultOptimizations()

optimize

public static SameDiff optimize(SameDiff graph,
                                String... requiredOutputs)

optimize

public static SameDiff optimize(SameDiff graph,
                                List<String> requiredOutputs)

optimize

public static SameDiff optimize(SameDiff graph,
                                List<String> requiredOutputs,
                                List<OptimizerSet> optimizations)

optimize

public static SameDiff optimize(SameDiff graph,
                                List<String> requiredOutputs,
                                List<OptimizerSet> optimizations,
                                OptimizationDebugger debugger)