Searched refs:createTransposeOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 123 TF::TransposeOp ConvertTFBatchMatMulOp<BatchMatMulOpType>::createTransposeOp( in createTransposeOp() function in mlir::TF::ConvertTFBatchMatMulOp
229 input_lhs = createTransposeOp(input_lhs, loc, rewriter); in matchAndRewrite()
237 input_rhs = createTransposeOp(input_rhs, loc, rewriter); in matchAndRewrite()
|
| D | einsum.cc | 52 TF::TransposeOp createTransposeOp(Value value, Location loc, in createTransposeOp() function
229 *lhs = createTransposeOp(*lhs, loc, lhs_transpose, rewriter); in transposeForBatchMatmul()
230 *rhs = createTransposeOp(*rhs, loc, rhs_transpose, rewriter); in transposeForBatchMatmul()
342 out = createTransposeOp(out, op.getLoc(), out_transpose, &rewriter); in rewriteToBatchMatmul()
|
| D | unroll_batch_matmul.h | 43 static TF::TransposeOp createTransposeOp(Value value, Location loc,
|