Lines Matching refs:GPUDevice
29 typedef Eigen::GpuDevice GPUDevice;
57 void TransposeSimple(const GPUDevice& d, const Tensor& in,
97 static bool run(const Eigen::GpuDevice& d, const Tensor& in,
115 tensorflow::functor::SwapDimension1And2InTensor3<GPUDevice, T,
123 tensorflow::functor::SwapDimension1And2InTensor3<GPUDevice, T,
128 tensorflow::functor::SwapDimension0And2InTensor3<GPUDevice, T,
146 static bool run(const Eigen::GpuDevice& d, const Tensor& in,
158 static bool run(const Eigen::GpuDevice& d, const Tensor& in,
172 struct Transpose<GPUDevice, T, conjugate> {
173 static void run(const GPUDevice& d, const Tensor& in,
179 internal::TransposeUsingEigen<GPUDevice, T, 2>(d, in, perm, conjugate,
186 internal::TransposeUsingEigen<GPUDevice, T, 3>(d, in, perm, conjugate,
193 internal::TransposeUsingEigen<GPUDevice, T, 4>(d, in, perm, conjugate,
200 internal::TransposeUsingEigen<GPUDevice, T, 5>(d, in, perm, conjugate,
207 internal::TransposeUsingEigen<GPUDevice, T, 6>(d, in, perm, conjugate,
214 internal::TransposeUsingEigen<GPUDevice, T, 7>(d, in, perm, conjugate,
221 internal::TransposeUsingEigen<GPUDevice, T, 8>(d, in, perm, conjugate,
233 struct Transpose<GPUDevice, string, conjugate> {
234 static void run(const GPUDevice& d, const Tensor& in,
241 template struct Transpose<GPUDevice, string, false>;
244 Status DoTranspose(const GPUDevice& device, const Tensor& in,
249 Status DoConjugateTranspose(const GPUDevice& device, const Tensor& in,
254 Status DoMatrixTranspose(const GPUDevice& device, const Tensor& in,
259 Status DoConjugateMatrixTranspose(const GPUDevice& device, const Tensor& in,
