Package org.tensorflow.framework

Class ConfigProto.Experimental.Builder

java.lang.Object
com.google.protobuf.AbstractMessageLite.Builder
com.google.protobuf.AbstractMessage.Builder<ConfigProto.Experimental.Builder>
com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>
org.tensorflow.framework.ConfigProto.Experimental.Builder
All Implemented Interfaces:
com.google.protobuf.Message.Builder, com.google.protobuf.MessageLite.Builder, com.google.protobuf.MessageLiteOrBuilder, com.google.protobuf.MessageOrBuilder, Cloneable, ConfigProto.ExperimentalOrBuilder
Enclosing class:
ConfigProto.Experimental
public static final class ConfigProto.Experimental.Builder extends com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder> implements ConfigProto.ExperimentalOrBuilder
 Everything inside Experimental is subject to change and is not subject
 to API stability guarantees in
 https://www.tensorflow.org/guide/version_compat.
Protobuf type tensorflow.ConfigProto.Experimental

  • Method Details

    • 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.Builder<ConfigProto.Experimental.Builder>

    • clear

      public ConfigProto.Experimental.Builder clear()
      Specified by:
      clear in interface com.google.protobuf.Message.Builder
      Specified by:
      clear in interface com.google.protobuf.MessageLite.Builder
      Overrides:
      clear in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • getDescriptorForType

      public com.google.protobuf.Descriptors.Descriptor getDescriptorForType()
      Specified by:
      getDescriptorForType in interface com.google.protobuf.Message.Builder
      Specified by:
      getDescriptorForType in interface com.google.protobuf.MessageOrBuilder
      Overrides:
      getDescriptorForType in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • getDefaultInstanceForType

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

    • build

      public ConfigProto.Experimental build()
      Specified by:
      build in interface com.google.protobuf.Message.Builder
      Specified by:
      build in interface com.google.protobuf.MessageLite.Builder

    • buildPartial

      public ConfigProto.Experimental buildPartial()
      Specified by:
      buildPartial in interface com.google.protobuf.Message.Builder
      Specified by:
      buildPartial in interface com.google.protobuf.MessageLite.Builder

    • clone

      public ConfigProto.Experimental.Builder clone()
      Specified by:
      clone in interface com.google.protobuf.Message.Builder
      Specified by:
      clone in interface com.google.protobuf.MessageLite.Builder
      Overrides:
      clone in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • setField

      public ConfigProto.Experimental.Builder setField(com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
      Specified by:
      setField in interface com.google.protobuf.Message.Builder
      Overrides:
      setField in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • clearField

      public ConfigProto.Experimental.Builder clearField(com.google.protobuf.Descriptors.FieldDescriptor field)
      Specified by:
      clearField in interface com.google.protobuf.Message.Builder
      Overrides:
      clearField in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • clearOneof

      public ConfigProto.Experimental.Builder clearOneof(com.google.protobuf.Descriptors.OneofDescriptor oneof)
      Specified by:
      clearOneof in interface com.google.protobuf.Message.Builder
      Overrides:
      clearOneof in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • setRepeatedField

      public ConfigProto.Experimental.Builder setRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, int index, Object value)
      Specified by:
      setRepeatedField in interface com.google.protobuf.Message.Builder
      Overrides:
      setRepeatedField in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • addRepeatedField

      public ConfigProto.Experimental.Builder addRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
      Specified by:
      addRepeatedField in interface com.google.protobuf.Message.Builder
      Overrides:
      addRepeatedField in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • mergeFrom

      public ConfigProto.Experimental.Builder mergeFrom(com.google.protobuf.Message other)
      Specified by:
      mergeFrom in interface com.google.protobuf.Message.Builder
      Overrides:
      mergeFrom in class com.google.protobuf.AbstractMessage.Builder<ConfigProto.Experimental.Builder>

    • mergeFrom

      public ConfigProto.Experimental.Builder mergeFrom(ConfigProto.Experimental other)

    • isInitialized

      public final boolean isInitialized()
      Specified by:
      isInitialized in interface com.google.protobuf.MessageLiteOrBuilder
      Overrides:
      isInitialized in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • mergeFrom

      public ConfigProto.Experimental.Builder mergeFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Specified by:
      mergeFrom in interface com.google.protobuf.Message.Builder
      Specified by:
      mergeFrom in interface com.google.protobuf.MessageLite.Builder
      Overrides:
      mergeFrom in class com.google.protobuf.AbstractMessage.Builder<ConfigProto.Experimental.Builder>
      Throws:
      IOException

    • getCollectiveGroupLeader

      public String getCollectiveGroupLeader()
       Task name for group resolution.
      string collective_group_leader = 1;
      Specified by:
      getCollectiveGroupLeader in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The collectiveGroupLeader.

    • getCollectiveGroupLeaderBytes

      public com.google.protobuf.ByteString getCollectiveGroupLeaderBytes()
       Task name for group resolution.
      string collective_group_leader = 1;
      Specified by:
      getCollectiveGroupLeaderBytes in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The bytes for collectiveGroupLeader.

    • setCollectiveGroupLeader

      public ConfigProto.Experimental.Builder setCollectiveGroupLeader(String value)
       Task name for group resolution.
      string collective_group_leader = 1;
      Parameters:
      value - The collectiveGroupLeader to set.
      Returns:
      This builder for chaining.

    • clearCollectiveGroupLeader

      public ConfigProto.Experimental.Builder clearCollectiveGroupLeader()
       Task name for group resolution.
      string collective_group_leader = 1;
      Returns:
      This builder for chaining.

    • setCollectiveGroupLeaderBytes

      public ConfigProto.Experimental.Builder setCollectiveGroupLeaderBytes(com.google.protobuf.ByteString value)
       Task name for group resolution.
      string collective_group_leader = 1;
      Parameters:
      value - The bytes for collectiveGroupLeader to set.
      Returns:
      This builder for chaining.

    • getExecutorType

      public String getExecutorType()
       Which executor to use, the default executor will be used
 if it is an empty string or "DEFAULT"
      string executor_type = 3;
      Specified by:
      getExecutorType in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The executorType.

    • getExecutorTypeBytes

      public com.google.protobuf.ByteString getExecutorTypeBytes()
       Which executor to use, the default executor will be used
 if it is an empty string or "DEFAULT"
      string executor_type = 3;
      Specified by:
      getExecutorTypeBytes in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The bytes for executorType.

    • setExecutorType

      public ConfigProto.Experimental.Builder setExecutorType(String value)
       Which executor to use, the default executor will be used
 if it is an empty string or "DEFAULT"
      string executor_type = 3;
      Parameters:
      value - The executorType to set.
      Returns:
      This builder for chaining.

    • clearExecutorType

      public ConfigProto.Experimental.Builder clearExecutorType()
       Which executor to use, the default executor will be used
 if it is an empty string or "DEFAULT"
      string executor_type = 3;
      Returns:
      This builder for chaining.

    • setExecutorTypeBytes

      public ConfigProto.Experimental.Builder setExecutorTypeBytes(com.google.protobuf.ByteString value)
       Which executor to use, the default executor will be used
 if it is an empty string or "DEFAULT"
      string executor_type = 3;
      Parameters:
      value - The bytes for executorType to set.
      Returns:
      This builder for chaining.

    • getRecvBufMaxChunk

      public int getRecvBufMaxChunk()
       Guidance to formatting of large RecvBuf fields for transfer.
 Any positive value sets the max chunk size.  0 defaults to 4096.
 Any negative value indicates no max, i.e. one chunk only.
      int32 recv_buf_max_chunk = 4;
      Specified by:
      getRecvBufMaxChunk in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The recvBufMaxChunk.

    • setRecvBufMaxChunk

      public ConfigProto.Experimental.Builder setRecvBufMaxChunk(int value)
       Guidance to formatting of large RecvBuf fields for transfer.
 Any positive value sets the max chunk size.  0 defaults to 4096.
 Any negative value indicates no max, i.e. one chunk only.
      int32 recv_buf_max_chunk = 4;
      Parameters:
      value - The recvBufMaxChunk to set.
      Returns:
      This builder for chaining.

    • clearRecvBufMaxChunk

      public ConfigProto.Experimental.Builder clearRecvBufMaxChunk()
       Guidance to formatting of large RecvBuf fields for transfer.
 Any positive value sets the max chunk size.  0 defaults to 4096.
 Any negative value indicates no max, i.e. one chunk only.
      int32 recv_buf_max_chunk = 4;
      Returns:
      This builder for chaining.

    • getUseNumaAffinity

      public boolean getUseNumaAffinity()
       If true, and supported by the platform, the runtime will attempt to
 use NUMA affinity where applicable.  One consequence will be the
 existence of as many CPU devices as there are available NUMA nodes.
      bool use_numa_affinity = 5;
      Specified by:
      getUseNumaAffinity in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The useNumaAffinity.

    • setUseNumaAffinity

      public ConfigProto.Experimental.Builder setUseNumaAffinity(boolean value)
       If true, and supported by the platform, the runtime will attempt to
 use NUMA affinity where applicable.  One consequence will be the
 existence of as many CPU devices as there are available NUMA nodes.
      bool use_numa_affinity = 5;
      Parameters:
      value - The useNumaAffinity to set.
      Returns:
      This builder for chaining.

    • clearUseNumaAffinity

      public ConfigProto.Experimental.Builder clearUseNumaAffinity()
       If true, and supported by the platform, the runtime will attempt to
 use NUMA affinity where applicable.  One consequence will be the
 existence of as many CPU devices as there are available NUMA nodes.
      bool use_numa_affinity = 5;
      Returns:
      This builder for chaining.

    • getCollectiveDeterministicSequentialExecution

      public boolean getCollectiveDeterministicSequentialExecution()
       If true, make collective op execution order sequential and deterministic
 for potentially concurrent collective instances.
      bool collective_deterministic_sequential_execution = 6;
      Specified by:
      getCollectiveDeterministicSequentialExecution in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The collectiveDeterministicSequentialExecution.

    • setCollectiveDeterministicSequentialExecution

      public ConfigProto.Experimental.Builder setCollectiveDeterministicSequentialExecution(boolean value)
       If true, make collective op execution order sequential and deterministic
 for potentially concurrent collective instances.
      bool collective_deterministic_sequential_execution = 6;
      Parameters:
      value - The collectiveDeterministicSequentialExecution to set.
      Returns:
      This builder for chaining.

    • clearCollectiveDeterministicSequentialExecution

      public ConfigProto.Experimental.Builder clearCollectiveDeterministicSequentialExecution()
       If true, make collective op execution order sequential and deterministic
 for potentially concurrent collective instances.
      bool collective_deterministic_sequential_execution = 6;
      Returns:
      This builder for chaining.

    • getCollectiveNccl

      public boolean getCollectiveNccl()
       If true, use NCCL for CollectiveOps.  This feature is highly
 experimental.
      bool collective_nccl = 7;
      Specified by:
      getCollectiveNccl in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The collectiveNccl.

    • setCollectiveNccl

      public ConfigProto.Experimental.Builder setCollectiveNccl(boolean value)
       If true, use NCCL for CollectiveOps.  This feature is highly
 experimental.
      bool collective_nccl = 7;
      Parameters:
      value - The collectiveNccl to set.
      Returns:
      This builder for chaining.

    • clearCollectiveNccl

      public ConfigProto.Experimental.Builder clearCollectiveNccl()
       If true, use NCCL for CollectiveOps.  This feature is highly
 experimental.
      bool collective_nccl = 7;
      Returns:
      This builder for chaining.

    • getShareSessionStateInClusterspecPropagation

      public boolean getShareSessionStateInClusterspecPropagation()
       In the following, session state means the value of a variable, elements
 in a hash table, or any other resource, accessible by worker sessions
 held by a TF server.

 When ClusterSpec propagation is enabled, the value of
 isolate_session_state is ignored when deciding whether to share session
 states in a TF server (for backwards compatibility reasons).
 - If share_session_state_in_clusterspec_propagation is true, the session
 states are shared.
 - If share_session_state_in_clusterspec_propagation is false, session
 states are isolated.

 When clusterspec propagation is not used, the value of
 share_session_state_in_clusterspec_propagation is ignored when deciding
 whether to share session states in a TF server.
 - If isolate_session_state is true, session states are isolated.
 - If isolate_session_state is false, session states are shared.

 TODO(b/129330037): Add a single API that consistently treats
 isolate_session_state and ClusterSpec propagation.
      bool share_session_state_in_clusterspec_propagation = 8;
      Specified by:
      getShareSessionStateInClusterspecPropagation in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The shareSessionStateInClusterspecPropagation.

    • setShareSessionStateInClusterspecPropagation

      public ConfigProto.Experimental.Builder setShareSessionStateInClusterspecPropagation(boolean value)
       In the following, session state means the value of a variable, elements
 in a hash table, or any other resource, accessible by worker sessions
 held by a TF server.

 When ClusterSpec propagation is enabled, the value of
 isolate_session_state is ignored when deciding whether to share session
 states in a TF server (for backwards compatibility reasons).
 - If share_session_state_in_clusterspec_propagation is true, the session
 states are shared.
 - If share_session_state_in_clusterspec_propagation is false, session
 states are isolated.

 When clusterspec propagation is not used, the value of
 share_session_state_in_clusterspec_propagation is ignored when deciding
 whether to share session states in a TF server.
 - If isolate_session_state is true, session states are isolated.
 - If isolate_session_state is false, session states are shared.

 TODO(b/129330037): Add a single API that consistently treats
 isolate_session_state and ClusterSpec propagation.
      bool share_session_state_in_clusterspec_propagation = 8;
      Parameters:
      value - The shareSessionStateInClusterspecPropagation to set.
      Returns:
      This builder for chaining.

    • clearShareSessionStateInClusterspecPropagation

      public ConfigProto.Experimental.Builder clearShareSessionStateInClusterspecPropagation()
       In the following, session state means the value of a variable, elements
 in a hash table, or any other resource, accessible by worker sessions
 held by a TF server.

 When ClusterSpec propagation is enabled, the value of
 isolate_session_state is ignored when deciding whether to share session
 states in a TF server (for backwards compatibility reasons).
 - If share_session_state_in_clusterspec_propagation is true, the session
 states are shared.
 - If share_session_state_in_clusterspec_propagation is false, session
 states are isolated.

 When clusterspec propagation is not used, the value of
 share_session_state_in_clusterspec_propagation is ignored when deciding
 whether to share session states in a TF server.
 - If isolate_session_state is true, session states are isolated.
 - If isolate_session_state is false, session states are shared.

 TODO(b/129330037): Add a single API that consistently treats
 isolate_session_state and ClusterSpec propagation.
      bool share_session_state_in_clusterspec_propagation = 8;
      Returns:
      This builder for chaining.

    • getDisableThreadSpinning

      public boolean getDisableThreadSpinning()
       If using a direct session, disable spinning while waiting for work in
 the thread pool. This may result in higher latency for completing ops,
 but in the case where there is a lot of spinning may result in lower
 CPU usage.
      bool disable_thread_spinning = 9;
      Specified by:
      getDisableThreadSpinning in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The disableThreadSpinning.

    • setDisableThreadSpinning

      public ConfigProto.Experimental.Builder setDisableThreadSpinning(boolean value)
       If using a direct session, disable spinning while waiting for work in
 the thread pool. This may result in higher latency for completing ops,
 but in the case where there is a lot of spinning may result in lower
 CPU usage.
      bool disable_thread_spinning = 9;
      Parameters:
      value - The disableThreadSpinning to set.
      Returns:
      This builder for chaining.

    • clearDisableThreadSpinning

      public ConfigProto.Experimental.Builder clearDisableThreadSpinning()
       If using a direct session, disable spinning while waiting for work in
 the thread pool. This may result in higher latency for completing ops,
 but in the case where there is a lot of spinning may result in lower
 CPU usage.
      bool disable_thread_spinning = 9;
      Returns:
      This builder for chaining.

    • getShareClusterDevicesInSession

      public boolean getShareClusterDevicesInSession()
       This was promoted to a non-experimental API. Please use
 ConfigProto.share_cluster_devices_in_session instead.
      bool share_cluster_devices_in_session = 10;
      Specified by:
      getShareClusterDevicesInSession in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The shareClusterDevicesInSession.

    • setShareClusterDevicesInSession

      public ConfigProto.Experimental.Builder setShareClusterDevicesInSession(boolean value)
       This was promoted to a non-experimental API. Please use
 ConfigProto.share_cluster_devices_in_session instead.
      bool share_cluster_devices_in_session = 10;
      Parameters:
      value - The shareClusterDevicesInSession to set.
      Returns:
      This builder for chaining.

    • clearShareClusterDevicesInSession

      public ConfigProto.Experimental.Builder clearShareClusterDevicesInSession()
       This was promoted to a non-experimental API. Please use
 ConfigProto.share_cluster_devices_in_session instead.
      bool share_cluster_devices_in_session = 10;
      Returns:
      This builder for chaining.

    • hasSessionMetadata

      public boolean hasSessionMetadata()
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;
      Specified by:
      hasSessionMetadata in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      Whether the sessionMetadata field is set.

    • getSessionMetadata

      public SessionMetadata getSessionMetadata()
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;
      Specified by:
      getSessionMetadata in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The sessionMetadata.

    • setSessionMetadata

      public ConfigProto.Experimental.Builder setSessionMetadata(SessionMetadata value)
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;

    • setSessionMetadata

      public ConfigProto.Experimental.Builder setSessionMetadata(SessionMetadata.Builder builderForValue)
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;

    • mergeSessionMetadata

      public ConfigProto.Experimental.Builder mergeSessionMetadata(SessionMetadata value)
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;

    • clearSessionMetadata

      public ConfigProto.Experimental.Builder clearSessionMetadata()
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;

    • getSessionMetadataBuilder

      public SessionMetadata.Builder getSessionMetadataBuilder()
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;

    • getSessionMetadataOrBuilder

      public SessionMetadataOrBuilder getSessionMetadataOrBuilder()
       Metadata about the session.

 If set, this can be used by the runtime and the Ops for debugging,
 monitoring, etc.

 NOTE: This is currently used and propagated only by the direct session
 and EagerContext.
      .tensorflow.SessionMetadata session_metadata = 11;
      Specified by:
      getSessionMetadataOrBuilder in interface ConfigProto.ExperimentalOrBuilder

    • getOptimizeForStaticGraph

      public boolean getOptimizeForStaticGraph()
       If true, the session may treat the graph as being static for optimization
 purposes.

 If this option is set to true when a session is created, the full
 GraphDef must be passed in a single call to Session::Create(), and
 Session::Extend() may not be supported.
      bool optimize_for_static_graph = 12;
      Specified by:
      getOptimizeForStaticGraph in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The optimizeForStaticGraph.

    • setOptimizeForStaticGraph

      public ConfigProto.Experimental.Builder setOptimizeForStaticGraph(boolean value)
       If true, the session may treat the graph as being static for optimization
 purposes.

 If this option is set to true when a session is created, the full
 GraphDef must be passed in a single call to Session::Create(), and
 Session::Extend() may not be supported.
      bool optimize_for_static_graph = 12;
      Parameters:
      value - The optimizeForStaticGraph to set.
      Returns:
      This builder for chaining.

    • clearOptimizeForStaticGraph

      public ConfigProto.Experimental.Builder clearOptimizeForStaticGraph()
       If true, the session may treat the graph as being static for optimization
 purposes.

 If this option is set to true when a session is created, the full
 GraphDef must be passed in a single call to Session::Create(), and
 Session::Extend() may not be supported.
      bool optimize_for_static_graph = 12;
      Returns:
      This builder for chaining.

    • getEnableMlirBridge

      public boolean getEnableMlirBridge()
       Whether to enable the MLIR-based TF->XLA bridge. This is only used if set
 to true. Default value or false is ignored. Use mlir_bridge_rollout for
 finer control.

 If this option is set to true when a session is created, MLIR is used to
 perform the set of graph transformations to put the graph in a form that
 can be executed with delegation of some computations to an accelerator.
 This builds on the model of XLA where a subset of the graph is
 encapsulated and attached to a "compile" operation, whose result is fed
 to an "execute" operation. The kernel for these operations is responsible
 to lower the encapsulated graph to a particular device.
      bool enable_mlir_bridge = 13;
      Specified by:
      getEnableMlirBridge in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The enableMlirBridge.

    • setEnableMlirBridge

      public ConfigProto.Experimental.Builder setEnableMlirBridge(boolean value)
       Whether to enable the MLIR-based TF->XLA bridge. This is only used if set
 to true. Default value or false is ignored. Use mlir_bridge_rollout for
 finer control.

 If this option is set to true when a session is created, MLIR is used to
 perform the set of graph transformations to put the graph in a form that
 can be executed with delegation of some computations to an accelerator.
 This builds on the model of XLA where a subset of the graph is
 encapsulated and attached to a "compile" operation, whose result is fed
 to an "execute" operation. The kernel for these operations is responsible
 to lower the encapsulated graph to a particular device.
      bool enable_mlir_bridge = 13;
      Parameters:
      value - The enableMlirBridge to set.
      Returns:
      This builder for chaining.

    • clearEnableMlirBridge

      public ConfigProto.Experimental.Builder clearEnableMlirBridge()
       Whether to enable the MLIR-based TF->XLA bridge. This is only used if set
 to true. Default value or false is ignored. Use mlir_bridge_rollout for
 finer control.

 If this option is set to true when a session is created, MLIR is used to
 perform the set of graph transformations to put the graph in a form that
 can be executed with delegation of some computations to an accelerator.
 This builds on the model of XLA where a subset of the graph is
 encapsulated and attached to a "compile" operation, whose result is fed
 to an "execute" operation. The kernel for these operations is responsible
 to lower the encapsulated graph to a particular device.
      bool enable_mlir_bridge = 13;
      Returns:
      This builder for chaining.

    • getMlirBridgeRolloutValue

      public int getMlirBridgeRolloutValue()
       Whether to enable the MLIR-based TF->XLA bridge.
      .tensorflow.ConfigProto.Experimental.MlirBridgeRollout mlir_bridge_rollout = 17;
      Specified by:
      getMlirBridgeRolloutValue in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The enum numeric value on the wire for mlirBridgeRollout.

    • setMlirBridgeRolloutValue

      public ConfigProto.Experimental.Builder setMlirBridgeRolloutValue(int value)
       Whether to enable the MLIR-based TF->XLA bridge.
      .tensorflow.ConfigProto.Experimental.MlirBridgeRollout mlir_bridge_rollout = 17;
      Parameters:
      value - The enum numeric value on the wire for mlirBridgeRollout to set.
      Returns:
      This builder for chaining.

    • getMlirBridgeRollout

      public ConfigProto.Experimental.MlirBridgeRollout getMlirBridgeRollout()
       Whether to enable the MLIR-based TF->XLA bridge.
      .tensorflow.ConfigProto.Experimental.MlirBridgeRollout mlir_bridge_rollout = 17;
      Specified by:
      getMlirBridgeRollout in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The mlirBridgeRollout.

    • setMlirBridgeRollout

      public ConfigProto.Experimental.Builder setMlirBridgeRollout(ConfigProto.Experimental.MlirBridgeRollout value)
       Whether to enable the MLIR-based TF->XLA bridge.
      .tensorflow.ConfigProto.Experimental.MlirBridgeRollout mlir_bridge_rollout = 17;
      Parameters:
      value - The mlirBridgeRollout to set.
      Returns:
      This builder for chaining.

    • clearMlirBridgeRollout

      public ConfigProto.Experimental.Builder clearMlirBridgeRollout()
       Whether to enable the MLIR-based TF->XLA bridge.
      .tensorflow.ConfigProto.Experimental.MlirBridgeRollout mlir_bridge_rollout = 17;
      Returns:
      This builder for chaining.

    • getEnableMlirGraphOptimization

      public boolean getEnableMlirGraphOptimization()
       Whether to enable the MLIR-based Graph optimizations.

 This will become a part of standard Tensorflow graph optimization
 pipeline, currently this is only used for gradual migration and testing
 new passes that are replacing existing optimizations in Grappler.
      bool enable_mlir_graph_optimization = 16;
      Specified by:
      getEnableMlirGraphOptimization in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The enableMlirGraphOptimization.

    • setEnableMlirGraphOptimization

      public ConfigProto.Experimental.Builder setEnableMlirGraphOptimization(boolean value)
       Whether to enable the MLIR-based Graph optimizations.

 This will become a part of standard Tensorflow graph optimization
 pipeline, currently this is only used for gradual migration and testing
 new passes that are replacing existing optimizations in Grappler.
      bool enable_mlir_graph_optimization = 16;
      Parameters:
      value - The enableMlirGraphOptimization to set.
      Returns:
      This builder for chaining.

    • clearEnableMlirGraphOptimization

      public ConfigProto.Experimental.Builder clearEnableMlirGraphOptimization()
       Whether to enable the MLIR-based Graph optimizations.

 This will become a part of standard Tensorflow graph optimization
 pipeline, currently this is only used for gradual migration and testing
 new passes that are replacing existing optimizations in Grappler.
      bool enable_mlir_graph_optimization = 16;
      Returns:
      This builder for chaining.

    • getDisableOutputPartitionGraphs

      public boolean getDisableOutputPartitionGraphs()
       If true, the session will not store an additional copy of the graph for
 each subgraph.

 If this option is set to true when a session is created, the
 `RunOptions.output_partition_graphs` options must not be set.
      bool disable_output_partition_graphs = 14;
      Specified by:
      getDisableOutputPartitionGraphs in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The disableOutputPartitionGraphs.

    • setDisableOutputPartitionGraphs

      public ConfigProto.Experimental.Builder setDisableOutputPartitionGraphs(boolean value)
       If true, the session will not store an additional copy of the graph for
 each subgraph.

 If this option is set to true when a session is created, the
 `RunOptions.output_partition_graphs` options must not be set.
      bool disable_output_partition_graphs = 14;
      Parameters:
      value - The disableOutputPartitionGraphs to set.
      Returns:
      This builder for chaining.

    • clearDisableOutputPartitionGraphs

      public ConfigProto.Experimental.Builder clearDisableOutputPartitionGraphs()
       If true, the session will not store an additional copy of the graph for
 each subgraph.

 If this option is set to true when a session is created, the
 `RunOptions.output_partition_graphs` options must not be set.
      bool disable_output_partition_graphs = 14;
      Returns:
      This builder for chaining.

    • getXlaFusionAutotunerThresh

      public long getXlaFusionAutotunerThresh()
       Minimum number of batches run through the XLA graph before XLA fusion
 autotuner is enabled. Default value of zero disables the autotuner.

 The XLA fusion autotuner can improve performance by executing a heuristic
 search on the compiler parameters.
      int64 xla_fusion_autotuner_thresh = 15;
      Specified by:
      getXlaFusionAutotunerThresh in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The xlaFusionAutotunerThresh.

    • setXlaFusionAutotunerThresh

      public ConfigProto.Experimental.Builder setXlaFusionAutotunerThresh(long value)
       Minimum number of batches run through the XLA graph before XLA fusion
 autotuner is enabled. Default value of zero disables the autotuner.

 The XLA fusion autotuner can improve performance by executing a heuristic
 search on the compiler parameters.
      int64 xla_fusion_autotuner_thresh = 15;
      Parameters:
      value - The xlaFusionAutotunerThresh to set.
      Returns:
      This builder for chaining.

    • clearXlaFusionAutotunerThresh

      public ConfigProto.Experimental.Builder clearXlaFusionAutotunerThresh()
       Minimum number of batches run through the XLA graph before XLA fusion
 autotuner is enabled. Default value of zero disables the autotuner.

 The XLA fusion autotuner can improve performance by executing a heuristic
 search on the compiler parameters.
      int64 xla_fusion_autotuner_thresh = 15;
      Returns:
      This builder for chaining.

    • getUseTfrt

      public boolean getUseTfrt()
       Whether runtime execution uses TFRT.
      bool use_tfrt = 18;
      Specified by:
      getUseTfrt in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The useTfrt.

    • setUseTfrt

      public ConfigProto.Experimental.Builder setUseTfrt(boolean value)
       Whether runtime execution uses TFRT.
      bool use_tfrt = 18;
      Parameters:
      value - The useTfrt to set.
      Returns:
      This builder for chaining.

    • clearUseTfrt

      public ConfigProto.Experimental.Builder clearUseTfrt()
       Whether runtime execution uses TFRT.
      bool use_tfrt = 18;
      Returns:
      This builder for chaining.

    • getEnableMultiHost

      public boolean getEnableMultiHost()
       If true, use Pathways with TFRT API for multi host support.
      bool enable_multi_host = 27;
      Specified by:
      getEnableMultiHost in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The enableMultiHost.

    • setEnableMultiHost

      public ConfigProto.Experimental.Builder setEnableMultiHost(boolean value)
       If true, use Pathways with TFRT API for multi host support.
      bool enable_multi_host = 27;
      Parameters:
      value - The enableMultiHost to set.
      Returns:
      This builder for chaining.

    • clearEnableMultiHost

      public ConfigProto.Experimental.Builder clearEnableMultiHost()
       If true, use Pathways with TFRT API for multi host support.
      bool enable_multi_host = 27;
      Returns:
      This builder for chaining.

    • getTfrtUseIfrt

      public boolean getTfrtUseIfrt()
       If true, use ifrt as the backend for TFRT. This is only used when
 `use_tfrt` is true.
      bool tfrt_use_ifrt = 32;
      Specified by:
      getTfrtUseIfrt in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The tfrtUseIfrt.

    • setTfrtUseIfrt

      public ConfigProto.Experimental.Builder setTfrtUseIfrt(boolean value)
       If true, use ifrt as the backend for TFRT. This is only used when
 `use_tfrt` is true.
      bool tfrt_use_ifrt = 32;
      Parameters:
      value - The tfrtUseIfrt to set.
      Returns:
      This builder for chaining.

    • clearTfrtUseIfrt

      public ConfigProto.Experimental.Builder clearTfrtUseIfrt()
       If true, use ifrt as the backend for TFRT. This is only used when
 `use_tfrt` is true.
      bool tfrt_use_ifrt = 32;
      Returns:
      This builder for chaining.

    • getBackendServerPort

      public int getBackendServerPort()
       Port for the Pathways server. Ignored if enable_multi_host=false.
      int32 backend_server_port = 28;
      Specified by:
      getBackendServerPort in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The backendServerPort.

    • setBackendServerPort

      public ConfigProto.Experimental.Builder setBackendServerPort(int value)
       Port for the Pathways server. Ignored if enable_multi_host=false.
      int32 backend_server_port = 28;
      Parameters:
      value - The backendServerPort to set.
      Returns:
      This builder for chaining.

    • clearBackendServerPort

      public ConfigProto.Experimental.Builder clearBackendServerPort()
       Port for the Pathways server. Ignored if enable_multi_host=false.
      int32 backend_server_port = 28;
      Returns:
      This builder for chaining.

    • getTargetTpu

      public boolean getTargetTpu()
       If true, TFRT will use TPU specific compiler passes and perform TPU
 specific initialization.
      bool target_tpu = 29;
      Specified by:
      getTargetTpu in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The targetTpu.

    • setTargetTpu

      public ConfigProto.Experimental.Builder setTargetTpu(boolean value)
       If true, TFRT will use TPU specific compiler passes and perform TPU
 specific initialization.
      bool target_tpu = 29;
      Parameters:
      value - The targetTpu to set.
      Returns:
      This builder for chaining.

    • clearTargetTpu

      public ConfigProto.Experimental.Builder clearTargetTpu()
       If true, TFRT will use TPU specific compiler passes and perform TPU
 specific initialization.
      bool target_tpu = 29;
      Returns:
      This builder for chaining.

    • getTargetGpu

      public boolean getTargetGpu()
       If true, TFRT will use GPU specific compiler passes and perform GPU
 specific initialization.
      bool target_gpu = 30;
      Specified by:
      getTargetGpu in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The targetGpu.

    • setTargetGpu

      public ConfigProto.Experimental.Builder setTargetGpu(boolean value)
       If true, TFRT will use GPU specific compiler passes and perform GPU
 specific initialization.
      bool target_gpu = 30;
      Parameters:
      value - The targetGpu to set.
      Returns:
      This builder for chaining.

    • clearTargetGpu

      public ConfigProto.Experimental.Builder clearTargetGpu()
       If true, TFRT will use GPU specific compiler passes and perform GPU
 specific initialization.
      bool target_gpu = 30;
      Returns:
      This builder for chaining.

    • getStreamMergeThreshold

      public int getStreamMergeThreshold()
       The threshold to merge small streams in TFRT. The stream with cost
 smaller than the threshold will be merged. Setting it to value 1
 disables all merges.
      int32 stream_merge_threshold = 31;
      Specified by:
      getStreamMergeThreshold in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The streamMergeThreshold.

    • setStreamMergeThreshold

      public ConfigProto.Experimental.Builder setStreamMergeThreshold(int value)
       The threshold to merge small streams in TFRT. The stream with cost
 smaller than the threshold will be merged. Setting it to value 1
 disables all merges.
      int32 stream_merge_threshold = 31;
      Parameters:
      value - The streamMergeThreshold to set.
      Returns:
      This builder for chaining.

    • clearStreamMergeThreshold

      public ConfigProto.Experimental.Builder clearStreamMergeThreshold()
       The threshold to merge small streams in TFRT. The stream with cost
 smaller than the threshold will be merged. Setting it to value 1
 disables all merges.
      int32 stream_merge_threshold = 31;
      Returns:
      This builder for chaining.

    • getDisableFunctionalOpsLowering

      public boolean getDisableFunctionalOpsLowering()
       Whether functional control flow op lowering should be disabled. This is
 useful when executing within a portable runtime where control flow op
 kernels may not be loaded due to selective registration.
      bool disable_functional_ops_lowering = 21;
      Specified by:
      getDisableFunctionalOpsLowering in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The disableFunctionalOpsLowering.

    • setDisableFunctionalOpsLowering

      public ConfigProto.Experimental.Builder setDisableFunctionalOpsLowering(boolean value)
       Whether functional control flow op lowering should be disabled. This is
 useful when executing within a portable runtime where control flow op
 kernels may not be loaded due to selective registration.
      bool disable_functional_ops_lowering = 21;
      Parameters:
      value - The disableFunctionalOpsLowering to set.
      Returns:
      This builder for chaining.

    • clearDisableFunctionalOpsLowering

      public ConfigProto.Experimental.Builder clearDisableFunctionalOpsLowering()
       Whether functional control flow op lowering should be disabled. This is
 useful when executing within a portable runtime where control flow op
 kernels may not be loaded due to selective registration.
      bool disable_functional_ops_lowering = 21;
      Returns:
      This builder for chaining.

    • getXlaPreferSingleGraphCluster

      public boolean getXlaPreferSingleGraphCluster()
       Provides a hint to XLA auto clustering to prefer forming a single large
 cluster that encompases most of the graph.
      bool xla_prefer_single_graph_cluster = 22;
      Specified by:
      getXlaPreferSingleGraphCluster in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The xlaPreferSingleGraphCluster.

    • setXlaPreferSingleGraphCluster

      public ConfigProto.Experimental.Builder setXlaPreferSingleGraphCluster(boolean value)
       Provides a hint to XLA auto clustering to prefer forming a single large
 cluster that encompases most of the graph.
      bool xla_prefer_single_graph_cluster = 22;
      Parameters:
      value - The xlaPreferSingleGraphCluster to set.
      Returns:
      This builder for chaining.

    • clearXlaPreferSingleGraphCluster

      public ConfigProto.Experimental.Builder clearXlaPreferSingleGraphCluster()
       Provides a hint to XLA auto clustering to prefer forming a single large
 cluster that encompases most of the graph.
      bool xla_prefer_single_graph_cluster = 22;
      Returns:
      This builder for chaining.

    • hasCoordinationConfig

      public boolean hasCoordinationConfig()
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;
      Specified by:
      hasCoordinationConfig in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      Whether the coordinationConfig field is set.

    • getCoordinationConfig

      public CoordinationConfig.CoordinationServiceConfig getCoordinationConfig()
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;
      Specified by:
      getCoordinationConfig in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The coordinationConfig.

    • setCoordinationConfig

      public ConfigProto.Experimental.Builder setCoordinationConfig(CoordinationConfig.CoordinationServiceConfig value)
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;

    • setCoordinationConfig

      public ConfigProto.Experimental.Builder setCoordinationConfig(CoordinationConfig.CoordinationServiceConfig.Builder builderForValue)
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;

    • mergeCoordinationConfig

      public ConfigProto.Experimental.Builder mergeCoordinationConfig(CoordinationConfig.CoordinationServiceConfig value)
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;

    • clearCoordinationConfig

      public ConfigProto.Experimental.Builder clearCoordinationConfig()
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;

    • getCoordinationConfigBuilder

      public CoordinationConfig.CoordinationServiceConfig.Builder getCoordinationConfigBuilder()
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;

    • getCoordinationConfigOrBuilder

      public CoordinationConfig.CoordinationServiceConfigOrBuilder getCoordinationConfigOrBuilder()
       Distributed coordination service configurations.
      .tensorflow.CoordinationServiceConfig coordination_config = 23;
      Specified by:
      getCoordinationConfigOrBuilder in interface ConfigProto.ExperimentalOrBuilder

    • getDisableOptimizeForStaticGraph

      public boolean getDisableOptimizeForStaticGraph()
       If true, the session will treat the graph as being non-static for
 optimization purposes.

 If this option is set to true when a session is created, the full
 GraphDef will be retained to enable calls to Session::Extend().
 Calling Extend() without setting this flag will result in errors.

 This option is meant to replace `optimize_for_static_graph` and it
 aims to negate its value.
      bool disable_optimize_for_static_graph = 24;
      Specified by:
      getDisableOptimizeForStaticGraph in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The disableOptimizeForStaticGraph.

    • setDisableOptimizeForStaticGraph

      public ConfigProto.Experimental.Builder setDisableOptimizeForStaticGraph(boolean value)
       If true, the session will treat the graph as being non-static for
 optimization purposes.

 If this option is set to true when a session is created, the full
 GraphDef will be retained to enable calls to Session::Extend().
 Calling Extend() without setting this flag will result in errors.

 This option is meant to replace `optimize_for_static_graph` and it
 aims to negate its value.
      bool disable_optimize_for_static_graph = 24;
      Parameters:
      value - The disableOptimizeForStaticGraph to set.
      Returns:
      This builder for chaining.

    • clearDisableOptimizeForStaticGraph

      public ConfigProto.Experimental.Builder clearDisableOptimizeForStaticGraph()
       If true, the session will treat the graph as being non-static for
 optimization purposes.

 If this option is set to true when a session is created, the full
 GraphDef will be retained to enable calls to Session::Extend().
 Calling Extend() without setting this flag will result in errors.

 This option is meant to replace `optimize_for_static_graph` and it
 aims to negate its value.
      bool disable_optimize_for_static_graph = 24;
      Returns:
      This builder for chaining.

    • getDisableEagerExecutorStreamingEnqueue

      public boolean getDisableEagerExecutorStreamingEnqueue()
       Whether eager remote execution will stream all the function calls or
 allow them to happen in parallel. When true, streaming execution is
 disabled, and parallel execution is allowed.
      bool disable_eager_executor_streaming_enqueue = 26;
      Specified by:
      getDisableEagerExecutorStreamingEnqueue in interface ConfigProto.ExperimentalOrBuilder
      Returns:
      The disableEagerExecutorStreamingEnqueue.

    • setDisableEagerExecutorStreamingEnqueue

      public ConfigProto.Experimental.Builder setDisableEagerExecutorStreamingEnqueue(boolean value)
       Whether eager remote execution will stream all the function calls or
 allow them to happen in parallel. When true, streaming execution is
 disabled, and parallel execution is allowed.
      bool disable_eager_executor_streaming_enqueue = 26;
      Parameters:
      value - The disableEagerExecutorStreamingEnqueue to set.
      Returns:
      This builder for chaining.

    • clearDisableEagerExecutorStreamingEnqueue

      public ConfigProto.Experimental.Builder clearDisableEagerExecutorStreamingEnqueue()
       Whether eager remote execution will stream all the function calls or
 allow them to happen in parallel. When true, streaming execution is
 disabled, and parallel execution is allowed.
      bool disable_eager_executor_streaming_enqueue = 26;
      Returns:
      This builder for chaining.

    • setUnknownFields

      public final ConfigProto.Experimental.Builder setUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)
      Specified by:
      setUnknownFields in interface com.google.protobuf.Message.Builder
      Overrides:
      setUnknownFields in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>

    • mergeUnknownFields

      public final ConfigProto.Experimental.Builder mergeUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)
      Specified by:
      mergeUnknownFields in interface com.google.protobuf.Message.Builder
      Overrides:
      mergeUnknownFields in class com.google.protobuf.GeneratedMessageV3.Builder<ConfigProto.Experimental.Builder>