From 2989b49a5a2329acd3a711f5e9350d002d028f42 Mon Sep 17 00:00:00 2001
From: Niranjan Hasabnis <niranjan.hasabnis@intel.com>
Date: Thu, 16 Apr 2020 14:41:20 -0700
Subject: [PATCH] [Intel MKL] Fixing mkl_eager_op_rewrite compile error
This PR fixes build error in --config=mkl introduced by an API change in EagerOperation.
---
tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite.cc b/tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite.cc
index 346ccc11ca8..7f5ec70d0a1 100644
--- a/tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite.cc
+++ b/tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite.cc
@@ -136,7 +136,7 @@ Status MklEagerOpRewrite::SetupNewOp(
->Set("_kernel", mkl_op_registry::kMklNameChangeOpLabel);
if (orig_op->Device() == kVariantDeviceNull) {
- string device_name = orig_op->GetDeviceName();
+ string device_name = orig_op->DeviceName();
(*new_mkl_op)->SetDeviceName(device_name.c_str());
} else if (VariantDeviceIsCustom(orig_op->Device())) {
(*new_mkl_op)->SetDevice(absl::get<CustomDevice*>(orig_op->Device()));