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 parametersProtobuf 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
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. |
boolean |
hasScopedAllocatorOpts()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
|
int |
hashCode()
|
final boolean | |
static RewriterConfig.Builder | |
static RewriterConfig.Builder |
newBuilder(RewriterConfig prototype)
|
RewriterConfig.Builder | |
static RewriterConfig |
parseDelimitedFrom(InputStream input)
|
static RewriterConfig |
parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static RewriterConfig |
parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static RewriterConfig |
parseFrom(com.google.protobuf.CodedInputStream input)
|
static RewriterConfig |
parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static RewriterConfig |
parseFrom(ByteBuffer data)
|
static RewriterConfig |
parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static RewriterConfig |
parseFrom(com.google.protobuf.ByteString data)
|
static RewriterConfig |
parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static RewriterConfig |
parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static
|
parser()
|
RewriterConfig.Builder | |
void |
writeTo(com.google.protobuf.CodedOutputStream output)
|
Inherited Methods
Constants
public static final int ARITHMETIC_OPTIMIZATION_FIELD_NUMBER
Constant Value:
7
public static final int AUTO_MIXED_PRECISION_FIELD_NUMBER
Constant Value:
23
public static final int AUTO_MIXED_PRECISION_MKL_FIELD_NUMBER
Constant Value:
25
public static final int AUTO_PARALLEL_FIELD_NUMBER
Constant Value:
5
public static final int COMMON_SUBGRAPH_ELIMINATION_FIELD_NUMBER
Constant Value:
24
public static final int CONSTANT_FOLDING_FIELD_NUMBER
Constant Value:
3
public static final int CPU_LAYOUT_CONVERSION_FIELD_NUMBER
Constant Value:
50
public static final int CUSTOM_OPTIMIZERS_FIELD_NUMBER
Constant Value:
200
public static final int DEBUG_STRIPPER_FIELD_NUMBER
Constant Value:
11
public static final int DEPENDENCY_OPTIMIZATION_FIELD_NUMBER
Constant Value:
8
public static final int DISABLE_META_OPTIMIZER_FIELD_NUMBER
Constant Value:
19
public static final int DISABLE_MODEL_PRUNING_FIELD_NUMBER
Constant Value:
2
public static final int EXPERIMENTAL_DISABLE_COMPRESSED_TENSOR_OPTIMIZATION_FIELD_NUMBER
Constant Value:
26
public static final int FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER
Constant Value:
21
public static final int FUNCTION_OPTIMIZATION_FIELD_NUMBER
Constant Value:
10
public static final int IMPLEMENTATION_SELECTOR_FIELD_NUMBER
Constant Value:
22
public static final int INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER
Constant Value:
300
public static final int LAYOUT_OPTIMIZER_FIELD_NUMBER
Constant Value:
1
public static final int LOOP_OPTIMIZATION_FIELD_NUMBER
Constant Value:
9
public static final int MEMORY_OPTIMIZATION_FIELD_NUMBER
Constant Value:
4
public static final int MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER
Constant Value:
6
public static final int META_OPTIMIZER_ITERATIONS_FIELD_NUMBER
Constant Value:
12
public static final int META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER
Constant Value:
20
public static final int MIN_GRAPH_NODES_FIELD_NUMBER
Constant Value:
17
public static final int OPTIMIZERS_FIELD_NUMBER
Constant Value:
100
public static final int PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER
Constant Value:
18
public static final int POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER
Constant Value:
301
public static final int REMAPPING_FIELD_NUMBER
Constant Value:
14
public static final int SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER
Constant Value:
15
public static final int SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER
Constant Value:
16
public static final int SHAPE_OPTIMIZATION_FIELD_NUMBER
Constant Value:
13
Public Methods
public boolean equals (Object obj)
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;
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;
public RewriterConfig.Toggle getAutoMixedPrecision ()
Optimize data types for CUDA (default is OFF). 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;
public RewriterConfig.Toggle getAutoMixedPrecisionMkl ()
Optimize data types for MKL (default is OFF). This will try to use bfloat16 on CPUs, which is faster. Note that this can change the numerical stability of the graph.
.tensorflow.RewriterConfig.Toggle auto_mixed_precision_mkl = 25;
public int getAutoMixedPrecisionMklValue ()
Optimize data types for MKL (default is OFF). This will try to use bfloat16 on CPUs, which is faster. Note that this can change the numerical stability of the graph.
.tensorflow.RewriterConfig.Toggle auto_mixed_precision_mkl = 25;
public int getAutoMixedPrecisionValue ()
Optimize data types for CUDA (default is OFF). 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;
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;
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;
public RewriterConfig.Toggle getCommonSubgraphElimination ()
Common subgraph elimination (default is ON) e.g. Simplify arithmetic ops; merge ops with same value (like constants).
.tensorflow.RewriterConfig.Toggle common_subgraph_elimination = 24;
public int getCommonSubgraphEliminationValue ()
Common subgraph elimination (default is ON) e.g. Simplify arithmetic ops; merge ops with same value (like constants).
.tensorflow.RewriterConfig.Toggle common_subgraph_elimination = 24;
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;
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;
public RewriterConfig.CpuLayout getCpuLayoutConversion ()
CPU Conversion settings between NHCW and NCHW.
.tensorflow.RewriterConfig.CpuLayout cpu_layout_conversion = 50;
public int getCpuLayoutConversionValue ()
CPU Conversion settings between NHCW and NCHW.
.tensorflow.RewriterConfig.CpuLayout cpu_layout_conversion = 50;
public RewriterConfig.CustomGraphOptimizer getCustomOptimizers (int index)
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
public int getCustomOptimizersCount ()
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
public List<RewriterConfig.CustomGraphOptimizer> getCustomOptimizersList ()
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
public RewriterConfig.CustomGraphOptimizerOrBuilder getCustomOptimizersOrBuilder (int index)
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
public List<? extends RewriterConfig.CustomGraphOptimizerOrBuilder> getCustomOptimizersOrBuilderList ()
list of CustomGraphOptimizers to apply.
repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
public RewriterConfig.Toggle getDebugStripper ()
Strips debug-related nodes from the graph (off by default).
.tensorflow.RewriterConfig.Toggle debug_stripper = 11;
public int getDebugStripperValue ()
Strips debug-related nodes from the graph (off by default).
.tensorflow.RewriterConfig.Toggle debug_stripper = 11;
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;
public int getDependencyOptimizationValue ()
Control dependency optimizations (default is ON). Remove redundant control dependencies, which may enable other optimization.
.tensorflow.RewriterConfig.Toggle dependency_optimization = 8;
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor ()
public boolean getDisableMetaOptimizer ()
Disable the entire meta optimizer (off by default).
bool disable_meta_optimizer = 19;
public boolean getDisableModelPruning ()
If true, don't remove unnecessary ops from the graph
bool disable_model_pruning = 2;
public boolean getExperimentalDisableCompressedTensorOptimization ()
Disable optimizations that assume compressed tensors. Note that this flag is experimental and may be removed in the future.
bool experimental_disable_compressed_tensor_optimization = 26;
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;
public RewriterConfig.Toggle getFunctionOptimization ()
Function optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle function_optimization = 10;
public int getFunctionOptimizationValue ()
Function optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle function_optimization = 10;
public RewriterConfig.Toggle getImplementationSelector ()
Enable the swap of kernel implementations based on the device placement (default is ON).
.tensorflow.RewriterConfig.Toggle implementation_selector = 22;
public int getImplementationSelectorValue ()
Enable the swap of kernel implementations based on the device placement (default is ON).
.tensorflow.RewriterConfig.Toggle implementation_selector = 22;
public VerifierConfig getInterOptimizerVerifierConfig ()
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
public VerifierConfigOrBuilder getInterOptimizerVerifierConfigOrBuilder ()
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
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;
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;
public RewriterConfig.Toggle getLoopOptimization ()
Loop optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle loop_optimization = 9;
public int getLoopOptimizationValue ()
Loop optimizations (default is ON).
.tensorflow.RewriterConfig.Toggle loop_optimization = 9;
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;
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;
public 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;
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;
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;
public int getMetaOptimizerIterationsValue ()
Controls how many times we run the optimizers in meta optimizer (default is once).
.tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;
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;
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;
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;
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;
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;
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;
public
getParserForType
()
public RewriterConfig.Toggle getPinToHostOptimization ()
Force small ops onto the CPU (default is OFF).
.tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;
public int getPinToHostOptimizationValue ()
Force small ops onto the CPU (default is OFF).
.tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;
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;
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;
public RewriterConfig.Toggle getRemapping ()
Remapping (default is ON) Remap subgraphs onto more efficient implementations.
.tensorflow.RewriterConfig.Toggle remapping = 14;
public int getRemappingValue ()
Remapping (default is ON) Remap subgraphs onto more efficient implementations.
.tensorflow.RewriterConfig.Toggle remapping = 14;
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;
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;
public ScopedAllocatorOptions getScopedAllocatorOpts ()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
public ScopedAllocatorOptionsOrBuilder getScopedAllocatorOptsOrBuilder ()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
public int getSerializedSize ()
public RewriterConfig.Toggle getShapeOptimization ()
Shape optimizations (default is ON) Simplify computations made on shapes.
.tensorflow.RewriterConfig.Toggle shape_optimization = 13;
public int getShapeOptimizationValue ()
Shape optimizations (default is ON) Simplify computations made on shapes.
.tensorflow.RewriterConfig.Toggle shape_optimization = 13;
public final com.google.protobuf.UnknownFieldSet getUnknownFields ()
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;
public boolean hasInterOptimizerVerifierConfig ()
VerifierConfig specifying the verifiers to be run after every optimizer.
.tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
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;
public boolean hasScopedAllocatorOpts ()
.tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
public int hashCode ()
public final boolean isInitialized ()
public static RewriterConfig parseDelimitedFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public static RewriterConfig parseFrom (ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
InvalidProtocolBufferException |
---|
public static RewriterConfig parseFrom (com.google.protobuf.CodedInputStream input)
Throws
IOException |
---|
public static RewriterConfig parseFrom (byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
InvalidProtocolBufferException |
---|
public static RewriterConfig parseFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public static RewriterConfig parseFrom (com.google.protobuf.ByteString data)
Throws
InvalidProtocolBufferException |
---|
public static RewriterConfig parseFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public static RewriterConfig parseFrom (com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
InvalidProtocolBufferException |
---|
public
static
parser
()
public void writeTo (com.google.protobuf.CodedOutputStream output)
Throws
IOException |
---|