Searched refs:TransposeSimple (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_functor_cpu.cc | 35 void TransposeSimple(const CPUDevice& device, const Tensor& in, in TransposeSimple() function
104 TransposeSimple<T, conjugate>(d, in, perm, out); in run()
|
| D | transpose_functor_gpu.cu.cc | 57 void TransposeSimple(const GPUDevice& d, const Tensor& in, in TransposeSimple() function
195 internal::TransposeSimple<T, conjugate>(d, in, perm, out); in run()
|