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

review comment changes

Browse Source
This commit is contained in:
nammbash 2019-10-22 14:17:42 -07:00
parent 2bb4fe3658
commit 33454c1bd0
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -36,11 +36,7 @@ class MklEagerOpRewrite : public EagerOpRewrite {
// Overload Operator== for std::find comparison
// used by SlowCheckIfKernelRegistered.
bool operator==(const MklEagerOp& rhs) const {
if (op_name.compare(rhs.op_name) == 0) {
return true;
} else {
return false;
}
return (op_name.compare(rhs.op_name) == 0);
}
};
@ -86,8 +82,7 @@ class MklEagerOpRewrite : public EagerOpRewrite {
bool FastCheckIfKernelRegistered(string op_name, DataType dt);
// This is called by FastCheckIfKernelRegistered once per unique op name and
// data
// type.
// data type.
bool SlowCheckIfKernelRegistered(string op_name, DataType dt);
// map used by FastCheckIfKernelRegistered.