Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc428 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
435 MklDnnData<T>::CreateBlockedMemDesc(input_dims, input_strides); in ComputeMklSlice()
441 MklDnnData<T>::CreateBlockedMemDesc(size_dims, output_strides); in ComputeMklSlice()
Dmkl_aggregate_ops.cc208 md = MklDnnData<T>::CreateBlockedMemDesc(dims, strides); in Compute()
Dmkl_relu_op.cc473 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
628 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
Dmkl_concat_op.cc679 dst_md = MklDnnData<T>::CreateBlockedMemDesc( in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1322 static inline memory::desc CreateBlockedMemDesc(const memory::dims& dim, in CreateBlockedMemDesc() function
1339 auto blocked_md = MklDnnData<T>::CreateBlockedMemDesc(dim, strides);