From 90078eb3445f92a7fe149e166589812e22ecd881 Mon Sep 17 00:00:00 2001 From: Christian Sigg Date: Sun, 28 Feb 2021 11:33:08 -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: 360055555 Change-Id: Ibe2c962444907140a7316d2e477897337a4d63b5 --- tensorflow/compiler/mlir/tfr/ir/tfr_ops.td | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td b/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td index 9d1e7fb8513..6971edc298f 100644 --- a/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td +++ b/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td @@ -441,7 +441,7 @@ def TFR_TFRFuncOp : TFR_Op<"func", [HasParent<"ModuleOp">, // non-derived ones. llvm::StringSet<> getDefinedAttributeNames() { llvm::StringSet<> all_attrs; - for (auto& attr : getAttrs()) { + for (auto& attr : (*this)->getAttrs()) { all_attrs.insert(attr.first.strref()); } for (const auto& operand : llvm::enumerate(getType().getInputs())) {