Merge pull request #59992 from tensorflow/fix-2.11-build

learning-to-play · web-flow · commit c78616f4b001 · 2023-03-14T16:31:51.000-07:00
Fix 2.11 build
diff --git a/tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc b/tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
@@ -5819,11 +5819,11 @@ class ConvertRandomShuffleOp : public OpRewritePattern<TF::RandomShuffleOp> {
   LogicalResult matchAndRewrite(TF::RandomShuffleOp op,
                                 PatternRewriter &rewriter) const override {
     auto no_op = [&]() {
-      rewriter.replaceOp(op, op.getValue());
+      rewriter.replaceOp(op, op.value());
       return success();
     };
 
-    auto input_type = op.getValue().getType().dyn_cast<RankedTensorType>();
+    auto input_type = op.value().getType().dyn_cast<RankedTensorType>();
     if (!input_type) return failure();
     if (input_type.hasStaticShape() && input_type.getNumElements() <= 1)
       // No shuffling is required, so copy input directly to output.
