MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ReorderStrides

Function ReorderStrides

tensorflow/core/kernels/mkl_transpose_op.cc:53–61  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

51 const gtl::ArraySlice<int32>& perm);
52
53static inline memory::dims ReorderStrides(const memory::dims& strides,
54 const gtl::ArraySlice<int32>& perm) {
55 memory::dims reordered_strides;
56 reordered_strides.resize(strides.size());
57 for (size_t i = 0; i < strides.size(); ++i) {
58 reordered_strides[perm[i]] = strides[i];
59 }
60 return reordered_strides;
61}
62
63// Transpose of N-dimensional tensor using OneDNN
64template <typename T>

Callers 1

MKLTransposeNDFunction · 0.85

Calls 2

resizeMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected