Home
last modified time | relevance | path

Searched defs:DoTranspose (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtranspose_op.cc189 Status TransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::TransposeCpuOp
196 Status ConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeCpuOp
227 Status ConjugateTransposeGpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::ConjugateTransposeGpuOp
H A Dtranspose_functor_gpu.cu.cc216 Status DoTranspose(const GPUDevice& device, const Tensor& in, in DoTranspose() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_transpose_op.cc108 Status MklTransposeCpuOp::DoTranspose(OpKernelContext* ctx, const Tensor& in, in DoTranspose() function in tensorflow::MklTransposeCpuOp
133 Status MklConjugateTransposeCpuOp::DoTranspose(OpKernelContext* ctx, in DoTranspose() function in tensorflow::MklConjugateTransposeCpuOp