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

Function DoMatrixTransposeImpl

tensorflow/core/kernels/transpose_functor.h:238–246  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

236
237template <typename Device>
238inline Status DoMatrixTransposeImpl(const Device& device, const Tensor& in,
239 bool conjugate, Tensor* out) {
240 const int ndims = in.dims();
241 if (ndims == 0) return Status::OK();
242 TransposePermsVec perm(ndims);
243 std::iota(perm.begin(), perm.end(), 0);
244 std::swap(perm[ndims - 2], perm[ndims - 1]);
245 return DoTransposeImpl(device, in, perm, conjugate, out);
246}
247
248#ifdef TENSORFLOW_USE_SYCL
249// For SYCL lets always go through Eigen

Callers 4

DoMatrixTransposeFunction · 0.85
DoMatrixTransposeFunction · 0.85

Calls 4

DoTransposeImplFunction · 0.85
dimsMethod · 0.45
beginMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected