Searched refs:CreateBlockedMemDesc (Results 1 – 5 of 5) sorted by relevance
410 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()417 MklDnnData<T>::CreateBlockedMemDesc(size_dims, output_strides); in ComputeMklSlice()
180 auto output_tf_md = MklDnnData<T>::CreateBlockedMemDesc( in Compute()
129 md1 = MklDnnData<T>::CreateBlockedMemDesc(dims, strides); in Compute()
480 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()592 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
1722 static inline memory::desc CreateBlockedMemDesc(const memory::dims& dim, in CreateBlockedMemDesc() function1736 auto blocked_md = MklDnnData<T>::CreateBlockedMemDesc(dim, strides);