commit 31a3e515a153ed2b646d0f11872acf802f66028d
author Catherine Payne <paynecl@google.com> Fri Apr 15 22:16:53 2022 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Apr 15 22:19:51 2022 -0700
tree 3466bf21b46ccb68527782108c7cd79e5d9cfc8e
parent 5c8c20c0180e1a783ff0708cb8f9b08dfdf065cb

Correcting error in instrumentation of new TFXLABridge

PiperOrigin-RevId: 442165238

tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc

diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
index aa22782..7980866 100644
--- a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
+++ b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
@@ -334,7 +334,7 @@
   Status status = mlir::TFTPU::RunTFXLABridge(module, enable_logging,
                                               CreateTFXLABridgePipeline);
   tensorflow::metrics::UpdateTfMlirBridgeFirstPhaseCounter(
-      /*device type*/ "cpu/gpu", /*bridge version*/ "v2",
+      /*device type*/ "cpu/gpu", /*bridge version*/ "tfxla",
       /*fallback_enabled*/ false,
       /*result*/ status == Status::OK() ? "success" : "failure");
   return status;