Searched refs:TransposeOp (Results 1 – 3 of 3) sorted by relevance
24 class TransposeOp : public OpKernel {26 explicit TransposeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in TransposeOp() function36 class TransposeCpuOp : public TransposeOp {38 explicit TransposeCpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in TransposeCpuOp()46 class MklTransposeCpuOp : public TransposeOp {48 explicit MklTransposeCpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in MklTransposeCpuOp()56 class TransposeGpuOp : public TransposeOp {58 explicit TransposeGpuOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in TransposeGpuOp()66 class TransposeSyclOp : public TransposeOp {68 explicit TransposeSyclOp(OpKernelConstruction* ctx) : TransposeOp(ctx) {} in TransposeSyclOp()[all …]
144 void TransposeOp::Compute(OpKernelContext* ctx) { in Compute()
34 class TransposeOp : public XlaOpKernel { class36 explicit TransposeOp(OpKernelConstruction* ctx, bool conjugate = false) in TransposeOp() function in tensorflow::__anon3956a3dc0111::TransposeOp99 class ConjugateTransposeOp : public TransposeOp {102 : TransposeOp(ctx, /*conjugate=*/true) {} in ConjugateTransposeOp()106 TransposeOp);