Removing unused enum values for tfxla bridge flag from TF Public API

PiperOrigin-RevId: 504986360
This commit is contained in:
Catherine Payne 2023-01-26 17:46:53 -08:00 committed by TensorFlower Gardener
parent 9f9b9edd1b
commit 8421f7a4fc
4 changed files with 23 additions and 24 deletions

View File

@ -166,6 +166,10 @@
* `tf.types.experimental.GenericFunction`
* The `experimental_get_compiler_ir` method supports tf.TensorSpec
compilation arguments.
* `tf.config.experimental.mlir_bridge_rollout`
* Removed enums `MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED` and
`MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED` which are no longer used by
the tf2xla bridge
# Thanks to our Contributors

View File

@ -629,18 +629,9 @@ message ConfigProto {
MLIR_BRIDGE_ROLLOUT_ENABLED = 1;
// Disabling the MLIR bridge disables it for all graphs in this session.
MLIR_BRIDGE_ROLLOUT_DISABLED = 2;
// Enable the MLIR bridge on a per graph basis based on an analysis of
// the features used in the graph. If the features used by the graph are
// supported by the MLIR bridge, the MLIR bridge will be used to run the
// graph.
MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED = 3;
// Enable the MLIR bridge in a fallback mode on a per graph basis based
// on an analysis of the features used in the graph.
// Running the MLIR bridge in the fallback mode means that it is
// executed and it commits all the changes to the TF graph in case
// of success. And it does not in case of failures and let the old bridge
// to process the TF graph.
MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED = 4;
reserved 3, 4;
reserved "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED",
"MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED";
}
// Whether to enable the MLIR-based TF->XLA bridge.
MlirBridgeRollout mlir_bridge_rollout = 17;

View File

@ -145,14 +145,16 @@ tf_proto {
name: "MLIR_BRIDGE_ROLLOUT_DISABLED"
number: 2
}
value {
name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED"
number: 3
reserved_range {
start: 3
end: 3
}
value {
name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED"
number: 4
reserved_range {
start: 4
end: 4
}
reserved_name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED"
reserved_name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED"
}
reserved_range {
start: 2

View File

@ -274,14 +274,16 @@ tf_proto {
name: "MLIR_BRIDGE_ROLLOUT_DISABLED"
number: 2
}
value {
name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED"
number: 3
reserved_range {
start: 3
end: 3
}
value {
name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED"
number: 4
reserved_range {
start: 4
end: 4
}
reserved_name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED"
reserved_name: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED"
}
reserved_range {
start: 2