Searched defs:TransposeOp (Results 1 – 3 of 3) sorted by relevance
35 class TransposeOp : public XlaOpKernel { class37 explicit TransposeOp(OpKernelConstruction* ctx, bool conjugate = false) in TransposeOp() function in tensorflow::__anon5623cd890111::TransposeOp
26 explicit TransposeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in TransposeOp() function
48 SmallVector<TransposeOp, 2>* transpose_ops) { in ReuseExistingTranspose()