| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_op.h | 31 virtual Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
41 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
51 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
61 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
71 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
83 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
95 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
107 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
119 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
|
| D | transpose_op.cc | 201 OP_REQUIRES_OK(ctx, DoTranspose(ctx, input, permutation, output)); in Compute()
205 Status TransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::TransposeCpuOp
208 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, in DoTranspose()
212 Status ConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeCpuOp
252 Status TransposeGpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in TF_CALL_ALL_TYPES()
255 return ::tensorflow::DoTranspose(ctx->eigen_device<GPUDevice>(), in, perm, in TF_CALL_ALL_TYPES()
258 Status ConjugateTransposeGpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeGpuOp
283 Status TransposeSyclOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::TransposeSyclOp
286 return ::tensorflow::DoTranspose(ctx->eigen_device<SYCLDevice>(), in, perm, in DoTranspose()
289 Status ConjugateTransposeSyclOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeSyclOp
|
| D | mkl_transpose_op.cc | 157 Status MklTransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::MklTransposeCpuOp
195 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, in DoTranspose()
199 Status MklConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::MklConjugateTransposeCpuOp
|
| D | transpose_functor_cpu.cc | 112 Status DoTranspose(const DEVICE& device, const Tensor& in, \
|
| D | transpose_functor.h | 35 Status DoTranspose(const Device& device, const Tensor& in,
|
| D | transpose_functor_gpu.cu.cc | 215 Status DoTranspose(const GPUDevice& device, const Tensor& in, in DoTranspose() function
|
| D | reduction_ops_common.h | 226 ctx, DoTranspose(d, data_reshaped, helper.permutation(), &shuffled));
|