Package org.tensorflow.framework
Class RewriterConfig
java.lang.Object
com.google.protobuf.AbstractMessageLite
com.google.protobuf.AbstractMessage
com.google.protobuf.GeneratedMessageV3
org.tensorflow.framework.RewriterConfig
- All Implemented Interfaces:
com.google.protobuf.Message,com.google.protobuf.MessageLite,com.google.protobuf.MessageLiteOrBuilder,com.google.protobuf.MessageOrBuilder,java.io.Serializable,RewriterConfigOrBuilder
public final class RewriterConfig extends com.google.protobuf.GeneratedMessageV3 implements RewriterConfigOrBuilder
Graph rewriting is experimental and subject to change, not covered by any API stability guarantees.Protobuf type
tensorflow.RewriterConfig- See Also:
- Serialized Form
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classRewriterConfig.BuilderGraph rewriting is experimental and subject to change, not covered by any API stability guarantees.static classRewriterConfig.CustomGraphOptimizerMessage to describe custom graph optimizer and its parametersstatic interfaceRewriterConfig.CustomGraphOptimizerOrBuilderstatic classRewriterConfig.MemOptTypeProtobuf enumtensorflow.RewriterConfig.MemOptTypestatic classRewriterConfig.NumIterationsTypeEnum controlling the number of times to run optimizers.static classRewriterConfig.ToggleProtobuf enumtensorflow.RewriterConfig.ToggleNested classes/interfaces inherited from class com.google.protobuf.GeneratedMessageV3
com.google.protobuf.GeneratedMessageV3.BuilderParent, com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage,BuilderType extends com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType,BuilderType>>, com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage>, com.google.protobuf.GeneratedMessageV3.ExtendableMessageOrBuilder<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage>, com.google.protobuf.GeneratedMessageV3.FieldAccessorTable, com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter -
Field Summary
-
Method Summary
Modifier and Type Method Description booleanequals(java.lang.Object obj)RewriterConfig.TogglegetArithmeticOptimization()Arithmetic optimizations (default is ON) e.g.intgetArithmeticOptimizationValue()Arithmetic optimizations (default is ON) e.g.RewriterConfig.TogglegetAutoMixedPrecision()Optimize data types (default is OFF).intgetAutoMixedPrecisionValue()Optimize data types (default is OFF).AutoParallelOptionsgetAutoParallel()Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.AutoParallelOptionsOrBuildergetAutoParallelOrBuilder()Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.RewriterConfig.TogglegetConstantFolding()Fold constants (default is ON) Statically infer the value of tensors when possible, and materialize the result using constants.intgetConstantFoldingValue()Fold constants (default is ON) Statically infer the value of tensors when possible, and materialize the result using constants.RewriterConfig.CustomGraphOptimizergetCustomOptimizers(int index)list of CustomGraphOptimizers to apply.intgetCustomOptimizersCount()list of CustomGraphOptimizers to apply.java.util.List<RewriterConfig.CustomGraphOptimizer>getCustomOptimizersList()list of CustomGraphOptimizers to apply.RewriterConfig.CustomGraphOptimizerOrBuildergetCustomOptimizersOrBuilder(int index)list of CustomGraphOptimizers to apply.java.util.List<? extends RewriterConfig.CustomGraphOptimizerOrBuilder>getCustomOptimizersOrBuilderList()list of CustomGraphOptimizers to apply.RewriterConfig.TogglegetDebugStripper()Strips debug-related nodes from the graph (off by default).intgetDebugStripperValue()Strips debug-related nodes from the graph (off by default).static RewriterConfiggetDefaultInstance()RewriterConfiggetDefaultInstanceForType()RewriterConfig.TogglegetDependencyOptimization()Control dependency optimizations (default is ON).intgetDependencyOptimizationValue()Control dependency optimizations (default is ON).static com.google.protobuf.Descriptors.DescriptorgetDescriptor()booleangetDisableMetaOptimizer()Disable the entire meta optimizer (off by default).booleangetDisableModelPruning()If true, don't remove unnecessary ops from the graphbooleangetFailOnOptimizerErrors()If true, any optimization pass failing will cause the MetaOptimizer to stop with an error.RewriterConfig.TogglegetFunctionOptimization()Function optimizations (default is ON).intgetFunctionOptimizationValue()Function optimizations (default is ON).RewriterConfig.TogglegetImplementationSelector()Enable the swap of kernel implementations based on the device placement (default is ON).intgetImplementationSelectorValue()Enable the swap of kernel implementations based on the device placement (default is ON).VerifierConfiggetInterOptimizerVerifierConfig()VerifierConfig specifying the verifiers to be run after every optimizer.VerifierConfigOrBuildergetInterOptimizerVerifierConfigOrBuilder()VerifierConfig specifying the verifiers to be run after every optimizer.RewriterConfig.TogglegetLayoutOptimizer()Optimize tensor layouts (default is ON) e.g.intgetLayoutOptimizerValue()Optimize tensor layouts (default is ON) e.g.RewriterConfig.TogglegetLoopOptimization()Loop optimizations (default is ON).intgetLoopOptimizationValue()Loop optimizations (default is ON).RewriterConfig.MemOptTypegetMemoryOptimization()Configures memory optimization passes through the meta-optimizer.intgetMemoryOptimizationValue()Configures memory optimization passes through the meta-optimizer.java.lang.StringgetMemoryOptimizerTargetNodeNameScope()A node name scope for node names which are valid outputs of recomputations.com.google.protobuf.ByteStringgetMemoryOptimizerTargetNodeNameScopeBytes()A node name scope for node names which are valid outputs of recomputations.RewriterConfig.NumIterationsTypegetMetaOptimizerIterations()Controls how many times we run the optimizers in meta optimizer (default is once).intgetMetaOptimizerIterationsValue()Controls how many times we run the optimizers in meta optimizer (default is once).longgetMetaOptimizerTimeoutMs()Maximum number of milliseconds to spend optimizing a single graph before timing out.intgetMinGraphNodes()The minimum number of nodes in a graph to optimizer.java.lang.StringgetOptimizers(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.ByteStringgetOptimizersBytes(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).intgetOptimizersCount()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.ProtocolStringListgetOptimizersList()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.Parser<RewriterConfig>getParserForType()RewriterConfig.TogglegetPinToHostOptimization()Force small ops onto the CPU (default is OFF).intgetPinToHostOptimizationValue()Force small ops onto the CPU (default is OFF).VerifierConfiggetPostOptimizationVerifierConfig()VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.VerifierConfigOrBuildergetPostOptimizationVerifierConfigOrBuilder()VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.RewriterConfig.TogglegetRemapping()Remapping (default is ON) Remap subgraphs onto more efficient implementations.intgetRemappingValue()Remapping (default is ON) Remap subgraphs onto more efficient implementations.RewriterConfig.TogglegetScopedAllocatorOptimization()Try to allocate some independent Op outputs contiguously in order to merge or eliminate downstream Ops (off by default).intgetScopedAllocatorOptimizationValue()Try to allocate some independent Op outputs contiguously in order to merge or eliminate downstream Ops (off by default).ScopedAllocatorOptionsgetScopedAllocatorOpts().tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;ScopedAllocatorOptionsOrBuildergetScopedAllocatorOptsOrBuilder().tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;intgetSerializedSize()RewriterConfig.TogglegetShapeOptimization()Shape optimizations (default is ON) Simplify computations made on shapes.intgetShapeOptimizationValue()Shape optimizations (default is ON) Simplify computations made on shapes.com.google.protobuf.UnknownFieldSetgetUnknownFields()booleanhasAutoParallel()Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.inthashCode()booleanhasInterOptimizerVerifierConfig()VerifierConfig specifying the verifiers to be run after every optimizer.booleanhasPostOptimizationVerifierConfig()VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.booleanhasScopedAllocatorOpts().tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTableinternalGetFieldAccessorTable()booleanisInitialized()static RewriterConfig.BuildernewBuilder()static RewriterConfig.BuildernewBuilder(RewriterConfig prototype)RewriterConfig.BuildernewBuilderForType()protected RewriterConfig.BuildernewBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)protected java.lang.ObjectnewInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)static RewriterConfigparseDelimitedFrom(java.io.InputStream input)static RewriterConfigparseDelimitedFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)static RewriterConfigparseFrom(byte[] data)static RewriterConfigparseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)static RewriterConfigparseFrom(com.google.protobuf.ByteString data)static RewriterConfigparseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)static RewriterConfigparseFrom(com.google.protobuf.CodedInputStream input)static RewriterConfigparseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)static RewriterConfigparseFrom(java.io.InputStream input)static RewriterConfigparseFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)static RewriterConfigparseFrom(java.nio.ByteBuffer data)static RewriterConfigparseFrom(java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)static com.google.protobuf.Parser<RewriterConfig>parser()RewriterConfig.BuildertoBuilder()voidwriteTo(com.google.protobuf.CodedOutputStream output)Methods inherited from class com.google.protobuf.GeneratedMessageV3
canUseUnsafe, computeStringSize, computeStringSizeNoTag, emptyBooleanList, emptyDoubleList, emptyFloatList, emptyIntList, emptyLongList, getAllFields, getDescriptorForType, getField, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, hasField, hasOneof, internalGetMapField, makeExtensionsImmutable, mergeFromAndMakeImmutableInternal, mutableCopy, mutableCopy, mutableCopy, mutableCopy, mutableCopy, newBooleanList, newBuilderForType, newDoubleList, newFloatList, newIntList, newLongList, parseDelimitedWithIOException, parseDelimitedWithIOException, parseUnknownField, parseUnknownFieldProto3, parseWithIOException, parseWithIOException, parseWithIOException, parseWithIOException, serializeBooleanMapTo, serializeIntegerMapTo, serializeLongMapTo, serializeStringMapTo, writeReplace, writeString, writeStringNoTagMethods inherited from class com.google.protobuf.AbstractMessage
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toStringMethods inherited from class com.google.protobuf.AbstractMessageLite
addAll, addAll, checkByteStringIsUtf8, toByteArray, toByteString, writeDelimitedTo, writeToMethods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Field Details
-
LAYOUT_OPTIMIZER_FIELD_NUMBER
public static final int LAYOUT_OPTIMIZER_FIELD_NUMBER- See Also:
- Constant Field Values
-
CONSTANT_FOLDING_FIELD_NUMBER
public static final int CONSTANT_FOLDING_FIELD_NUMBER- See Also:
- Constant Field Values
-
SHAPE_OPTIMIZATION_FIELD_NUMBER
public static final int SHAPE_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
REMAPPING_FIELD_NUMBER
public static final int REMAPPING_FIELD_NUMBER- See Also:
- Constant Field Values
-
ARITHMETIC_OPTIMIZATION_FIELD_NUMBER
public static final int ARITHMETIC_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
DEPENDENCY_OPTIMIZATION_FIELD_NUMBER
public static final int DEPENDENCY_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
LOOP_OPTIMIZATION_FIELD_NUMBER
public static final int LOOP_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
FUNCTION_OPTIMIZATION_FIELD_NUMBER
public static final int FUNCTION_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
DEBUG_STRIPPER_FIELD_NUMBER
public static final int DEBUG_STRIPPER_FIELD_NUMBER- See Also:
- Constant Field Values
-
DISABLE_MODEL_PRUNING_FIELD_NUMBER
public static final int DISABLE_MODEL_PRUNING_FIELD_NUMBER- See Also:
- Constant Field Values
-
SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER
public static final int SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER
public static final int PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
IMPLEMENTATION_SELECTOR_FIELD_NUMBER
public static final int IMPLEMENTATION_SELECTOR_FIELD_NUMBER- See Also:
- Constant Field Values
-
AUTO_MIXED_PRECISION_FIELD_NUMBER
public static final int AUTO_MIXED_PRECISION_FIELD_NUMBER- See Also:
- Constant Field Values
-
DISABLE_META_OPTIMIZER_FIELD_NUMBER
public static final int DISABLE_META_OPTIMIZER_FIELD_NUMBER- See Also:
- Constant Field Values
-
META_OPTIMIZER_ITERATIONS_FIELD_NUMBER
public static final int META_OPTIMIZER_ITERATIONS_FIELD_NUMBER- See Also:
- Constant Field Values
-
MIN_GRAPH_NODES_FIELD_NUMBER
public static final int MIN_GRAPH_NODES_FIELD_NUMBER- See Also:
- Constant Field Values
-
MEMORY_OPTIMIZATION_FIELD_NUMBER
public static final int MEMORY_OPTIMIZATION_FIELD_NUMBER- See Also:
- Constant Field Values
-
MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER
public static final int MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER- See Also:
- Constant Field Values
-
META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER
public static final int META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER- See Also:
- Constant Field Values
-
AUTO_PARALLEL_FIELD_NUMBER
public static final int AUTO_PARALLEL_FIELD_NUMBER- See Also:
- Constant Field Values
-
FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER
public static final int FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER- See Also:
- Constant Field Values
-
SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER
public static final int SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER- See Also:
- Constant Field Values
-
OPTIMIZERS_FIELD_NUMBER
public static final int OPTIMIZERS_FIELD_NUMBER- See Also:
- Constant Field Values
-
CUSTOM_OPTIMIZERS_FIELD_NUMBER
public static final int CUSTOM_OPTIMIZERS_FIELD_NUMBER- See Also:
- Constant Field Values
-
INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER
public static final int INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER- See Also:
- Constant Field Values
-
POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER
public static final int POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER- See Also:
- Constant Field Values
-
-
Method Details
-
newInstance
protected java.lang.Object newInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)- Overrides:
newInstancein classcom.google.protobuf.GeneratedMessageV3
-
getUnknownFields
public final com.google.protobuf.UnknownFieldSet getUnknownFields()- Specified by:
getUnknownFieldsin interfacecom.google.protobuf.MessageOrBuilder- Overrides:
getUnknownFieldsin classcom.google.protobuf.GeneratedMessageV3
-
getDescriptor
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() -
internalGetFieldAccessorTable
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()- Specified by:
internalGetFieldAccessorTablein classcom.google.protobuf.GeneratedMessageV3
-
getLayoutOptimizerValue
public int getLayoutOptimizerValue()Optimize tensor layouts (default is ON) e.g. This will try to use NCHW layout on GPU which is faster.
.tensorflow.RewriterConfig.Toggle layout_optimizer = 1;- Specified by:
getLayoutOptimizerValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for layoutOptimizer.
-
getLayoutOptimizer
Optimize tensor layouts (default is ON) e.g. This will try to use NCHW layout on GPU which is faster.
.tensorflow.RewriterConfig.Toggle layout_optimizer = 1;- Specified by:
getLayoutOptimizerin interfaceRewriterConfigOrBuilder- Returns:
- The layoutOptimizer.
-
getConstantFoldingValue
public int getConstantFoldingValue()Fold constants (default is ON) Statically infer the value of tensors when possible, and materialize the result using constants.
.tensorflow.RewriterConfig.Toggle constant_folding = 3;- Specified by:
getConstantFoldingValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for constantFolding.
-
getConstantFolding
Fold constants (default is ON) Statically infer the value of tensors when possible, and materialize the result using constants.
.tensorflow.RewriterConfig.Toggle constant_folding = 3;- Specified by:
getConstantFoldingin interfaceRewriterConfigOrBuilder- Returns:
- The constantFolding.
-
getShapeOptimizationValue
public int getShapeOptimizationValue()Shape optimizations (default is ON) Simplify computations made on shapes.
.tensorflow.RewriterConfig.Toggle shape_optimization = 13;- Specified by:
getShapeOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for shapeOptimization.
-
getShapeOptimization
Shape optimizations (default is ON) Simplify computations made on shapes.
.tensorflow.RewriterConfig.Toggle shape_optimization = 13;- Specified by:
getShapeOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The shapeOptimization.
-
getRemappingValue
public int getRemappingValue()Remapping (default is ON) Remap subgraphs onto more efficient implementations.
.tensorflow.RewriterConfig.Toggle remapping = 14;- Specified by:
getRemappingValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for remapping.
-
getRemapping
Remapping (default is ON) Remap subgraphs onto more efficient implementations.
.tensorflow.RewriterConfig.Toggle remapping = 14;- Specified by:
getRemappingin interfaceRewriterConfigOrBuilder- Returns:
- The remapping.
-
getArithmeticOptimizationValue
public int getArithmeticOptimizationValue()Arithmetic optimizations (default is ON) e.g. Simplify arithmetic ops; merge ops with same value (like constants).
.tensorflow.RewriterConfig.Toggle arithmetic_optimization = 7;- Specified by:
getArithmeticOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for arithmeticOptimization.
-
getArithmeticOptimization
Arithmetic optimizations (default is ON) e.g. Simplify arithmetic ops; merge ops with same value (like constants).
.tensorflow.RewriterConfig.Toggle arithmetic_optimization = 7;- Specified by:
getArithmeticOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The arithmeticOptimization.
-
getDependencyOptimizationValue
public int getDependencyOptimizationValue()Control dependency optimizations (default is ON). Remove redundant control dependencies, which may enable other optimization.
.tensorflow.RewriterConfig.Toggle dependency_optimization = 8;- Specified by:
getDependencyOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for dependencyOptimization.
-
getDependencyOptimization
Control dependency optimizations (default is ON). Remove redundant control dependencies, which may enable other optimization.
.tensorflow.RewriterConfig.Toggle dependency_optimization = 8;- Specified by:
getDependencyOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The dependencyOptimization.
-
getLoopOptimizationValue
public int getLoopOptimizationValue()Loop optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle loop_optimization = 9;- Specified by:
getLoopOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for loopOptimization.
-
getLoopOptimization
Loop optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle loop_optimization = 9;- Specified by:
getLoopOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The loopOptimization.
-
getFunctionOptimizationValue
public int getFunctionOptimizationValue()Function optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle function_optimization = 10;- Specified by:
getFunctionOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for functionOptimization.
-
getFunctionOptimization
Function optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle function_optimization = 10;- Specified by:
getFunctionOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The functionOptimization.
-
getDebugStripperValue
public int getDebugStripperValue()Strips debug-related nodes from the graph (off by default).
.tensorflow.RewriterConfig.Toggle debug_stripper = 11;- Specified by:
getDebugStripperValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for debugStripper.
-
getDebugStripper
Strips debug-related nodes from the graph (off by default).
.tensorflow.RewriterConfig.Toggle debug_stripper = 11;- Specified by:
getDebugStripperin interfaceRewriterConfigOrBuilder- Returns:
- The debugStripper.
-
getDisableModelPruning
public boolean getDisableModelPruning()If true, don't remove unnecessary ops from the graph
bool disable_model_pruning = 2;- Specified by:
getDisableModelPruningin interfaceRewriterConfigOrBuilder- Returns:
- The disableModelPruning.
-
getScopedAllocatorOptimizationValue
public int getScopedAllocatorOptimizationValue()Try to allocate some independent Op outputs contiguously in order to merge or eliminate downstream Ops (off by default).
.tensorflow.RewriterConfig.Toggle scoped_allocator_optimization = 15;- Specified by:
getScopedAllocatorOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for scopedAllocatorOptimization.
-
getScopedAllocatorOptimization
Try to allocate some independent Op outputs contiguously in order to merge or eliminate downstream Ops (off by default).
.tensorflow.RewriterConfig.Toggle scoped_allocator_optimization = 15;- Specified by:
getScopedAllocatorOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The scopedAllocatorOptimization.
-
getPinToHostOptimizationValue
public int getPinToHostOptimizationValue()Force small ops onto the CPU (default is OFF).
.tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;- Specified by:
getPinToHostOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for pinToHostOptimization.
-
getPinToHostOptimization
Force small ops onto the CPU (default is OFF).
.tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;- Specified by:
getPinToHostOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The pinToHostOptimization.
-
getImplementationSelectorValue
public int getImplementationSelectorValue()Enable the swap of kernel implementations based on the device placement (default is ON).
.tensorflow.RewriterConfig.Toggle implementation_selector = 22;- Specified by:
getImplementationSelectorValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for implementationSelector.
-
getImplementationSelector
Enable the swap of kernel implementations based on the device placement (default is ON).
.tensorflow.RewriterConfig.Toggle implementation_selector = 22;- Specified by:
getImplementationSelectorin interfaceRewriterConfigOrBuilder- Returns:
- The implementationSelector.
-
getAutoMixedPrecisionValue
public int getAutoMixedPrecisionValue()Optimize data types (default is OFF). e.g., This will try to use float16 on GPU which is faster. Note that this can change the numerical stability of the graph and may require the use of loss scaling to maintain model convergence.
.tensorflow.RewriterConfig.Toggle auto_mixed_precision = 23;- Specified by:
getAutoMixedPrecisionValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for autoMixedPrecision.
-
getAutoMixedPrecision
Optimize data types (default is OFF). e.g., This will try to use float16 on GPU which is faster. Note that this can change the numerical stability of the graph and may require the use of loss scaling to maintain model convergence.
.tensorflow.RewriterConfig.Toggle auto_mixed_precision = 23;- Specified by:
getAutoMixedPrecisionin interfaceRewriterConfigOrBuilder- Returns:
- The autoMixedPrecision.
-
getDisableMetaOptimizer
public boolean getDisableMetaOptimizer()Disable the entire meta optimizer (off by default).
bool disable_meta_optimizer = 19;- Specified by:
getDisableMetaOptimizerin interfaceRewriterConfigOrBuilder- Returns:
- The disableMetaOptimizer.
-
getMetaOptimizerIterationsValue
public int getMetaOptimizerIterationsValue()Controls how many times we run the optimizers in meta optimizer (default is once).
.tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;- Specified by:
getMetaOptimizerIterationsValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for metaOptimizerIterations.
-
getMetaOptimizerIterations
Controls how many times we run the optimizers in meta optimizer (default is once).
.tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;- Specified by:
getMetaOptimizerIterationsin interfaceRewriterConfigOrBuilder- Returns:
- The metaOptimizerIterations.
-
getMinGraphNodes
public int getMinGraphNodes()The minimum number of nodes in a graph to optimizer. For smaller graphs, optimization is skipped. 0 means the system picks an appropriate number. < 0 means do not skip optimization.
int32 min_graph_nodes = 17;- Specified by:
getMinGraphNodesin interfaceRewriterConfigOrBuilder- Returns:
- The minGraphNodes.
-
getMemoryOptimizationValue
public int getMemoryOptimizationValue()Configures memory optimization passes through the meta-optimizer. Has no effect on manually requested memory optimization passes in the optimizers field.
.tensorflow.RewriterConfig.MemOptType memory_optimization = 4;- Specified by:
getMemoryOptimizationValuein interfaceRewriterConfigOrBuilder- Returns:
- The enum numeric value on the wire for memoryOptimization.
-
getMemoryOptimization
Configures memory optimization passes through the meta-optimizer. Has no effect on manually requested memory optimization passes in the optimizers field.
.tensorflow.RewriterConfig.MemOptType memory_optimization = 4;- Specified by:
getMemoryOptimizationin interfaceRewriterConfigOrBuilder- Returns:
- The memoryOptimization.
-
getMemoryOptimizerTargetNodeNameScope
public java.lang.String getMemoryOptimizerTargetNodeNameScope()A node name scope for node names which are valid outputs of recomputations. Inputs to nodes that match this scope may be recomputed (subject either to manual annotation of those input nodes or to manual annotation and heuristics depending on memory_optimization), but the nodes themselves will not be recomputed. This matches any sub-scopes as well, meaning the scope can appear not just as a top-level scope. For example, if the value is "gradients/", the default, it will match node name "gradients/foo", "foo/gradients/bar", but not "foo_gradients/"
string memory_optimizer_target_node_name_scope = 6;- Specified by:
getMemoryOptimizerTargetNodeNameScopein interfaceRewriterConfigOrBuilder- Returns:
- The memoryOptimizerTargetNodeNameScope.
-
getMemoryOptimizerTargetNodeNameScopeBytes
public com.google.protobuf.ByteString getMemoryOptimizerTargetNodeNameScopeBytes()A node name scope for node names which are valid outputs of recomputations. Inputs to nodes that match this scope may be recomputed (subject either to manual annotation of those input nodes or to manual annotation and heuristics depending on memory_optimization), but the nodes themselves will not be recomputed. This matches any sub-scopes as well, meaning the scope can appear not just as a top-level scope. For example, if the value is "gradients/", the default, it will match node name "gradients/foo", "foo/gradients/bar", but not "foo_gradients/"
string memory_optimizer_target_node_name_scope = 6;- Specified by:
getMemoryOptimizerTargetNodeNameScopeBytesin interfaceRewriterConfigOrBuilder- Returns:
- The bytes for memoryOptimizerTargetNodeNameScope.
-
getMetaOptimizerTimeoutMs
public long getMetaOptimizerTimeoutMs()Maximum number of milliseconds to spend optimizing a single graph before timing out. If equal to 0 the system picks a default (currently 5 minutes). If less than 0 the optimizer will never time out.
int64 meta_optimizer_timeout_ms = 20;- Specified by:
getMetaOptimizerTimeoutMsin interfaceRewriterConfigOrBuilder- Returns:
- The metaOptimizerTimeoutMs.
-
hasAutoParallel
public boolean hasAutoParallel()Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.
.tensorflow.AutoParallelOptions auto_parallel = 5;- Specified by:
hasAutoParallelin interfaceRewriterConfigOrBuilder- Returns:
- Whether the autoParallel field is set.
-
getAutoParallel
Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.
.tensorflow.AutoParallelOptions auto_parallel = 5;- Specified by:
getAutoParallelin interfaceRewriterConfigOrBuilder- Returns:
- The autoParallel.
-
getAutoParallelOrBuilder
Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.
.tensorflow.AutoParallelOptions auto_parallel = 5;- Specified by:
getAutoParallelOrBuilderin interfaceRewriterConfigOrBuilder
-
getFailOnOptimizerErrors
public boolean getFailOnOptimizerErrors()If true, any optimization pass failing will cause the MetaOptimizer to stop with an error. By default - or when set to false, failing passes are skipped silently.
bool fail_on_optimizer_errors = 21;- Specified by:
getFailOnOptimizerErrorsin interfaceRewriterConfigOrBuilder- Returns:
- The failOnOptimizerErrors.
-
hasScopedAllocatorOpts
public boolean hasScopedAllocatorOpts().tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;- Specified by:
hasScopedAllocatorOptsin interfaceRewriterConfigOrBuilder- Returns:
- Whether the scopedAllocatorOpts field is set.
-
getScopedAllocatorOpts
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;- Specified by:
getScopedAllocatorOptsin interfaceRewriterConfigOrBuilder- Returns:
- The scopedAllocatorOpts.
-
getScopedAllocatorOptsOrBuilder
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;- Specified by:
getScopedAllocatorOptsOrBuilderin interfaceRewriterConfigOrBuilder
-
getOptimizersList
public 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). Of the RewriterConfig options, only the AutoParallel configuration options (the auto_parallel field) apply to manually requested optimization passes ("autoparallel"). Memory optimization passes ("memory") invoked here are not configurable (in contrast to memory optimization passes through the meta-optimizer) and act only on manual op annotations. Custom optimizers (see custom_optimizers) that are not part of this schedule will be run after - in the order that they were specified.repeated string optimizers = 100;- Specified by:
getOptimizersListin interfaceRewriterConfigOrBuilder- Returns:
- A list containing the optimizers.
-
getOptimizersCount
public 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). Of the RewriterConfig options, only the AutoParallel configuration options (the auto_parallel field) apply to manually requested optimization passes ("autoparallel"). Memory optimization passes ("memory") invoked here are not configurable (in contrast to memory optimization passes through the meta-optimizer) and act only on manual op annotations. Custom optimizers (see custom_optimizers) that are not part of this schedule will be run after - in the order that they were specified.repeated string optimizers = 100;- Specified by:
getOptimizersCountin interfaceRewriterConfigOrBuilder- Returns:
- The count of optimizers.
-
getOptimizers
public java.lang.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). Of the RewriterConfig options, only the AutoParallel configuration options (the auto_parallel field) apply to manually requested optimization passes ("autoparallel"). Memory optimization passes ("memory") invoked here are not configurable (in contrast to memory optimization passes through the meta-optimizer) and act only on manual op annotations. Custom optimizers (see custom_optimizers) that are not part of this schedule will be run after - in the order that they were specified.repeated string optimizers = 100;- Specified by:
getOptimizersin interfaceRewriterConfigOrBuilder- Parameters:
index- The index of the element to return.- Returns:
- The optimizers at the given index.
-
getOptimizersBytes
public 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). Of the RewriterConfig options, only the AutoParallel configuration options (the auto_parallel field) apply to manually requested optimization passes ("autoparallel"). Memory optimization passes ("memory") invoked here are not configurable (in contrast to memory optimization passes through the meta-optimizer) and act only on manual op annotations. Custom optimizers (see custom_optimizers) that are not part of this schedule will be run after - in the order that they were specified.repeated string optimizers = 100;- Specified by:
getOptimizersBytesin interfaceRewriterConfigOrBuilder- Parameters:
index- The index of the value to return.- Returns:
- The bytes of the optimizers at the given index.
-
getCustomOptimizersList
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;- Specified by:
getCustomOptimizersListin interfaceRewriterConfigOrBuilder
-
getCustomOptimizersOrBuilderList
public java.util.List<? extends RewriterConfig.CustomGraphOptimizerOrBuilder> getCustomOptimizersOrBuilderList()list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;- Specified by:
getCustomOptimizersOrBuilderListin interfaceRewriterConfigOrBuilder
-
getCustomOptimizersCount
public int getCustomOptimizersCount()list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;- Specified by:
getCustomOptimizersCountin interfaceRewriterConfigOrBuilder
-
getCustomOptimizers
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;- Specified by:
getCustomOptimizersin interfaceRewriterConfigOrBuilder
-
getCustomOptimizersOrBuilder
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;- Specified by:
getCustomOptimizersOrBuilderin interfaceRewriterConfigOrBuilder
-
hasInterOptimizerVerifierConfig
public boolean hasInterOptimizerVerifierConfig()VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;- Specified by:
hasInterOptimizerVerifierConfigin interfaceRewriterConfigOrBuilder- Returns:
- Whether the interOptimizerVerifierConfig field is set.
-
getInterOptimizerVerifierConfig
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;- Specified by:
getInterOptimizerVerifierConfigin interfaceRewriterConfigOrBuilder- Returns:
- The interOptimizerVerifierConfig.
-
getInterOptimizerVerifierConfigOrBuilder
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;- Specified by:
getInterOptimizerVerifierConfigOrBuilderin interfaceRewriterConfigOrBuilder
-
hasPostOptimizationVerifierConfig
public boolean hasPostOptimizationVerifierConfig()VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.
.tensorflow.VerifierConfig post_optimization_verifier_config = 301;- Specified by:
hasPostOptimizationVerifierConfigin interfaceRewriterConfigOrBuilder- Returns:
- Whether the postOptimizationVerifierConfig field is set.
-
getPostOptimizationVerifierConfig
VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.
.tensorflow.VerifierConfig post_optimization_verifier_config = 301;- Specified by:
getPostOptimizationVerifierConfigin interfaceRewriterConfigOrBuilder- Returns:
- The postOptimizationVerifierConfig.
-
getPostOptimizationVerifierConfigOrBuilder
VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.
.tensorflow.VerifierConfig post_optimization_verifier_config = 301;- Specified by:
getPostOptimizationVerifierConfigOrBuilderin interfaceRewriterConfigOrBuilder
-
isInitialized
public final boolean isInitialized()- Specified by:
isInitializedin interfacecom.google.protobuf.MessageLiteOrBuilder- Overrides:
isInitializedin classcom.google.protobuf.GeneratedMessageV3
-
writeTo
public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException- Specified by:
writeToin interfacecom.google.protobuf.MessageLite- Overrides:
writeToin classcom.google.protobuf.GeneratedMessageV3- Throws:
java.io.IOException
-
getSerializedSize
public int getSerializedSize()- Specified by:
getSerializedSizein interfacecom.google.protobuf.MessageLite- Overrides:
getSerializedSizein classcom.google.protobuf.GeneratedMessageV3
-
equals
public boolean equals(java.lang.Object obj)- Specified by:
equalsin interfacecom.google.protobuf.Message- Overrides:
equalsin classcom.google.protobuf.AbstractMessage
-
hashCode
public int hashCode()- Specified by:
hashCodein interfacecom.google.protobuf.Message- Overrides:
hashCodein classcom.google.protobuf.AbstractMessage
-
parseFrom
public static RewriterConfig parseFrom(java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static RewriterConfig parseFrom(java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static RewriterConfig parseFrom(com.google.protobuf.ByteString data) throws com.google.protobuf.InvalidProtocolBufferException- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static RewriterConfig parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static RewriterConfig parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static RewriterConfig parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
- Throws:
java.io.IOException
-
parseFrom
public static RewriterConfig parseFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException- Throws:
java.io.IOException
-
parseDelimitedFrom
public static RewriterConfig parseDelimitedFrom(java.io.InputStream input) throws java.io.IOException- Throws:
java.io.IOException
-
parseDelimitedFrom
public static RewriterConfig parseDelimitedFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException- Throws:
java.io.IOException
-
parseFrom
public static RewriterConfig parseFrom(com.google.protobuf.CodedInputStream input) throws java.io.IOException- Throws:
java.io.IOException
-
parseFrom
public static RewriterConfig parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException- Throws:
java.io.IOException
-
newBuilderForType
- Specified by:
newBuilderForTypein interfacecom.google.protobuf.Message- Specified by:
newBuilderForTypein interfacecom.google.protobuf.MessageLite
-
newBuilder
-
newBuilder
-
toBuilder
- Specified by:
toBuilderin interfacecom.google.protobuf.Message- Specified by:
toBuilderin interfacecom.google.protobuf.MessageLite
-
newBuilderForType
protected RewriterConfig.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)- Specified by:
newBuilderForTypein classcom.google.protobuf.GeneratedMessageV3
-
getDefaultInstance
-
parser
-
getParserForType
- Specified by:
getParserForTypein interfacecom.google.protobuf.Message- Specified by:
getParserForTypein interfacecom.google.protobuf.MessageLite- Overrides:
getParserForTypein classcom.google.protobuf.GeneratedMessageV3
-
getDefaultInstanceForType
- Specified by:
getDefaultInstanceForTypein interfacecom.google.protobuf.MessageLiteOrBuilder- Specified by:
getDefaultInstanceForTypein interfacecom.google.protobuf.MessageOrBuilder
-