experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Intel MKL] Fixing mkl_eager_op_rewrite compile error

Browse Source 
This PR fixes build error in --config=mkl introduced by an API change in EagerOperation.
This commit is contained in:
Niranjan Hasabnis 2020-04-16 14:41:20 -07:00
parent 83eeaab4b3
commit 2989b49a5a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite.cc
View File

@ -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()));