Searched refs:blocked_md (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 183 mkldnn_memory_desc_t* blocked_md);
470 mkldnn_memory_desc_t blocked_md; in GetTfLayout() local
472 CreateBlockedMemDescHelper(dims, strides, data_.T_, &blocked_md)); in GetTfLayout()
473 return memory::desc(blocked_md); in GetTfLayout()
1203 mkldnn_memory_desc_t* blocked_md) { in CreateBlockedMemDescHelper() argument
1213 mkldnn_memory_desc_init_by_strides(blocked_md, kNumDims, input_dims, in CreateBlockedMemDescHelper()
1324 mkldnn_memory_desc_t blocked_md; in CreateBlockedMemDesc() local
1326 CreateBlockedMemDescHelper(dim, strides, MklDnnType<T>(), &blocked_md)); in CreateBlockedMemDesc()
1327 return memory::desc(blocked_md); in CreateBlockedMemDesc()
1339 auto blocked_md = MklDnnData<T>::CreateBlockedMemDesc(dim, strides);
[all …]
|