Have a question? Connect with the community at the TensorFlow Forum Visit Forum

RewriterConfig

public final class RewriterConfig

 Graph rewriting is experimental and subject to change, not covered by any
 API stability guarantees.
 
Protobuf type tensorflow.RewriterConfig

Nested Classes

class RewriterConfig.Builder
 Graph rewriting is experimental and subject to change, not covered by any
 API stability guarantees. 
enum RewriterConfig.CpuLayout
 Enum for layout conversion between NCHW and NHWC on CPU. 
class RewriterConfig.CustomGraphOptimizer
 Message to describe custom graph optimizer and its parameters
 
Protobuf type tensorflow.RewriterConfig.CustomGraphOptimizer
interface RewriterConfig.CustomGraphOptimizerOrBuilder
enum RewriterConfig.MemOptType Protobuf enum tensorflow.RewriterConfig.MemOptType
enum RewriterConfig.NumIterationsType
 Enum controlling the number of times to run optimizers. 
enum RewriterConfig.Toggle Protobuf enum tensorflow.RewriterConfig.Toggle

Constants

int ARITHMETIC_OPTIMIZATION_FIELD_NUMBER
int AUTO_MIXED_PRECISION_FIELD_NUMBER
int AUTO_MIXED_PRECISION_MKL_FIELD_NUMBER
int AUTO_PARALLEL_FIELD_NUMBER
int COMMON_SUBGRAPH_ELIMINATION_FIELD_NUMBER
int CONSTANT_FOLDING_FIELD_NUMBER
int CPU_LAYOUT_CONVERSION_FIELD_NUMBER
int CUSTOM_OPTIMIZERS_FIELD_NUMBER
int DEBUG_STRIPPER_FIELD_NUMBER
int DEPENDENCY_OPTIMIZATION_FIELD_NUMBER
int DISABLE_META_OPTIMIZER_FIELD_NUMBER
int DISABLE_MODEL_PRUNING_FIELD_NUMBER
int EXPERIMENTAL_DISABLE_COMPRESSED_TENSOR_OPTIMIZATION_FIELD_NUMBER
int FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER
int FUNCTION_OPTIMIZATION_FIELD_NUMBER
int IMPLEMENTATION_SELECTOR_FIELD_NUMBER
int INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER
int LAYOUT_OPTIMIZER_FIELD_NUMBER
int LOOP_OPTIMIZATION_FIELD_NUMBER
int MEMORY_OPTIMIZATION_FIELD_NUMBER
int MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER
int META_OPTIMIZER_ITERATIONS_FIELD_NUMBER
int META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER
int MIN_GRAPH_NODES_FIELD_NUMBER
int OPTIMIZERS_FIELD_NUMBER
int PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER
int POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER
int REMAPPING_FIELD_NUMBER
int SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER
int SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER
int SHAPE_OPTIMIZATION_FIELD_NUMBER

Public Methods

boolean
equals (Object obj)
RewriterConfig.Toggle
getArithmeticOptimization ()
 Arithmetic optimizations (default is ON)
 e.g.
int
getArithmeticOptimizationValue ()
 Arithmetic optimizations (default is ON)
 e.g.
RewriterConfig.Toggle
getAutoMixedPrecision ()
 Optimize data types for CUDA (default is OFF).
RewriterConfig.Toggle
getAutoMixedPrecisionMkl ()
 Optimize data types for MKL (default is OFF).
int
getAutoMixedPrecisionMklValue ()
 Optimize data types for MKL (default is OFF).
int
getAutoMixedPrecisionValue ()
 Optimize data types for CUDA (default is OFF).
AutoParallelOptions
getAutoParallel ()
 Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
AutoParallelOptionsOrBuilder
getAutoParallelOrBuilder ()
 Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
RewriterConfig.Toggle
getCommonSubgraphElimination ()
 Common subgraph elimination (default is ON)
 e.g.
int
getCommonSubgraphEliminationValue ()
 Common subgraph elimination (default is ON)
 e.g.
RewriterConfig.Toggle
getConstantFolding ()
 Fold constants (default is ON)
 Statically infer the value of tensors when possible, and materialize the
 result using constants.
int
getConstantFoldingValue ()
 Fold constants (default is ON)
 Statically infer the value of tensors when possible, and materialize the
 result using constants.
RewriterConfig.CpuLayout
getCpuLayoutConversion ()
 CPU Conversion settings between NHCW and NCHW.
int
getCpuLayoutConversionValue ()
 CPU Conversion settings between NHCW and NCHW.
RewriterConfig.CustomGraphOptimizer
getCustomOptimizers (int index)
 list of CustomGraphOptimizers to apply.
int
getCustomOptimizersCount ()
 list of CustomGraphOptimizers to apply.
List< RewriterConfig.CustomGraphOptimizer >
getCustomOptimizersList ()
 list of CustomGraphOptimizers to apply.
RewriterConfig.CustomGraphOptimizerOrBuilder
getCustomOptimizersOrBuilder (int index)
 list of CustomGraphOptimizers to apply.
List<? extends RewriterConfig.CustomGraphOptimizerOrBuilder >
getCustomOptimizersOrBuilderList ()
 list of CustomGraphOptimizers to apply.
RewriterConfig.Toggle
getDebugStripper ()
 Strips debug-related nodes from the graph (off by default).
int
getDebugStripperValue ()
 Strips debug-related nodes from the graph (off by default).
static RewriterConfig
RewriterConfig
RewriterConfig.Toggle
getDependencyOptimization ()
 Control dependency optimizations (default is ON).
int
getDependencyOptimizationValue ()
 Control dependency optimizations (default is ON).
final static com.google.protobuf.Descriptors.Descriptor
boolean
getDisableMetaOptimizer ()
 Disable the entire meta optimizer (off by default).
boolean
getDisableModelPruning ()
 If true, don't remove unnecessary ops from the graph
 
bool disable_model_pruning = 2;
boolean
getExperimentalDisableCompressedTensorOptimization ()
 Disable optimizations that assume compressed tensors.
boolean
getFailOnOptimizerErrors ()
 If true, any optimization pass failing will cause the MetaOptimizer to
 stop with an error.
RewriterConfig.Toggle
getFunctionOptimization ()
 Function optimizations (default is ON).
int
getFunctionOptimizationValue ()
 Function optimizations (default is ON).
RewriterConfig.Toggle
getImplementationSelector ()
 Enable the swap of kernel implementations based on the device placement
 (default is ON).
int
getImplementationSelectorValue ()
 Enable the swap of kernel implementations based on the device placement
 (default is ON).
VerifierConfig
getInterOptimizerVerifierConfig ()
 VerifierConfig specifying the verifiers to be run after every optimizer.
VerifierConfigOrBuilder
getInterOptimizerVerifierConfigOrBuilder ()
 VerifierConfig specifying the verifiers to be run after every optimizer.
RewriterConfig.Toggle
getLayoutOptimizer ()
 Optimize tensor layouts (default is ON)
 e.g.
int
getLayoutOptimizerValue ()
 Optimize tensor layouts (default is ON)
 e.g.
RewriterConfig.Toggle
getLoopOptimization ()
 Loop optimizations (default is ON).
int
getLoopOptimizationValue ()
 Loop optimizations (default is ON).
RewriterConfig.MemOptType
getMemoryOptimization ()
 Configures memory optimization passes through the meta-optimizer.
int
getMemoryOptimizationValue ()
 Configures memory optimization passes through the meta-optimizer.
String
getMemoryOptimizerTargetNodeNameScope ()
 A node name scope for node names which are valid outputs of recomputations.
com.google.protobuf.ByteString
getMemoryOptimizerTargetNodeNameScopeBytes ()
 A node name scope for node names which are valid outputs of recomputations.
RewriterConfig.NumIterationsType
getMetaOptimizerIterations ()
 Controls how many times we run the optimizers in meta optimizer (default
 is once).
int
getMetaOptimizerIterationsValue ()
 Controls how many times we run the optimizers in meta optimizer (default
 is once).
long
getMetaOptimizerTimeoutMs ()
 Maximum number of milliseconds to spend optimizing a single graph before
 timing out.
int
getMinGraphNodes ()
 The minimum number of nodes in a graph to optimizer.
String
getOptimizers (int index)
 If non-empty, will use this as an alternative way to specify a list of
 optimizations to turn on and the order of the optimizations (replacing the
 meta-optimizer).
com.google.protobuf.ByteString
getOptimizersBytes (int index)
 If non-empty, will use this as an alternative way to specify a list of
 optimizations to turn on and the order of the optimizations (replacing the
 meta-optimizer).
int
getOptimizersCount ()
 If non-empty, will use this as an alternative way to specify a list of
 optimizations to turn on and the order of the optimizations (replacing the
 meta-optimizer).
com.google.protobuf.ProtocolStringList
getOptimizersList ()
 If non-empty, will use this as an alternative way to specify a list of
 optimizations to turn on and the order of the optimizations (replacing the
 meta-optimizer).
RewriterConfig.Toggle
getPinToHostOptimization ()
 Force small ops onto the CPU (default is OFF).
int
getPinToHostOptimizationValue ()
 Force small ops onto the CPU (default is OFF).
VerifierConfig
getPostOptimizationVerifierConfig ()
 VerifierConfig specifying the verifiers to be run at the end, after all
 optimizers have run.
VerifierConfigOrBuilder
getPostOptimizationVerifierConfigOrBuilder ()
 VerifierConfig specifying the verifiers to be run at the end, after all
 optimizers have run.
RewriterConfig.Toggle
getRemapping ()
 Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
int
getRemappingValue ()
 Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
RewriterConfig.Toggle
getScopedAllocatorOptimization ()
 Try to allocate some independent Op outputs contiguously in order to
 merge or eliminate downstream Ops (off by default).
int
getScopedAllocatorOptimizationValue ()
 Try to allocate some independent Op outputs contiguously in order to
 merge or eliminate downstream Ops (off by default).
ScopedAllocatorOptions
getScopedAllocatorOpts ()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
ScopedAllocatorOptionsOrBuilder
getScopedAllocatorOptsOrBuilder ()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
int
RewriterConfig.Toggle
getShapeOptimization ()
 Shape optimizations (default is ON)
 Simplify computations made on shapes.
int
getShapeOptimizationValue ()
 Shape optimizations (default is ON)
 Simplify computations made on shapes.
final com.google.protobuf.UnknownFieldSet
boolean
hasAutoParallel ()
 Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
boolean
hasInterOptimizerVerifierConfig ()
 VerifierConfig specifying the verifiers to be run after every optimizer.
boolean
hasPostOptimizationVerifierConfig ()
 VerifierConfig specifying the verifiers to be run at the end, after all
 optimizers have run.