Home
last modified time | relevance | path

Searched refs:TransposeUsingEigen (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtranspose_functor_cpu.cc76 internal::TransposeUsingEigen<CPUDevice, T, 2>(d, in, perm, conjugate, in run()
80 internal::TransposeUsingEigen<CPUDevice, T, 3>(d, in, perm, conjugate, in run()
84 internal::TransposeUsingEigen<CPUDevice, T, 4>(d, in, perm, conjugate, in run()
88 internal::TransposeUsingEigen<CPUDevice, T, 5>(d, in, perm, conjugate, in run()
92 internal::TransposeUsingEigen<CPUDevice, T, 6>(d, in, perm, conjugate, in run()
96 internal::TransposeUsingEigen<CPUDevice, T, 7>(d, in, perm, conjugate, in run()
100 internal::TransposeUsingEigen<CPUDevice, T, 8>(d, in, perm, conjugate, in run()
149 TransposeUsingEigen<SYCLDevice, T, 1>(d, in, perm, conjugate, out); in TransposeSYCL()
152 TransposeUsingEigen<SYCLDevice, T, 2>(d, in, perm, conjugate, out); in TransposeSYCL()
155 TransposeUsingEigen<SYCLDevice, T, 3>(d, in, perm, conjugate, out); in TransposeSYCL()
[all …]
Dtranspose_functor.h147 void TransposeUsingEigen(const Device& d, const Tensor& in, in TransposeUsingEigen() function
Dtranspose_functor_gpu.cu.cc173 internal::TransposeUsingEigen<GPUDevice, T, DIM>(d, in, perm, conjugate, \