HomeSort by relevance Sort by last modified time
    Searched defs:DoTranspose (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_transpose_op.cc 92 Status MklTransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in,
114 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm,
118 Status MklConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx,
transpose_op.cc 201 OP_REQUIRES_OK(ctx, DoTranspose(ctx, input, permutation, output));
205 Status TransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in,
208 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm,
212 Status ConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx,
254 Status TransposeGpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in,
257 return ::tensorflow::DoTranspose(ctx->eigen_device<GPUDevice>(), in, perm,
260 Status ConjugateTransposeGpuOp::DoTranspose(OpKernelContext* ctx,
285 Status TransposeSyclOp::DoTranspose(OpKernelContext* ctx, const Tensor& in,
288 return ::tensorflow::DoTranspose(ctx->eigen_device<SYCLDevice>(), in, perm,
291 Status ConjugateTransposeSyclOp::DoTranspose(OpKernelContext* ctx
    [all...]
transpose_functor_gpu.cu.cc 244 Status DoTranspose(const GPUDevice& device, const Tensor& in,

Completed in 750 milliseconds