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

Merge pull request #38614 from Intel-tensorflow:nhasabni/mkl_eager_op_compile_fix

Browse Source 
PiperOrigin-RevId: 306939573
Change-Id: Id5bbcca8608abcb590f246dfb661d812808cac09
This commit is contained in:
TensorFlower Gardener 2020-04-16 16:14:07 -07:00
commit d5743e269c
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()));