Home | History | Annotate | Download | only in kernels

Lines Matching refs:Transpose

27 // Transpose tensor 'in' into tensor 'out' according to dimension
38 // Conjugate and transpose tensor 'in' into tensor 'out' according to dimension
62 struct Transpose {
145 // Uses Eigen to transpose.
179 Transpose<Device, uint8>::run(d, in, perm, out);
188 Transpose<Device, uint16>::run(d, in, perm, out);
194 Transpose<Device, uint32>::run(d, in, perm, out);
199 Transpose<Device, uint64>::run(d, in, perm, out);
207 "Conjugate transpose of complex64 not supported for GCC on "
210 Transpose<Device, complex64, /*conjugate=*/true>::run(d, in, perm, out);
213 Transpose<Device, uint64>::run(d, in, perm, out);
219 Transpose<Device, complex128, /*conjugate=*/true>::run(d, in, perm,
222 Transpose<Device, complex128, /*conjugate=*/false>::run(d, in, perm,
228 Transpose<Device, string>::run(d, in, perm, out);