Searched refs:TransposeUsingTile (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_functor_gpu.cu.cc | 96 struct TransposeUsingTile { struct
145 struct TransposeUsingTile<complex64, conjugate> { struct
149 return TransposeUsingTile<uint64>::run(d, in, perm, out); in run()
151 return TransposeUsingTile<float2, true>::run(d, in, perm, out); in run()
157 struct TransposeUsingTile<complex128, conjugate> { struct
161 return TransposeUsingTile<float4>::run(d, in, perm, out); in run()
163 return TransposeUsingTile<double2, true>::run(d, in, perm, out); in run()
182 if (internal::TransposeUsingTile<T, conjugate>::run(d, in, perm, out)) { in run()
|