| /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,
73 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
85 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
97 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
|
| D | transpose_op.cc | 186 OP_REQUIRES_OK(ctx, DoTranspose(ctx, input, permutation, output)); in Compute()
190 Status TransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::TransposeCpuOp
193 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, in DoTranspose()
197 Status ConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeCpuOp
222 Status TransposeGpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in TF_CALL_ALL_TYPES()
225 return ::tensorflow::DoTranspose(ctx->eigen_device<GPUDevice>(), in, perm, in TF_CALL_ALL_TYPES()
228 Status ConjugateTransposeGpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeGpuOp
|
| D | transpose_functor_cpu.cc | 112 Status DoTranspose(const DEVICE& device, const Tensor& in, \
|
| D | fused_batch_norm_op.cc | 139 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()()
220 const Status s = ::tensorflow::DoTranspose( in operator ()()
277 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()()
331 const Status s = ::tensorflow::DoTranspose( in operator ()()
374 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()()
377 OP_REQUIRES_OK(context, ::tensorflow::DoTranspose( in operator ()()
508 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()()
|
| D | transpose_functor.h | 35 Status DoTranspose(const Device& device, const Tensor& in,
|
| D | transpose_functor_gpu.cu.cc | 216 Status DoTranspose(const GPUDevice& device, const Tensor& in, in DoTranspose() function
|
| D | reduction_ops_common.h | 228 OP_REQUIRES_OK(ctx, DoTranspose(d, data_reshaped, helper.permutation(),
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_transpose_op.cc | 167 Status MklTransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::MklTransposeCpuOp
208 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, in DoTranspose()
212 Status MklConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::MklConjugateTransposeCpuOp
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 355 DoTranspose(device, lhs, gtl::ArraySlice<int>(perm.data(), ndims), in TransposeLhsForGtsvBatched()
|
| D | einsum_op_impl.h | 354 TF_RETURN_IF_ERROR(DoTranspose(device, input, permutation, output)); in TransposeOperand()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 107 TF_RETURN_IF_ERROR(DoTranspose<CPUDevice>(ctx->eigen_device<CPUDevice>(), in TransposeTensor()
|