Package org.tensorflow.framework

Class RewriterConfig.Builder

  • All Implemented Interfaces:
    Cloneable, org.nd4j.shade.protobuf.Message.Builder, org.nd4j.shade.protobuf.MessageLite.Builder, org.nd4j.shade.protobuf.MessageLiteOrBuilder, org.nd4j.shade.protobuf.MessageOrBuilder, RewriterConfigOrBuilder
    Enclosing class:
    RewriterConfig
    public static final class RewriterConfig.Builder
extends org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>
implements RewriterConfigOrBuilder
     Graph rewriting is experimental and subject to change, not covered by any
 API stability guarantees.
    Protobuf type tensorflow.RewriterConfig

    • Method Detail

      • getDescriptor

        public static final org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptor()

      • internalGetFieldAccessorTable

        protected org.nd4j.shade.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
        Specified by:
        internalGetFieldAccessorTable in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • clear

        public RewriterConfig.Builder clear()
        Specified by:
        clear in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        clear in interface org.nd4j.shade.protobuf.MessageLite.Builder
        Overrides:
        clear in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • getDescriptorForType

        public org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptorForType()
        Specified by:
        getDescriptorForType in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        getDescriptorForType in interface org.nd4j.shade.protobuf.MessageOrBuilder
        Overrides:
        getDescriptorForType in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • getDefaultInstanceForType

        public RewriterConfig getDefaultInstanceForType()
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageOrBuilder

      • build

        public RewriterConfig build()
        Specified by:
        build in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        build in interface org.nd4j.shade.protobuf.MessageLite.Builder

      • buildPartial

        public RewriterConfig buildPartial()
        Specified by:
        buildPartial in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        buildPartial in interface org.nd4j.shade.protobuf.MessageLite.Builder

      • clone

        public RewriterConfig.Builder clone()
        Specified by:
        clone in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        clone in interface org.nd4j.shade.protobuf.MessageLite.Builder
        Overrides:
        clone in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • setField

        public RewriterConfig.Builder setField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field,
                                       Object value)
        Specified by:
        setField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        setField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • clearField

        public RewriterConfig.Builder clearField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field)
        Specified by:
        clearField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        clearField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • clearOneof

        public RewriterConfig.Builder clearOneof​(org.nd4j.shade.protobuf.Descriptors.OneofDescriptor oneof)
        Specified by:
        clearOneof in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        clearOneof in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • setRepeatedField

        public RewriterConfig.Builder setRepeatedField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field,
                                               int index,
                                               Object value)
        Specified by:
        setRepeatedField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        setRepeatedField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • addRepeatedField

        public RewriterConfig.Builder addRepeatedField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field,
                                               Object value)
        Specified by:
        addRepeatedField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        addRepeatedField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • mergeFrom

        public RewriterConfig.Builder mergeFrom​(org.nd4j.shade.protobuf.Message other)
        Specified by:
        mergeFrom in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        mergeFrom in class org.nd4j.shade.protobuf.AbstractMessage.Builder<RewriterConfig.Builder>

      • isInitialized

        public final boolean isInitialized()
        Specified by:
        isInitialized in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Overrides:
        isInitialized in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • mergeFrom

        public RewriterConfig.Builder mergeFrom​(org.nd4j.shade.protobuf.CodedInputStream input,
                                        org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
                                 throws IOException
        Specified by:
        mergeFrom in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        mergeFrom in interface org.nd4j.shade.protobuf.MessageLite.Builder
        Overrides:
        mergeFrom in class org.nd4j.shade.protobuf.AbstractMessage.Builder<RewriterConfig.Builder>
        Throws:
        IOException

      • 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.

      • setLayoutOptimizerValue

        public RewriterConfig.Builder setLayoutOptimizerValue​(int value)
         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;
        Parameters:
        value - The enum numeric value on the wire for layoutOptimizer to set.
        Returns:
        This builder for chaining.

      • 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.

      • setLayoutOptimizer

        public RewriterConfig.Builder setLayoutOptimizer​(RewriterConfig.Toggle value)
         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;
        Parameters:
        value - The layoutOptimizer to set.
        Returns:
        This builder for chaining.

      • clearLayoutOptimizer

        public RewriterConfig.Builder clearLayoutOptimizer()
         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;
        Returns:
        This builder for chaining.

      • 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.

      • setConstantFoldingValue

        public RewriterConfig.Builder setConstantFoldingValue​(int value)
         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;
        Parameters:
        value - The enum numeric value on the wire for constantFolding to set.
        Returns:
        This builder for chaining.

      • 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.

      • setConstantFolding

        public RewriterConfig.Builder setConstantFolding​(RewriterConfig.Toggle value)
         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;
        Parameters:
        value - The constantFolding to set.
        Returns:
        This builder for chaining.

      • clearConstantFolding

        public RewriterConfig.Builder clearConstantFolding()
         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;
        Returns:
        This builder for chaining.

      • 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.

      • setShapeOptimizationValue

        public RewriterConfig.Builder setShapeOptimizationValue​(int value)
         Shape optimizations (default is ON)
 Simplify computations made on shapes.
        .tensorflow.RewriterConfig.Toggle shape_optimization = 13;
        Parameters:
        value - The enum numeric value on the wire for shapeOptimization to set.
        Returns:
        This builder for chaining.

      • setShapeOptimization

        public RewriterConfig.Builder setShapeOptimization​(RewriterConfig.Toggle value)
         Shape optimizations (default is ON)
 Simplify computations made on shapes.
        .tensorflow.RewriterConfig.Toggle shape_optimization = 13;
        Parameters:
        value - The shapeOptimization to set.
        Returns:
        This builder for chaining.

      • clearShapeOptimization

        public RewriterConfig.Builder clearShapeOptimization()
         Shape optimizations (default is ON)
 Simplify computations made on shapes.
        .tensorflow.RewriterConfig.Toggle shape_optimization = 13;
        Returns:
        This builder for chaining.

      • 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.

      • setRemappingValue

        public RewriterConfig.Builder setRemappingValue​(int value)
         Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
        .tensorflow.RewriterConfig.Toggle remapping = 14;
        Parameters:
        value - The enum numeric value on the wire for remapping to set.
        Returns:
        This builder for chaining.

      • setRemapping

        public RewriterConfig.Builder setRemapping​(RewriterConfig.Toggle value)
         Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
        .tensorflow.RewriterConfig.Toggle remapping = 14;
        Parameters:
        value - The remapping to set.
        Returns:
        This builder for chaining.

      • clearRemapping

        public RewriterConfig.Builder clearRemapping()
         Remapping (default is ON)
 Remap subgraphs onto more efficient implementations.
        .tensorflow.RewriterConfig.Toggle remapping = 14;
        Returns:
        This builder for chaining.

      • 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.

      • setArithmeticOptimizationValue

        public RewriterConfig.Builder setArithmeticOptimizationValue​(int value)
         Arithmetic optimizations (default is ON)
 e.g. Simplify arithmetic ops; merge ops with same value (like constants).
        .tensorflow.RewriterConfig.Toggle arithmetic_optimization = 7;
        Parameters:
        value - The enum numeric value on the wire for arithmeticOptimization to set.
        Returns:
        This builder for chaining.

      • 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.

      • setArithmeticOptimization

        public RewriterConfig.Builder setArithmeticOptimization​(RewriterConfig.Toggle value)
         Arithmetic optimizations (default is ON)
 e.g. Simplify arithmetic ops; merge ops with same value (like constants).
        .tensorflow.RewriterConfig.Toggle arithmetic_optimization = 7;
        Parameters:
        value - The arithmeticOptimization to set.
        Returns:
        This builder for chaining.

      • clearArithmeticOptimization

        public RewriterConfig.Builder clearArithmeticOptimization()
         Arithmetic optimizations (default is ON)
 e.g. Simplify arithmetic ops; merge ops with same value (like constants).
        .tensorflow.RewriterConfig.Toggle arithmetic_optimization = 7;
        Returns:
        This builder for chaining.

      • 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.

      • setDependencyOptimizationValue

        public RewriterConfig.Builder setDependencyOptimizationValue​(int value)
         Control dependency optimizations (default is ON).
 Remove redundant control dependencies, which may enable other optimization.
        .tensorflow.RewriterConfig.Toggle dependency_optimization = 8;
        Parameters:
        value - The enum numeric value on the wire for dependencyOptimization to set.
        Returns:
        This builder for chaining.

      • 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.

      • setDependencyOptimization

        public RewriterConfig.Builder setDependencyOptimization​(RewriterConfig.Toggle value)
         Control dependency optimizations (default is ON).
 Remove redundant control dependencies, which may enable other optimization.
        .tensorflow.RewriterConfig.Toggle dependency_optimization = 8;
        Parameters:
        value - The dependencyOptimization to set.
        Returns:
        This builder for chaining.

      • clearDependencyOptimization

        public RewriterConfig.Builder clearDependencyOptimization()
         Control dependency optimizations (default is ON).
 Remove redundant control dependencies, which may enable other optimization.
        .tensorflow.RewriterConfig.Toggle dependency_optimization = 8;
        Returns:
        This builder for chaining.

      • 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.

      • setLoopOptimizationValue

        public RewriterConfig.Builder setLoopOptimizationValue​(int value)
         Loop optimizations (default is ON).
        .tensorflow.RewriterConfig.Toggle loop_optimization = 9;
        Parameters:
        value - The enum numeric value on the wire for loopOptimization to set.
        Returns:
        This builder for chaining.

      • setLoopOptimization

        public RewriterConfig.Builder setLoopOptimization​(RewriterConfig.Toggle value)
         Loop optimizations (default is ON).
        .tensorflow.RewriterConfig.Toggle loop_optimization = 9;
        Parameters:
        value - The loopOptimization to set.
        Returns:
        This builder for chaining.

      • clearLoopOptimization

        public RewriterConfig.Builder clearLoopOptimization()
         Loop optimizations (default is ON).
        .tensorflow.RewriterConfig.Toggle loop_optimization = 9;
        Returns:
        This builder for chaining.

      • 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.

      • setFunctionOptimizationValue

        public RewriterConfig.Builder setFunctionOptimizationValue​(int value)
         Function optimizations (default is ON).
        .tensorflow.RewriterConfig.Toggle function_optimization = 10;
        Parameters:
        value - The enum numeric value on the wire for functionOptimization to set.
        Returns:
        This builder for chaining.

      • setFunctionOptimization

        public RewriterConfig.Builder setFunctionOptimization​(RewriterConfig.Toggle value)
         Function optimizations (default is ON).
        .tensorflow.RewriterConfig.Toggle function_optimization = 10;
        Parameters:
        value - The functionOptimization to set.
        Returns:
        This builder for chaining.

      • clearFunctionOptimization

        public RewriterConfig.Builder clearFunctionOptimization()
         Function optimizations (default is ON).
        .tensorflow.RewriterConfig.Toggle function_optimization = 10;
        Returns:
        This builder for chaining.

      • 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.

      • setDebugStripperValue

        public RewriterConfig.Builder setDebugStripperValue​(int value)
         Strips debug-related nodes from the graph (off by default).
        .tensorflow.RewriterConfig.Toggle debug_stripper = 11;
        Parameters:
        value - The enum numeric value on the wire for debugStripper to set.
        Returns:
        This builder for chaining.

      • setDebugStripper

        public RewriterConfig.Builder setDebugStripper​(RewriterConfig.Toggle value)
         Strips debug-related nodes from the graph (off by default).
        .tensorflow.RewriterConfig.Toggle debug_stripper = 11;
        Parameters:
        value - The debugStripper to set.
        Returns:
        This builder for chaining.

      • clearDebugStripper

        public RewriterConfig.Builder clearDebugStripper()
         Strips debug-related nodes from the graph (off by default).
        .tensorflow.RewriterConfig.Toggle debug_stripper = 11;
        Returns:
        This builder for chaining.

      • 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.

      • setDisableModelPruning

        public RewriterConfig.Builder setDisableModelPruning​(boolean value)
         If true, don't remove unnecessary ops from the graph
        bool disable_model_pruning = 2;
        Parameters:
        value - The disableModelPruning to set.
        Returns:
        This builder for chaining.

      • clearDisableModelPruning

        public RewriterConfig.Builder clearDisableModelPruning()
         If true, don't remove unnecessary ops from the graph
        bool disable_model_pruning = 2;
        Returns:
        This builder for chaining.

      • 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.

      • setScopedAllocatorOptimizationValue

        public RewriterConfig.Builder setScopedAllocatorOptimizationValue​(int value)
         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;
        Parameters:
        value - The enum numeric value on the wire for scopedAllocatorOptimization to set.
        Returns:
        This builder for chaining.

      • 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.

      • setScopedAllocatorOptimization

        public RewriterConfig.Builder setScopedAllocatorOptimization​(RewriterConfig.Toggle value)
         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;
        Parameters:
        value - The scopedAllocatorOptimization to set.
        Returns:
        This builder for chaining.

      • clearScopedAllocatorOptimization

        public RewriterConfig.Builder clearScopedAllocatorOptimization()
         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;
        Returns:
        This builder for chaining.

      • 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.

      • setMetaOptimizerIterationsValue

        public RewriterConfig.Builder setMetaOptimizerIterationsValue​(int value)
         Controls how many times we run the optimizers in meta optimizer (default
 is once).
        .tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;
        Parameters:
        value - The enum numeric value on the wire for metaOptimizerIterations to set.
        Returns:
        This builder for chaining.

      • setMetaOptimizerIterations

        public RewriterConfig.Builder setMetaOptimizerIterations​(RewriterConfig.NumIterationsType value)
         Controls how many times we run the optimizers in meta optimizer (default
 is once).
        .tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;
        Parameters:
        value - The metaOptimizerIterations to set.
        Returns:
        This builder for chaining.

      • clearMetaOptimizerIterations

        public RewriterConfig.Builder clearMetaOptimizerIterations()
         Controls how many times we run the optimizers in meta optimizer (default
 is once).
        .tensorflow.RewriterConfig.NumIterationsType meta_optimizer_iterations = 12;
        Returns:
        This builder for chaining.

      • 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.

      • setMemoryOptimizationValue

        public RewriterConfig.Builder setMemoryOptimizationValue​(int value)
         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;
        Parameters:
        value - The enum numeric value on the wire for memoryOptimization to set.
        Returns:
        This builder for chaining.

      • 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.

      • setMemoryOptimization

        public RewriterConfig.Builder setMemoryOptimization​(RewriterConfig.MemOptType value)
         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;
        Parameters:
        value - The memoryOptimization to set.
        Returns:
        This builder for chaining.

      • clearMemoryOptimization

        public RewriterConfig.Builder clearMemoryOptimization()
         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;
        Returns:
        This builder for chaining.

      • getMemoryOptimizerTargetNodeNameScope

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

      • getMemoryOptimizerTargetNodeNameScopeBytes

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

      • setMemoryOptimizerTargetNodeNameScope

        public RewriterConfig.Builder setMemoryOptimizerTargetNodeNameScope​(String value)
         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;
        Parameters:
        value - The memoryOptimizerTargetNodeNameScope to set.
        Returns:
        This builder for chaining.

      • clearMemoryOptimizerTargetNodeNameScope

        public RewriterConfig.Builder clearMemoryOptimizerTargetNodeNameScope()
         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;
        Returns:
        This builder for chaining.

      • setMemoryOptimizerTargetNodeNameScopeBytes

        public RewriterConfig.Builder setMemoryOptimizerTargetNodeNameScopeBytes​(org.nd4j.shade.protobuf.ByteString value)
         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;
        Parameters:
        value - The bytes for memoryOptimizerTargetNodeNameScope to set.
        Returns:
        This builder for chaining.

      • 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.

      • setAutoParallel

        public RewriterConfig.Builder setAutoParallel​(AutoParallelOptions value)
         Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
        .tensorflow.AutoParallelOptions auto_parallel = 5;

      • setAutoParallel

        public RewriterConfig.Builder setAutoParallel​(AutoParallelOptions.Builder builderForValue)
         Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
        .tensorflow.AutoParallelOptions auto_parallel = 5;

      • mergeAutoParallel

        public RewriterConfig.Builder mergeAutoParallel​(AutoParallelOptions value)
         Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
        .tensorflow.AutoParallelOptions auto_parallel = 5;

      • clearAutoParallel

        public RewriterConfig.Builder clearAutoParallel()
         Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
        .tensorflow.AutoParallelOptions auto_parallel = 5;

      • getAutoParallelBuilder

        public AutoParallelOptions.Builder getAutoParallelBuilder()
         Configures AutoParallel optimization passes either through the
 meta-optimizer or when manually specified through the optimizers field.
        .tensorflow.AutoParallelOptions auto_parallel = 5;

      • hasScopedAllocatorOpts

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

      • clearScopedAllocatorOpts

        public RewriterConfig.Builder clearScopedAllocatorOpts()
        .tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;

      • getScopedAllocatorOptsBuilder

        public ScopedAllocatorOptions.Builder getScopedAllocatorOptsBuilder()
        .tensorflow.ScopedAllocatorOptions scoped_allocator_opts = 16;

      • getOptimizersList

        public org.nd4j.shade.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 registered optimizers will be run after the base optimizers, in
 the order that they are 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 registered optimizers will be run after the base optimizers, in
 the order that they are 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 registered optimizers will be run after the base optimizers, in
 the order that they are 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 org.nd4j.shade.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 registered optimizers will be run after the base optimizers, in
 the order that they are 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.

      • setOptimizers

        public RewriterConfig.Builder setOptimizers​(int index,
                                            String value)
         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 registered optimizers will be run after the base optimizers, in
 the order that they are specified.
        repeated string optimizers = 100;
        Parameters:
        index - The index to set the value at.
        value - The optimizers to set.
        Returns:
        This builder for chaining.

      • addOptimizers

        public RewriterConfig.Builder addOptimizers​(String value)
         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 registered optimizers will be run after the base optimizers, in
 the order that they are specified.
        repeated string optimizers = 100;
        Parameters:
        value - The optimizers to add.
        Returns:
        This builder for chaining.

      • addAllOptimizers

        public RewriterConfig.Builder addAllOptimizers​(Iterable<String> values)
         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 registered optimizers will be run after the base optimizers, in
 the order that they are specified.
        repeated string optimizers = 100;
        Parameters:
        values - The optimizers to add.
        Returns:
        This builder for chaining.

      • clearOptimizers

        public RewriterConfig.Builder clearOptimizers()
         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 registered optimizers will be run after the base optimizers, in
 the order that they are specified.
        repeated string optimizers = 100;
        Returns:
        This builder for chaining.

      • addOptimizersBytes

        public RewriterConfig.Builder addOptimizersBytes​(org.nd4j.shade.protobuf.ByteString value)
         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 registered optimizers will be run after the base optimizers, in
 the order that they are specified.
        repeated string optimizers = 100;
        Parameters:
        value - The bytes of the optimizers to add.
        Returns:
        This builder for chaining.

      • getCustomOptimizersCount

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

      • clearCustomOptimizers

        public RewriterConfig.Builder clearCustomOptimizers()
         list of CustomGraphOptimizers to apply.
        repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;

      • removeCustomOptimizers

        public RewriterConfig.Builder removeCustomOptimizers​(int index)
         list of CustomGraphOptimizers to apply.
        repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;

      • getCustomOptimizersBuilder

        public RewriterConfig.CustomGraphOptimizer.Builder getCustomOptimizersBuilder​(int index)
         list of CustomGraphOptimizers to apply.
        repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;

      • addCustomOptimizersBuilder

        public RewriterConfig.CustomGraphOptimizer.Builder addCustomOptimizersBuilder()
         list of CustomGraphOptimizers to apply.
        repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;

      • addCustomOptimizersBuilder

        public RewriterConfig.CustomGraphOptimizer.Builder addCustomOptimizersBuilder​(int index)
         list of CustomGraphOptimizers to apply.
        repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;

      • getCustomOptimizersBuilderList

        public List<RewriterConfig.CustomGraphOptimizer.Builder> getCustomOptimizersBuilderList()
         list of CustomGraphOptimizers to apply.
        repeated .tensorflow.RewriterConfig.CustomGraphOptimizer custom_optimizers = 200;

      • setUnknownFields

        public final RewriterConfig.Builder setUnknownFields​(org.nd4j.shade.protobuf.UnknownFieldSet unknownFields)
        Specified by:
        setUnknownFields in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        setUnknownFields in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>

      • mergeUnknownFields

        public final RewriterConfig.Builder mergeUnknownFields​(org.nd4j.shade.protobuf.UnknownFieldSet unknownFields)
        Specified by:
        mergeUnknownFields in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        mergeUnknownFields in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<RewriterConfig.Builder>