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 and Description
GraphOptimizer()

Method Summary

Modifier and Type	Method and Description
static List<OptimizerSet>	defaultOptimizations()
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)
static SameDiff	optimize(SameDiff graph, String... requiredOutputs)

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)