Searched refs:TransposeUsingTile (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_functor_gpu.cu.cc | 97 struct TransposeUsingTile { struct
146 struct TransposeUsingTile<complex64, conjugate> { struct
150 return TransposeUsingTile<uint64>::run(d, in, perm, out); in run()
152 return TransposeUsingTile<float2, true>::run(d, in, perm, out); in run()
158 struct TransposeUsingTile<complex128, conjugate> { struct
162 return TransposeUsingTile<float4>::run(d, in, perm, out); in run()
164 return TransposeUsingTile<double2, true>::run(d, in, perm, out); in run()
183 if (internal::TransposeUsingTile<T, conjugate>::run(d, in, perm, out)) { in run()
|