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

Update tensorflow/core/util/mkl_util.h

Browse Source 
Addressing review comments

Co-authored-by: Penporn Koanantakool <38085909+penpornk@users.noreply.github.com>
This commit is contained in:
Sharada Shiddibhavi 2020-06-18 13:14:37 -07:00 committed by GitHub
parent 5016da3128
commit ededf6f4b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/util/mkl_util.h
View File

@ -1538,13 +1538,8 @@ class MklDnnData {
/// Set function for data buffer of user memory primitive.
inline void SetUsrMemDataHandle(const Tensor* tensor,
std::shared_ptr<stream> t_stream = nullptr) {
CHECK_NOTNULL(user_memory_);
CHECK_NOTNULL(tensor);
#ifdef ENABLE_MKLDNN_THREADPOOL
user_memory_->set_data_handle(GetTensorBuffer(tensor), *t_stream);
#else
user_memory_->set_data_handle(GetTensorBuffer(tensor));
#endif // ENABLE_MKLDNN_THREADPOOL
SetUsrMemDataHandle(GetTensorBuffer(tensor), t_stream);
}
}
/// allocate function for data buffer