Home
last modified time | relevance | path

Searched refs:op_pd (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1858 inline bool IsReorderNeeded(const memory::primitive_desc& op_pd) const { in IsReorderNeeded() argument
1860 return op_pd != user_memory_->get_primitive_desc(); in IsReorderNeeded()
1897 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() argument
1901 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
1903 reorder_memory_ = new memory(op_pd); in CheckReorderToOpMem()
1913 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd) { in CheckReorderToOpMem() argument
1915 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
1919 reorder_memory_ = new memory(op_pd); in CheckReorderToOpMem()
1938 inline bool CheckReorderToOpMem(const memory::primitive_desc& op_pd, in CheckReorderToOpMem() argument
1944 if (IsReorderNeeded(op_pd)) { in CheckReorderToOpMem()
[all …]