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, 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:

  • Field Details

    • CPU_LAYOUT_CONVERSION_FIELD_NUMBER

      public static final int CPU_LAYOUT_CONVERSION_FIELD_NUMBER
      See Also:

    • LAYOUT_OPTIMIZER_FIELD_NUMBER

      public static final int LAYOUT_OPTIMIZER_FIELD_NUMBER
      See Also:

    • CONSTANT_FOLDING_FIELD_NUMBER

      public static final int CONSTANT_FOLDING_FIELD_NUMBER
      See Also:

    • SHAPE_OPTIMIZATION_FIELD_NUMBER

      public static final int SHAPE_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • REMAPPING_FIELD_NUMBER

      public static final int REMAPPING_FIELD_NUMBER
      See Also:

    • COMMON_SUBGRAPH_ELIMINATION_FIELD_NUMBER

      public static final int COMMON_SUBGRAPH_ELIMINATION_FIELD_NUMBER
      See Also:

    • ARITHMETIC_OPTIMIZATION_FIELD_NUMBER

      public static final int ARITHMETIC_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • DEPENDENCY_OPTIMIZATION_FIELD_NUMBER

      public static final int DEPENDENCY_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • LOOP_OPTIMIZATION_FIELD_NUMBER

      public static final int LOOP_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • FUNCTION_OPTIMIZATION_FIELD_NUMBER

      public static final int FUNCTION_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • DEBUG_STRIPPER_FIELD_NUMBER

      public static final int DEBUG_STRIPPER_FIELD_NUMBER
      See Also:

    • DISABLE_MODEL_PRUNING_FIELD_NUMBER

      public static final int DISABLE_MODEL_PRUNING_FIELD_NUMBER
      See Also:

    • SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER

      public static final int SCOPED_ALLOCATOR_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER

      public static final int PIN_TO_HOST_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • IMPLEMENTATION_SELECTOR_FIELD_NUMBER

      public static final int IMPLEMENTATION_SELECTOR_FIELD_NUMBER
      See Also:

    • AUTO_MIXED_PRECISION_FIELD_NUMBER

      public static final int AUTO_MIXED_PRECISION_FIELD_NUMBER
      See Also:

    • AUTO_MIXED_PRECISION_MKL_FIELD_NUMBER

      public static final int AUTO_MIXED_PRECISION_MKL_FIELD_NUMBER
      See Also:

    • AUTO_MIXED_PRECISION_ONEDNN_BFLOAT16_FIELD_NUMBER

      public static final int AUTO_MIXED_PRECISION_ONEDNN_BFLOAT16_FIELD_NUMBER
      See Also:

    • AUTO_MIXED_PRECISION_CPU_FIELD_NUMBER

      public static final int AUTO_MIXED_PRECISION_CPU_FIELD_NUMBER
      See Also:

    • DISABLE_META_OPTIMIZER_FIELD_NUMBER

      public static final int DISABLE_META_OPTIMIZER_FIELD_NUMBER
      See Also:

    • DISABLE_TFG_OPTIMIZER_FIELD_NUMBER

      public static final int DISABLE_TFG_OPTIMIZER_FIELD_NUMBER
      See Also:

    • USE_PLUGIN_OPTIMIZERS_FIELD_NUMBER

      public static final int USE_PLUGIN_OPTIMIZERS_FIELD_NUMBER
      See Also:

    • EXPERIMENTAL_CONDITIONAL_CODE_MOTION_FIELD_NUMBER

      public static final int EXPERIMENTAL_CONDITIONAL_CODE_MOTION_FIELD_NUMBER
      See Also:

    • META_OPTIMIZER_ITERATIONS_FIELD_NUMBER

      public static final int META_OPTIMIZER_ITERATIONS_FIELD_NUMBER
      See Also:

    • MIN_GRAPH_NODES_FIELD_NUMBER

      public static final int MIN_GRAPH_NODES_FIELD_NUMBER
      See Also:

    • EXPERIMENTAL_DISABLE_COMPRESSED_TENSOR_OPTIMIZATION_FIELD_NUMBER

      public static final int EXPERIMENTAL_DISABLE_COMPRESSED_TENSOR_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • EXPERIMENTAL_DISABLE_FOLDING_QUANTIZATION_EMULATION_FIELD_NUMBER

      public static final int EXPERIMENTAL_DISABLE_FOLDING_QUANTIZATION_EMULATION_FIELD_NUMBER
      See Also:

    • MEMORY_OPTIMIZATION_FIELD_NUMBER

      public static final int MEMORY_OPTIMIZATION_FIELD_NUMBER
      See Also:

    • MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER

      public static final int MEMORY_OPTIMIZER_TARGET_NODE_NAME_SCOPE_FIELD_NUMBER
      See Also:

    • META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER

      public static final int META_OPTIMIZER_TIMEOUT_MS_FIELD_NUMBER
      See Also:

    • AUTO_PARALLEL_FIELD_NUMBER

      public static final int AUTO_PARALLEL_FIELD_NUMBER
      See Also:

    • FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER

      public static final int FAIL_ON_OPTIMIZER_ERRORS_FIELD_NUMBER
      See Also:

    • SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER

      public static final int SCOPED_ALLOCATOR_OPTS_FIELD_NUMBER
      See Also:

    • OPTIMIZERS_FIELD_NUMBER

      public static final int OPTIMIZERS_FIELD_NUMBER
      See Also:

    • CUSTOM_OPTIMIZERS_FIELD_NUMBER

      public static final int CUSTOM_OPTIMIZERS_FIELD_NUMBER
      See Also:

    • INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER

      public static final int INTER_OPTIMIZER_VERIFIER_CONFIG_FIELD_NUMBER
      See Also:

    • POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER

      public static final int POST_OPTIMIZATION_VERIFIER_CONFIG_FIELD_NUMBER
      See Also:

  • Method Details

    • newInstance

      protected Object newInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
      Overrides:
      newInstance in class com.google.protobuf.GeneratedMessageV3

    • getDescriptor

      public static final com.google.protobuf.Descriptors.Descriptor getDescriptor()

    • internalGetFieldAccessorTable

      protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
      Specified by:
      internalGetFieldAccessorTable in class com.google.protobuf.GeneratedMessageV3

    • getCpuLayoutConversionValue

      public int getCpuLayoutConversionValue()
       CPU Conversion settings between NHCW and NCHW.
      .tensorflow.RewriterConfig.CpuLayout cpu_layout_conversion = 50;
      Specified by:
      getCpuLayoutConversionValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for cpuLayoutConversion.

    • getCpuLayoutConversion

      public RewriterConfig.CpuLayout getCpuLayoutConversion()
       CPU Conversion settings between NHCW and NCHW.
      .tensorflow.RewriterConfig.CpuLayout cpu_layout_conversion = 50;
      Specified by:
      getCpuLayoutConversion in interface RewriterConfigOrBuilder
      Returns:
      The cpuLayoutConversion.

    • 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:
      getLayoutOptimizerValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for layoutOptimizer.

    • getLayoutOptimizer

      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;
      Specified by:
      getLayoutOptimizer in interface RewriterConfigOrBuilder
      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:
      getConstantFoldingValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for constantFolding.

    • getConstantFolding

      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;
      Specified by:
      getConstantFolding in interface RewriterConfigOrBuilder
      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:
      getShapeOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for shapeOptimization.

    • getShapeOptimization

      public RewriterConfig.Toggle getShapeOptimization()
       Shape optimizations (default is ON)
 Simplify computations made on shapes.
      .tensorflow.RewriterConfig.Toggle shape_optimization = 13;
      Specified by:
      getShapeOptimization in interface RewriterConfigOrBuilder
      Returns:
      The shapeOptimization.

    • getRemappingValue

      public int getRemappingValue()
       Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
      .tensorflow.RewriterConfig.Toggle remapping = 14;
      Specified by:
      getRemappingValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for remapping.

    • getRemapping

      public RewriterConfig.Toggle getRemapping()
       Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
      .tensorflow.RewriterConfig.Toggle remapping = 14;
      Specified by:
      getRemapping in interface RewriterConfigOrBuilder
      Returns:
      The remapping.

    • getCommonSubgraphEliminationValue

      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;
      Specified by:
      getCommonSubgraphEliminationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for commonSubgraphElimination.

    • getCommonSubgraphElimination

      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;
      Specified by:
      getCommonSubgraphElimination in interface RewriterConfigOrBuilder
      Returns:
      The commonSubgraphElimination.

    • 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:
      getArithmeticOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for arithmeticOptimization.

    • getArithmeticOptimization

      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;
      Specified by:
      getArithmeticOptimization in interface RewriterConfigOrBuilder
      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:
      getDependencyOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for dependencyOptimization.

    • getDependencyOptimization

      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;
      Specified by:
      getDependencyOptimization in interface RewriterConfigOrBuilder
      Returns:
      The dependencyOptimization.

    • getLoopOptimizationValue

      public int getLoopOptimizationValue()
       Loop optimizations (default is ON).
      .tensorflow.RewriterConfig.Toggle loop_optimization = 9;
      Specified by:
      getLoopOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for loopOptimization.

    • getLoopOptimization

      public RewriterConfig.Toggle getLoopOptimization()
       Loop optimizations (default is ON).
      .tensorflow.RewriterConfig.Toggle loop_optimization = 9;
      Specified by:
      getLoopOptimization in interface RewriterConfigOrBuilder
      Returns:
      The loopOptimization.

    • getFunctionOptimizationValue

      public int getFunctionOptimizationValue()
       Function optimizations (default is ON).
      .tensorflow.RewriterConfig.Toggle function_optimization = 10;
      Specified by:
      getFunctionOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for functionOptimization.

    • getFunctionOptimization

      public RewriterConfig.Toggle getFunctionOptimization()
       Function optimizations (default is ON).
      .tensorflow.RewriterConfig.Toggle function_optimization = 10;
      Specified by:
      getFunctionOptimization in interface RewriterConfigOrBuilder
      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:
      getDebugStripperValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for debugStripper.

    • getDebugStripper

      public RewriterConfig.Toggle getDebugStripper()
       Strips debug-related nodes from the graph (off by default).
      .tensorflow.RewriterConfig.Toggle debug_stripper = 11;
      Specified by:
      getDebugStripper in interface RewriterConfigOrBuilder
      Returns:
      The debugStripper.

    • getDisableModelPruning

      public boolean getDisableModelPruning()
       If true, don't remove unnecessary ops from the graph
      bool disable_model_pruning = 2;
      Specified by:
      getDisableModelPruning in interface RewriterConfigOrBuilder
      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:
      getScopedAllocatorOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for scopedAllocatorOptimization.

    • getScopedAllocatorOptimization

      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;
      Specified by:
      getScopedAllocatorOptimization in interface RewriterConfigOrBuilder
      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:
      getPinToHostOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for pinToHostOptimization.

    • getPinToHostOptimization

      public RewriterConfig.Toggle getPinToHostOptimization()
       Force small ops onto the CPU (default is OFF).
      .tensorflow.RewriterConfig.Toggle pin_to_host_optimization = 18;
      Specified by:
      getPinToHostOptimization in interface RewriterConfigOrBuilder
      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:
      getImplementationSelectorValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for implementationSelector.

    • getImplementationSelector

      public RewriterConfig.Toggle getImplementationSelector()
       Enable the swap of kernel implementations based on the device placement
 (default is ON).
      .tensorflow.RewriterConfig.Toggle implementation_selector = 22;
      Specified by:
      getImplementationSelector in interface RewriterConfigOrBuilder
      Returns:
      The implementationSelector.

    • getAutoMixedPrecisionValue

      public int getAutoMixedPrecisionValue()
       Optimize data types for CUDA/oneDNN (default is OFF).
 This will try to use float16 on GPU/CPU 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:
      getAutoMixedPrecisionValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for autoMixedPrecision.

    • getAutoMixedPrecision

      public RewriterConfig.Toggle getAutoMixedPrecision()
       Optimize data types for CUDA/oneDNN (default is OFF).
 This will try to use float16 on GPU/CPU 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:
      getAutoMixedPrecision in interface RewriterConfigOrBuilder
      Returns:
      The autoMixedPrecision.

    • getAutoMixedPrecisionMklValue

      public int getAutoMixedPrecisionMklValue()
       Optimize data types for oneDNN (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.
 Note: this is deprecated.
 It is replaced by auto_mixed_precision_onednn_bfloat16
      .tensorflow.RewriterConfig.Toggle auto_mixed_precision_mkl = 25;
      Specified by:
      getAutoMixedPrecisionMklValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for autoMixedPrecisionMkl.

    • getAutoMixedPrecisionMkl

      public RewriterConfig.Toggle getAutoMixedPrecisionMkl()
       Optimize data types for oneDNN (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.
 Note: this is deprecated.
 It is replaced by auto_mixed_precision_onednn_bfloat16
      .tensorflow.RewriterConfig.Toggle auto_mixed_precision_mkl = 25;
      Specified by:
      getAutoMixedPrecisionMkl in interface RewriterConfigOrBuilder
      Returns:
      The autoMixedPrecisionMkl.

    • getAutoMixedPrecisionOnednnBfloat16Value

      public int getAutoMixedPrecisionOnednnBfloat16Value()
       Optimize data types for oneDNN (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.
 Note: this is equivalent to the deprecated option auto_mixed_precision_mkl
      .tensorflow.RewriterConfig.Toggle auto_mixed_precision_onednn_bfloat16 = 31;
      Specified by:
      getAutoMixedPrecisionOnednnBfloat16Value in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for autoMixedPrecisionOnednnBfloat16.

    • getAutoMixedPrecisionOnednnBfloat16

      public RewriterConfig.Toggle getAutoMixedPrecisionOnednnBfloat16()
       Optimize data types for oneDNN (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.
 Note: this is equivalent to the deprecated option auto_mixed_precision_mkl
      .tensorflow.RewriterConfig.Toggle auto_mixed_precision_onednn_bfloat16 = 31;
      Specified by:
      getAutoMixedPrecisionOnednnBfloat16 in interface RewriterConfigOrBuilder
      Returns:
      The autoMixedPrecisionOnednnBfloat16.

    • getAutoMixedPrecisionCpuValue

      public int getAutoMixedPrecisionCpuValue()
       Emulate a model using data type float16 on CPU (default is OFF).
 This will try to emulate the float16 inputs and outputs of an operator
 on CPU to have better correlation with float16 on GPU; however the
 computation in the operator is based on float32.
 Note that this can change the numerical stability of the graph.
      .tensorflow.RewriterConfig.Toggle auto_mixed_precision_cpu = 29;
      Specified by:
      getAutoMixedPrecisionCpuValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for autoMixedPrecisionCpu.

    • getAutoMixedPrecisionCpu

      public RewriterConfig.Toggle getAutoMixedPrecisionCpu()
       Emulate a model using data type float16 on CPU (default is OFF).
 This will try to emulate the float16 inputs and outputs of an operator
 on CPU to have better correlation with float16 on GPU; however the
 computation in the operator is based on float32.
 Note that this can change the numerical stability of the graph.
      .tensorflow.RewriterConfig.Toggle auto_mixed_precision_cpu = 29;
      Specified by:
      getAutoMixedPrecisionCpu in interface RewriterConfigOrBuilder
      Returns:
      The autoMixedPrecisionCpu.

    • getDisableMetaOptimizer

      public boolean getDisableMetaOptimizer()
       Disable the entire meta optimizer (off by default).
      bool disable_meta_optimizer = 19;
      Specified by:
      getDisableMetaOptimizer in interface RewriterConfigOrBuilder
      Returns:
      The disableMetaOptimizer.

    • getDisableTfgOptimizer

      public boolean getDisableTfgOptimizer()
       Disable the TFG optimizer (off by default).
      bool disable_tfg_optimizer = 32;
      Specified by:
      getDisableTfgOptimizer in interface RewriterConfigOrBuilder
      Returns:
      The disableTfgOptimizer.

    • getUsePluginOptimizersValue

      public int getUsePluginOptimizersValue()
       Optimizers registered by plugin (default is ON)
      .tensorflow.RewriterConfig.Toggle use_plugin_optimizers = 28;
      Specified by:
      getUsePluginOptimizersValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for usePluginOptimizers.

    • getUsePluginOptimizers

      public RewriterConfig.Toggle getUsePluginOptimizers()
       Optimizers registered by plugin (default is ON)
      .tensorflow.RewriterConfig.Toggle use_plugin_optimizers = 28;
      Specified by:
      getUsePluginOptimizers in interface RewriterConfigOrBuilder
      Returns:
      The usePluginOptimizers.

    • getExperimentalConditionalCodeMotionValue

      public int getExperimentalConditionalCodeMotionValue()
       Conditional code motion (default is ON).
      .tensorflow.RewriterConfig.Toggle experimental_conditional_code_motion = 30;
      Specified by:
      getExperimentalConditionalCodeMotionValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for experimentalConditionalCodeMotion.

    • getExperimentalConditionalCodeMotion

      public RewriterConfig.Toggle getExperimentalConditionalCodeMotion()
       Conditional code motion (default is ON).
      .tensorflow.RewriterConfig.Toggle experimental_conditional_code_motion = 30;
      Specified by:
      getExperimentalConditionalCodeMotion in interface RewriterConfigOrBuilder
      Returns:
      The experimentalConditionalCodeMotion.

    • 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:
      getMetaOptimizerIterationsValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for metaOptimizerIterations.

    • getMetaOptimizerIterations

      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;
      Specified by:
      getMetaOptimizerIterations in interface RewriterConfigOrBuilder
      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:
      getMinGraphNodes in interface RewriterConfigOrBuilder
      Returns:
      The minGraphNodes.

    • getExperimentalDisableCompressedTensorOptimization

      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;
      Specified by:
      getExperimentalDisableCompressedTensorOptimization in interface RewriterConfigOrBuilder
      Returns:
      The experimentalDisableCompressedTensorOptimization.

    • getExperimentalDisableFoldingQuantizationEmulation

      public boolean getExperimentalDisableFoldingQuantizationEmulation()
       Disable folding quantization emulation ops such as FakeQuantWithMinMax* and
 QuantizeAndDequantize*. Some compilers (e.g. the TF-to-tflite converter)
 have to extract quantization configs (e.g. min/max range, number of bits,
 and per-channel) from the quantization emulation ops. Note that this flag
 is experimental and may be removed in the future. See b/174138564 for more
 details.
      bool experimental_disable_folding_quantization_emulation = 27;
      Specified by:
      getExperimentalDisableFoldingQuantizationEmulation in interface RewriterConfigOrBuilder
      Returns:
      The experimentalDisableFoldingQuantizationEmulation.

    • 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:
      getMemoryOptimizationValue in interface RewriterConfigOrBuilder
      Returns:
      The enum numeric value on the wire for memoryOptimization.

    • getMemoryOptimization

      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;
      Specified by:
      getMemoryOptimization in interface RewriterConfigOrBuilder
      Returns:
      The memoryOptimization.

    • getMemoryOptimizerTargetNodeNameScope

      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;
      Specified by:
      getMemoryOptimizerTargetNodeNameScope in interface RewriterConfigOrBuilder
      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:
      getMemoryOptimizerTargetNodeNameScopeBytes in interface RewriterConfigOrBuilder
      Returns:
      The bytes for memoryOptimizerTargetNodeNameScope.

    • getMetaOptimizerTimeoutMs

      public long getMetaOptimizerTimeoutMs()
       Maximum number of milliseconds to spend optimizing a single graph before
 timing out. If less than or equal to 0 (default value) the optimizer will
 never time out.
      int64 meta_optimizer_timeout_ms = 20;
      Specified by:
      getMetaOptimizerTimeoutMs in interface RewriterConfigOrBuilder
      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:
      hasAutoParallel in interface RewriterConfigOrBuilder
      Returns:
      Whether the autoParallel field is set.

    • getAutoParallel

      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;
      Specified by:
      getAutoParallel in interface RewriterConfigOrBuilder
      Returns:
      The autoParallel.

    • getAutoParallelOrBuilder

      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;
      Specified by:
      getAutoParallelOrBuilder in interface RewriterConfigOrBuilder

    • 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:
      getFailOnOptimizerErrors in interface RewriterConfigOrBuilder
      Returns:
      The failOnOptimizerErrors.

    • hasScopedAllocatorOpts

      public boolean hasScopedAllocatorOpts()
      .tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
      Specified by:
      hasScopedAllocatorOpts in interface RewriterConfigOrBuilder
      Returns:
      Whether the scopedAllocatorOpts field is set.

    • getScopedAllocatorOpts

      public ScopedAllocatorOptions getScopedAllocatorOpts()
      .tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
      Specified by:
      getScopedAllocatorOpts in interface RewriterConfigOrBuilder
      Returns:
      The scopedAllocatorOpts.

    • getScopedAllocatorOptsOrBuilder

      public ScopedAllocatorOptionsOrBuilder getScopedAllocatorOptsOrBuilder()
      .tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;
      Specified by:
      getScopedAllocatorOptsOrBuilder in interface RewriterConfigOrBuilder

    • 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:
      getOptimizersList in interface RewriterConfigOrBuilder
      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:
      getOptimizersCount in interface RewriterConfigOrBuilder
      Returns:
      The count of optimizers.

    • getOptimizers

      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;
      Specified by:
      getOptimizers in interface RewriterConfigOrBuilder
      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:
      getOptimizersBytes in interface RewriterConfigOrBuilder
      Parameters:
      index - The index of the value to return.
      Returns:
      The bytes of the optimizers at the given index.

    • getCustomOptimizersList

      public List<RewriterConfig.CustomGraphOptimizer> getCustomOptimizersList()
       list of CustomGraphOptimizers to apply.
      repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
      Specified by:
      getCustomOptimizersList in interface RewriterConfigOrBuilder

    • getCustomOptimizersOrBuilderList

      public List<? extends RewriterConfig.CustomGraphOptimizerOrBuilder> getCustomOptimizersOrBuilderList()
       list of CustomGraphOptimizers to apply.
      repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
      Specified by:
      getCustomOptimizersOrBuilderList in interface RewriterConfigOrBuilder

    • getCustomOptimizersCount

      public int getCustomOptimizersCount()
       list of CustomGraphOptimizers to apply.
      repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
      Specified by:
      getCustomOptimizersCount in interface RewriterConfigOrBuilder

    • getCustomOptimizers

      public RewriterConfig.CustomGraphOptimizer getCustomOptimizers(int index)
       list of CustomGraphOptimizers to apply.
      repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
      Specified by:
      getCustomOptimizers in interface RewriterConfigOrBuilder

    • getCustomOptimizersOrBuilder

      public RewriterConfig.CustomGraphOptimizerOrBuilder getCustomOptimizersOrBuilder(int index)
       list of CustomGraphOptimizers to apply.
      repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;
      Specified by:
      getCustomOptimizersOrBuilder in interface RewriterConfigOrBuilder

    • hasInterOptimizerVerifierConfig

      public boolean hasInterOptimizerVerifierConfig()
       VerifierConfig specifying the verifiers to be run after every optimizer.
      .tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
      Specified by:
      hasInterOptimizerVerifierConfig in interface RewriterConfigOrBuilder
      Returns:
      Whether the interOptimizerVerifierConfig field is set.

    • getInterOptimizerVerifierConfig

      public VerifierConfig getInterOptimizerVerifierConfig()
       VerifierConfig specifying the verifiers to be run after every optimizer.
      .tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
      Specified by:
      getInterOptimizerVerifierConfig in interface RewriterConfigOrBuilder
      Returns:
      The interOptimizerVerifierConfig.

    • getInterOptimizerVerifierConfigOrBuilder

      public VerifierConfigOrBuilder getInterOptimizerVerifierConfigOrBuilder()
       VerifierConfig specifying the verifiers to be run after every optimizer.
      .tensorflow.VerifierConfig inter_optimizer_verifier_config = 300;
      Specified by:
      getInterOptimizerVerifierConfigOrBuilder in interface RewriterConfigOrBuilder

    • 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:
      hasPostOptimizationVerifierConfig in interface RewriterConfigOrBuilder
      Returns:
      Whether the postOptimizationVerifierConfig field is set.

    • getPostOptimizationVerifierConfig

      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;
      Specified by:
      getPostOptimizationVerifierConfig in interface RewriterConfigOrBuilder
      Returns:
      The postOptimizationVerifierConfig.

    • getPostOptimizationVerifierConfigOrBuilder

      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;
      Specified by:
      getPostOptimizationVerifierConfigOrBuilder in interface RewriterConfigOrBuilder

    • isInitialized

      public final boolean isInitialized()
      Specified by:
      isInitialized in interface com.google.protobuf.MessageLiteOrBuilder
      Overrides:
      isInitialized in class com.google.protobuf.GeneratedMessageV3

    • writeTo

      public void writeTo(com.google.protobuf.CodedOutputStream output) throws IOException
      Specified by:
      writeTo in interface com.google.protobuf.MessageLite
      Overrides:
      writeTo in class com.google.protobuf.GeneratedMessageV3
      Throws:
      IOException

    • getSerializedSize

      public int getSerializedSize()
      Specified by:
      getSerializedSize in interface com.google.protobuf.MessageLite
      Overrides:
      getSerializedSize in class com.google.protobuf.GeneratedMessageV3

    • equals

      public boolean equals(Object obj)
      Specified by:
      equals in interface com.google.protobuf.Message
      Overrides:
      equals in class com.google.protobuf.AbstractMessage

    • hashCode

      public int hashCode()
      Specified by:
      hashCode in interface com.google.protobuf.Message
      Overrides:
      hashCode in class com.google.protobuf.AbstractMessage

    • parseFrom

      public static RewriterConfig parseFrom(ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static RewriterConfig parseFrom(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

      public static RewriterConfig parseFrom(InputStream input) throws IOException
      Throws:
      IOException

    • parseFrom

      public static RewriterConfig parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Throws:
      IOException

    • parseDelimitedFrom

      public static RewriterConfig parseDelimitedFrom(InputStream input) throws IOException
      Throws:
      IOException

    • parseDelimitedFrom

      public static RewriterConfig parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Throws:
      IOException

    • parseFrom

      public static RewriterConfig parseFrom(com.google.protobuf.CodedInputStream input) throws IOException
      Throws:
      IOException

    • parseFrom

      public static RewriterConfig parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Throws:
      IOException

    • newBuilderForType

      public RewriterConfig.Builder newBuilderForType()
      Specified by:
      newBuilderForType in interface com.google.protobuf.Message
      Specified by:
      newBuilderForType in interface com.google.protobuf.MessageLite

    • newBuilder

      public static RewriterConfig.Builder newBuilder()

    • newBuilder

      public static RewriterConfig.Builder newBuilder(RewriterConfig prototype)

    • toBuilder

      public RewriterConfig.Builder toBuilder()
      Specified by:
      toBuilder in interface com.google.protobuf.Message
      Specified by:
      toBuilder in interface com.google.protobuf.MessageLite

    • newBuilderForType

      protected RewriterConfig.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)
      Specified by:
      newBuilderForType in class com.google.protobuf.GeneratedMessageV3

    • getDefaultInstance

      public static RewriterConfig getDefaultInstance()

    • parser

      public static com.google.protobuf.Parser<RewriterConfig> parser()

    • getParserForType

      public com.google.protobuf.Parser<RewriterConfig> getParserForType()
      Specified by:
      getParserForType in interface com.google.protobuf.Message
      Specified by:
      getParserForType in interface com.google.protobuf.MessageLite
      Overrides:
      getParserForType in class com.google.protobuf.GeneratedMessageV3

    • getDefaultInstanceForType

      public RewriterConfig getDefaultInstanceForType()
      Specified by:
      getDefaultInstanceForType in interface com.google.protobuf.MessageLiteOrBuilder
      Specified by:
      getDefaultInstanceForType in interface com.google.protobuf.MessageOrBuilder