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
Modifier and Type | Class and Description |
---|---|
static class |
RewriterConfig.Builder
Graph rewriting is experimental and subject to change, not covered by any
API stability guarantees.
|
static class |
RewriterConfig.CustomGraphOptimizer
Message to describe custom graph optimizer and its parameters
|
static interface |
RewriterConfig.CustomGraphOptimizerOrBuilder |
static class |
RewriterConfig.MemOptType
Protobuf enum
tensorflow.RewriterConfig.MemOptType |
static class |
RewriterConfig.NumIterationsType
Enum controlling the number of times to run optimizers.
|
static class |
RewriterConfig.Toggle
Protobuf enum
tensorflow.RewriterConfig.Toggle |
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
Modifier and Type | Method and Description |
---|---|
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 (default is OFF).
|
int |
getAutoMixedPrecisionValue()
Optimize data types (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 |
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.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 |
getDefaultInstance() |
RewriterConfig |
getDefaultInstanceForType() |
RewriterConfig.Toggle |
getDependencyOptimization()
Control dependency optimizations (default is ON).
|
int |
getDependencyOptimizationValue()
Control dependency optimizations (default is ON).
|
static com.google.protobuf.Descriptors.Descriptor |
getDescriptor() |
boolean |
getDisableMetaOptimizer()
Disable the entire meta optimizer (off by default).
|
boolean |
getDisableModelPruning()
If true, don't remove unnecessary ops from the graph
|
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 recompuations.
|
com.google.protobuf.ByteString |
getMemoryOptimizerTargetNodeNameScopeBytes()
A node name scope for node names which are valid outputs of recompuations.
|
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).
|
com.google.protobuf.Parser<RewriterConfig> |
getParserForType() |
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 |
getSerializedSize() |
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.
|
com.google.protobuf.UnknownFieldSet |
getUnknownFields() |
boolean |
hasAutoParallel()
Configures AutoParallel optimization passes either through the
meta-optimizer or when manually specified through the optimizers field.
|
int |
hashCode() |
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.
|
boolean |
hasScopedAllocatorOpts()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16; |
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable |
internalGetFieldAccessorTable() |
boolean |
isInitialized() |
static RewriterConfig.Builder |
newBuilder() |
static RewriterConfig.Builder |
newBuilder(RewriterConfig prototype) |
RewriterConfig.Builder |
newBuilderForType() |
protected RewriterConfig.Builder |
newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent) |
static RewriterConfig |
parseDelimitedFrom(InputStream input) |
static RewriterConfig |
parseDelimitedFrom(InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static RewriterConfig |
parseFrom(byte[] data) |
static RewriterConfig |
parseFrom(byte[] data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static RewriterConfig |
parseFrom(ByteBuffer data) |
static RewriterConfig |
parseFrom(ByteBuffer data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static RewriterConfig |
parseFrom(com.google.protobuf.ByteString data) |
static RewriterConfig |
parseFrom(com.google.protobuf.ByteString data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static RewriterConfig |
parseFrom(com.google.protobuf.CodedInputStream input) |
static RewriterConfig |
parseFrom(com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static RewriterConfig |
parseFrom(InputStream input) |
static RewriterConfig |
parseFrom(InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static com.google.protobuf.Parser<RewriterConfig> |
parser() |
RewriterConfig.Builder |
toBuilder() |
void |
writeTo(com.google.protobuf.CodedOutputStream output) |
canUseUnsafe, computeStringSize, computeStringSizeNoTag, getAllFields, getDescriptorForType, getField, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, hasField, hasOneof, internalGetMapField, makeExtensionsImmutable, newBuilderForType, parseDelimitedWithIOException, parseDelimitedWithIOException, parseUnknownField, parseUnknownFieldProto3, parseWithIOException, parseWithIOException, parseWithIOException, parseWithIOException, serializeBooleanMapTo, serializeIntegerMapTo, serializeLongMapTo, serializeStringMapTo, writeReplace, writeString, writeStringNoTag
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toString
addAll, addAll, checkByteStringIsUtf8, toByteArray, toByteString, writeDelimitedTo, writeTo
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
public static final int LAYOUT_OPTIMIZER_FIELD_NUMBER
public static final int CONSTANT_FOLDING_FIELD_NUMBER
public static final int SHAPE_OPTIMIZATION_FIELD_NUMBER
public static final int REMAPPING_FIELD_NUMBER
public static final int ARITHMETIC_OPTIMIZATION_FIELD_NUMBER
public static final int DEPENDENCY_OPTIMIZATION_FIELD_NUMBER
public static final int LOOP_OPTIMIZATION_FIELD_NUMBER
public static final int FUNCTION_OPTIMIZATION_FIELD_NUMBER
public static final int DEBUG_STRIPPER_FIELD_NUMBER
public static final int DISABLE_MODEL_PRUNING_FIELD_NUMBER
public static final int SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER
public static final int PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER
public static final int IMPLEMENTATION_SELECTOR_FIELD_NUMBER
public static final int AUTO_MIXED_PRECISION_FIELD_NUMBER
public static final int DISABLE_META_OPTIMIZER_FIELD_NUMBER
public static final int META_OPTIMIZER_ITERATIONS_FIELD_NUMBER
public static final int MIN_GRAPH_NODES_FIELD_NUMBER
public static final int MEMORY_OPTIMIZATION_FIELD_NUMBER
public static final int MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER
public static final int META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER
public static final int AUTO_PARALLEL_FIELD_NUMBER
public static final int FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER
public static final int SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER
public static final int OPTIMIZERS_FIELD_NUMBER
public static final int CUSTOM_OPTIMIZERS_FIELD_NUMBER
public static final int INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER
public static final int POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER
public final com.google.protobuf.UnknownFieldSet getUnknownFields()
getUnknownFields
in interface com.google.protobuf.MessageOrBuilder
getUnknownFields
in class com.google.protobuf.GeneratedMessageV3
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor()
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
internalGetFieldAccessorTable
in class com.google.protobuf.GeneratedMessageV3
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;
getLayoutOptimizerValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle 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;
getLayoutOptimizer
in interface RewriterConfigOrBuilder
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;
getConstantFoldingValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle 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;
getConstantFolding
in interface RewriterConfigOrBuilder
public int getShapeOptimizationValue()
Shape optimizations (default is ON) Simplify computations made on shapes.
.tensorflow.RewriterConfig.Toggle shape_optimization = 13;
getShapeOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getShapeOptimization()
Shape optimizations (default is ON) Simplify computations made on shapes.
.tensorflow.RewriterConfig.Toggle shape_optimization = 13;
getShapeOptimization
in interface RewriterConfigOrBuilder
public int getRemappingValue()
Remapping (default is ON) Remap subgraphs onto more efficient implementations.
.tensorflow.RewriterConfig.Toggle remapping = 14;
getRemappingValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getRemapping()
Remapping (default is ON) Remap subgraphs onto more efficient implementations.
.tensorflow.RewriterConfig.Toggle remapping = 14;
getRemapping
in interface RewriterConfigOrBuilder
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;
getArithmeticOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getArithmeticOptimization()
Arithmetic optimizations (default is ON) e.g. Simplify arithmetic ops; merge ops with same value (like constants).
.tensorflow.RewriterConfig.Toggle arithmetic_optimization = 7;
getArithmeticOptimization
in interface RewriterConfigOrBuilder
public int getDependencyOptimizationValue()
Control dependency optimizations (default is ON). Remove redundant control dependencies, which may enable other optimization.
.tensorflow.RewriterConfig.Toggle dependency_optimization = 8;
getDependencyOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getDependencyOptimization()
Control dependency optimizations (default is ON). Remove redundant control dependencies, which may enable other optimization.
.tensorflow.RewriterConfig.Toggle dependency_optimization = 8;
getDependencyOptimization
in interface RewriterConfigOrBuilder
public int getLoopOptimizationValue()
Loop optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle loop_optimization = 9;
getLoopOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getLoopOptimization()
Loop optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle loop_optimization = 9;
getLoopOptimization
in interface RewriterConfigOrBuilder
public int getFunctionOptimizationValue()
Function optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle function_optimization = 10;
getFunctionOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getFunctionOptimization()
Function optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle function_optimization = 10;
getFunctionOptimization
in interface RewriterConfigOrBuilder
public int getDebugStripperValue()
Strips debug-related nodes from the graph (off by default).
.tensorflow.RewriterConfig.Toggle debug_stripper = 11;
getDebugStripperValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getDebugStripper()
Strips debug-related nodes from the graph (off by default).
.tensorflow.RewriterConfig.Toggle debug_stripper = 11;
getDebugStripper
in interface RewriterConfigOrBuilder
public boolean getDisableModelPruning()
If true, don't remove unnecessary ops from the graph
bool disable_model_pruning = 2;
getDisableModelPruning
in interface RewriterConfigOrBuilder
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;
getScopedAllocatorOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle 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;
getScopedAllocatorOptimization
in interface RewriterConfigOrBuilder
public int getPinToHostOptimizationValue()
Force small ops onto the CPU (default is OFF).
.tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;
getPinToHostOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getPinToHostOptimization()
Force small ops onto the CPU (default is OFF).
.tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;
getPinToHostOptimization
in interface RewriterConfigOrBuilder
public int getImplementationSelectorValue()
Enable the swap of kernel implementations based on the device placement (default is ON).
.tensorflow.RewriterConfig.Toggle implementation_selector = 22;
getImplementationSelectorValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle getImplementationSelector()
Enable the swap of kernel implementations based on the device placement (default is ON).
.tensorflow.RewriterConfig.Toggle implementation_selector = 22;
getImplementationSelector
in interface RewriterConfigOrBuilder
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;
getAutoMixedPrecisionValue
in interface RewriterConfigOrBuilder
public RewriterConfig.Toggle 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;
getAutoMixedPrecision
in interface RewriterConfigOrBuilder
public boolean getDisableMetaOptimizer()
Disable the entire meta optimizer (off by default).
bool disable_meta_optimizer = 19;
getDisableMetaOptimizer
in interface RewriterConfigOrBuilder
public int getMetaOptimizerIterationsValue()
Controls how many times we run the optimizers in meta optimizer (default is once).
.tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;
getMetaOptimizerIterationsValue
in interface RewriterConfigOrBuilder
public RewriterConfig.NumIterationsType getMetaOptimizerIterations()
Controls how many times we run the optimizers in meta optimizer (default is once).
.tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;
getMetaOptimizerIterations
in interface RewriterConfigOrBuilder
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;
getMinGraphNodes
in interface RewriterConfigOrBuilder
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;
getMemoryOptimizationValue
in interface RewriterConfigOrBuilder
public RewriterConfig.MemOptType 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;
getMemoryOptimization
in interface RewriterConfigOrBuilder
public String getMemoryOptimizerTargetNodeNameScope()
A node name scope for node names which are valid outputs of recompuations. 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;
getMemoryOptimizerTargetNodeNameScope
in interface RewriterConfigOrBuilder
public com.google.protobuf.ByteString getMemoryOptimizerTargetNodeNameScopeBytes()
A node name scope for node names which are valid outputs of recompuations. 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;
getMemoryOptimizerTargetNodeNameScopeBytes
in interface RewriterConfigOrBuilder
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;
getMetaOptimizerTimeoutMs
in interface RewriterConfigOrBuilder
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;
hasAutoParallel
in interface RewriterConfigOrBuilder
public AutoParallelOptions getAutoParallel()
Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.
.tensorflow.AutoParallelOptions auto_parallel = 5;
getAutoParallel
in interface RewriterConfigOrBuilder
public AutoParallelOptionsOrBuilder getAutoParallelOrBuilder()
Configures AutoParallel optimization passes either through the meta-optimizer or when manually specified through the optimizers field.
.tensorflow.AutoParallelOptions auto_parallel = 5;
getAutoParallelOrBuilder
in interface RewriterConfigOrBuilder
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;
getFailOnOptimizerErrors
in interface RewriterConfigOrBuilder
public boolean hasScopedAllocatorOpts()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
hasScopedAllocatorOpts
in interface RewriterConfigOrBuilder
public ScopedAllocatorOptions getScopedAllocatorOpts()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
getScopedAllocatorOpts
in interface RewriterConfigOrBuilder
public ScopedAllocatorOptionsOrBuilder getScopedAllocatorOptsOrBuilder()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
getScopedAllocatorOptsOrBuilder
in interface RewriterConfigOrBuilder
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;
getOptimizersList
in interface RewriterConfigOrBuilder
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;
getOptimizersCount
in interface RewriterConfigOrBuilder
public 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;
getOptimizers
in interface RewriterConfigOrBuilder
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;
getOptimizersBytes
in interface RewriterConfigOrBuilder
public List<RewriterConfig.CustomGraphOptimizer> getCustomOptimizersList()
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
getCustomOptimizersList
in interface RewriterConfigOrBuilder
public List<? extends RewriterConfig.CustomGraphOptimizerOrBuilder> getCustomOptimizersOrBuilderList()
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
getCustomOptimizersOrBuilderList
in interface RewriterConfigOrBuilder
public int getCustomOptimizersCount()
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
getCustomOptimizersCount
in interface RewriterConfigOrBuilder
public RewriterConfig.CustomGraphOptimizer getCustomOptimizers(int index)
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
getCustomOptimizers
in interface RewriterConfigOrBuilder
public RewriterConfig.CustomGraphOptimizerOrBuilder getCustomOptimizersOrBuilder(int index)
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
getCustomOptimizersOrBuilder
in interface RewriterConfigOrBuilder
public boolean hasInterOptimizerVerifierConfig()
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
hasInterOptimizerVerifierConfig
in interface RewriterConfigOrBuilder
public VerifierConfig getInterOptimizerVerifierConfig()
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
getInterOptimizerVerifierConfig
in interface RewriterConfigOrBuilder
public VerifierConfigOrBuilder getInterOptimizerVerifierConfigOrBuilder()
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
getInterOptimizerVerifierConfigOrBuilder
in interface RewriterConfigOrBuilder
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;
hasPostOptimizationVerifierConfig
in interface RewriterConfigOrBuilder
public VerifierConfig getPostOptimizationVerifierConfig()
VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.
.tensorflow.VerifierConfig post_optimization_verifier_config = 301;
getPostOptimizationVerifierConfig
in interface RewriterConfigOrBuilder
public VerifierConfigOrBuilder getPostOptimizationVerifierConfigOrBuilder()
VerifierConfig specifying the verifiers to be run at the end, after all optimizers have run.
.tensorflow.VerifierConfig post_optimization_verifier_config = 301;
getPostOptimizationVerifierConfigOrBuilder
in interface RewriterConfigOrBuilder
public final boolean isInitialized()
isInitialized
in interface com.google.protobuf.MessageLiteOrBuilder
isInitialized
in class com.google.protobuf.GeneratedMessageV3
public void writeTo(com.google.protobuf.CodedOutputStream output) throws IOException
writeTo
in interface com.google.protobuf.MessageLite
writeTo
in class com.google.protobuf.GeneratedMessageV3
IOException
public int getSerializedSize()
getSerializedSize
in interface com.google.protobuf.MessageLite
getSerializedSize
in class com.google.protobuf.GeneratedMessageV3
public boolean equals(Object obj)
equals
in interface com.google.protobuf.Message
equals
in class com.google.protobuf.AbstractMessage
public int hashCode()
hashCode
in interface com.google.protobuf.Message
hashCode
in class com.google.protobuf.AbstractMessage
public static RewriterConfig parseFrom(ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static RewriterConfig parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static RewriterConfig parseFrom(com.google.protobuf.ByteString data) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static RewriterConfig parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static RewriterConfig parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static RewriterConfig parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static RewriterConfig parseFrom(InputStream input) throws IOException
IOException
public static RewriterConfig parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
IOException
public static RewriterConfig parseDelimitedFrom(InputStream input) throws IOException
IOException
public static RewriterConfig parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
IOException
public static RewriterConfig parseFrom(com.google.protobuf.CodedInputStream input) throws IOException
IOException
public static RewriterConfig parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
IOException
public RewriterConfig.Builder newBuilderForType()
newBuilderForType
in interface com.google.protobuf.Message
newBuilderForType
in interface com.google.protobuf.MessageLite
public static RewriterConfig.Builder newBuilder()
public static RewriterConfig.Builder newBuilder(RewriterConfig prototype)
public RewriterConfig.Builder toBuilder()
toBuilder
in interface com.google.protobuf.Message
toBuilder
in interface com.google.protobuf.MessageLite
protected RewriterConfig.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)
newBuilderForType
in class com.google.protobuf.GeneratedMessageV3
public static RewriterConfig getDefaultInstance()
public static com.google.protobuf.Parser<RewriterConfig> parser()
public com.google.protobuf.Parser<RewriterConfig> getParserForType()
getParserForType
in interface com.google.protobuf.Message
getParserForType
in interface com.google.protobuf.MessageLite
getParserForType
in class com.google.protobuf.GeneratedMessageV3
public RewriterConfig getDefaultInstanceForType()
getDefaultInstanceForType
in interface com.google.protobuf.MessageLiteOrBuilder
getDefaultInstanceForType
in interface com.google.protobuf.MessageOrBuilder
Copyright © 2015–2019. All rights reserved.