From 7d034e94dcc81356ba896775924aecc23519a3b1 Mon Sep 17 00:00:00 2001
From: Christian Sigg <csigg@google.com>
Date: Mon, 21 Dec 2020 02:29:14 -0800
Subject: [PATCH] Use mlir::OpState::operator->() to get to methods of
mlir::Operation.
This is a preparation step to remove those methods from OpState.
PiperOrigin-RevId: 348436381
Change-Id: I814d25b2bdab58b7123ff33b7793c7c3a4fcf7cf
---
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc b/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
index 41cce577f20..eba71da2382 100644
--- a/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
+++ b/tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
@@ -69,8 +69,8 @@ inline LogicalResult GetIntegerArraySafe(
LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() {
func_.eraseBody();
func_.addEntryBlock();
- func_.setAttr(kTFImplements,
- StringAttr::get(kMaxUnpooling, func_.getContext()));
+ func_->setAttr(kTFImplements,
+ StringAttr::get(kMaxUnpooling, func_.getContext()));
OpBuilder builder(func_.getBody());
std::string custom_option_buffer;