| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_op.h | 24 class TransposeOp : public OpKernel {
26 explicit TransposeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in TransposeOp() function
36 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()
67 class ConjugateTransposeCpuOp : public TransposeOp {
70 : TransposeOp(ctx) {} in ConjugateTransposeCpuOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | layout_optimization.cc | 41 TransposeOp ReuseExistingTranspose(const OpOperand* operand, in ReuseExistingTranspose()
44 SmallVector<TransposeOp, 2>* transpose_ops) { in ReuseExistingTranspose() argument
54 TransposeOp transpose = tranpose_op; in ReuseExistingTranspose()
175 arg, builder.create<TransposeOp>(loc, op->getOperand(arg), arg_perm)); in runOnFunction()
185 auto transposed_res = builder.create<TransposeOp>(loc, result, res_perm); in runOnFunction()
198 SmallVector<TransposeOp, 2> transpose_ops; in MoveTransposeBefore()
208 TransposeOp transpose = dyn_cast<TransposeOp>(user); in MoveTransposeBefore()
263 result.setType(cast<TransposeOp>(*result.getUsers().begin()).y().getType()); in MoveTransposeBefore()
279 TransposeOp transpose = ReuseExistingTranspose( in MoveTransposeBefore()
284 builder.create<TransposeOp>(loc, operand.get(), permutation_op); in MoveTransposeBefore()
[all …]
|
| D | unroll_batch_matmul.h | 43 static TF::TransposeOp createTransposeOp(Value value, Location loc,
|
| D | unroll_batch_matmul.cc | 123 TF::TransposeOp ConvertTFBatchMatMulOp<BatchMatMulOpType>::createTransposeOp( in createTransposeOp()
151 return rewriter.create<TF::TransposeOp>(loc, transposed_type, value, perm_op); in createTransposeOp()
|
| D | einsum.cc | 52 TF::TransposeOp createTransposeOp(Value value, Location loc, in createTransposeOp()
67 return rewriter->create<TF::TransposeOp>(loc, transposed_type, value, in createTransposeOp()
|
| /external/llvm-project/mlir/examples/toy/Ch3/mlir/ |
| D | ToyCombine.cpp | 28 struct SimplifyRedundantTranspose : public mlir::OpRewritePattern<TransposeOp> {
33 : OpRewritePattern<TransposeOp>(context, /*benefit=*/1) {} in SimplifyRedundantTranspose()
39 matchAndRewrite(TransposeOp op, in matchAndRewrite()
43 TransposeOp transposeInputOp = transposeInput.getDefiningOp<TransposeOp>(); in matchAndRewrite()
57 void TransposeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| D | Dialect.cpp | 228 void TransposeOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
234 static mlir::LogicalResult verify(TransposeOp op) { in verify()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | ToyCombine.cpp | 33 struct SimplifyRedundantTranspose : public mlir::OpRewritePattern<TransposeOp> {
38 : OpRewritePattern<TransposeOp>(context, /*benefit=*/1) {} in SimplifyRedundantTranspose()
44 matchAndRewrite(TransposeOp op, in matchAndRewrite()
48 TransposeOp transposeInputOp = transposeInput.getDefiningOp<TransposeOp>(); in matchAndRewrite()
62 void TransposeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| D | Dialect.cpp | 309 void TransposeOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
315 void TransposeOp::inferShapes() { in inferShapes()
321 static mlir::LogicalResult verify(TransposeOp op) { in verify()
|
| /external/llvm-project/mlir/examples/toy/Ch5/mlir/ |
| D | ToyCombine.cpp | 33 struct SimplifyRedundantTranspose : public mlir::OpRewritePattern<TransposeOp> {
38 : OpRewritePattern<TransposeOp>(context, /*benefit=*/1) {} in SimplifyRedundantTranspose()
44 matchAndRewrite(TransposeOp op, in matchAndRewrite()
48 TransposeOp transposeInputOp = transposeInput.getDefiningOp<TransposeOp>(); in matchAndRewrite()
62 void TransposeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| D | Dialect.cpp | 309 void TransposeOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
315 void TransposeOp::inferShapes() { in inferShapes()
321 static mlir::LogicalResult verify(TransposeOp op) { in verify()
|
| /external/llvm-project/mlir/examples/toy/Ch4/mlir/ |
| D | ToyCombine.cpp | 33 struct SimplifyRedundantTranspose : public mlir::OpRewritePattern<TransposeOp> {
38 : OpRewritePattern<TransposeOp>(context, /*benefit=*/1) {} in SimplifyRedundantTranspose()
44 matchAndRewrite(TransposeOp op, in matchAndRewrite()
48 TransposeOp transposeInputOp = transposeInput.getDefiningOp<TransposeOp>(); in matchAndRewrite()
62 void TransposeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| D | Dialect.cpp | 309 void TransposeOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
315 void TransposeOp::inferShapes() { in inferShapes()
321 static mlir::LogicalResult verify(TransposeOp op) { in verify()
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | ToyCombine.cpp | 51 struct SimplifyRedundantTranspose : public mlir::OpRewritePattern<TransposeOp> {
56 : OpRewritePattern<TransposeOp>(context, /*benefit=*/1) {} in SimplifyRedundantTranspose()
62 matchAndRewrite(TransposeOp op, in matchAndRewrite()
66 TransposeOp transposeInputOp = transposeInput.getDefiningOp<TransposeOp>(); in matchAndRewrite()
80 void TransposeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| D | Dialect.cpp | 388 void TransposeOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
394 void TransposeOp::inferShapes() { in inferShapes()
400 static mlir::LogicalResult verify(TransposeOp op) { in verify()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | transpose_op.cc | 35 class TransposeOp : public XlaOpKernel { class
37 explicit TransposeOp(OpKernelConstruction* ctx, bool conjugate = false) in TransposeOp() function in tensorflow::__anon5c50effa0111::TransposeOp
100 class ConjugateTransposeOp : public TransposeOp {
103 : TransposeOp(ctx, /*conjugate=*/true) {} in ConjugateTransposeOp()
107 TransposeOp);
|
| /external/llvm-project/mlir/docs/Tutorials/Toy/ |
| D | Ch-3.md | 79 struct SimplifyRedundantTranspose : public mlir::OpRewritePattern<TransposeOp> {
84 : OpRewritePattern<TransposeOp>(context, /*benefit=*/1) {}
90 matchAndRewrite(TransposeOp op,
94 TransposeOp transposeInputOp = transposeInput.getDefiningOp<TransposeOp>();
116 void TransposeOp::getCanonicalizationPatterns(
147 adding a new trait, `NoSideEffect`, to our `TransposeOp`:
150 def TransposeOp : Toy_Op<"transpose", [NoSideEffect]> {...}
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 1240 class TransposeOpLowering : public OpRewritePattern<vector::TransposeOp> {
1242 using OpRewritePattern<vector::TransposeOp>::OpRewritePattern;
1246 : OpRewritePattern<vector::TransposeOp>(context), in TransposeOpLowering()
1249 LogicalResult matchAndRewrite(vector::TransposeOp op, in matchAndRewrite()
1720 lhs = rewriter.create<vector::TransposeOp>(loc, lhs, perm); in matchAndRewrite()
1724 lhs = rewriter.create<vector::TransposeOp>(loc, lhs, perm); in matchAndRewrite()
1725 rhs = rewriter.create<vector::TransposeOp>(loc, rhs, perm); in matchAndRewrite()
1732 rhs = rewriter.create<vector::TransposeOp>(loc, rhs, perm); in matchAndRewrite()
1737 rhs = rewriter.create<vector::TransposeOp>(loc, lhs, perm); in matchAndRewrite()
1743 rhs = rewriter.create<vector::TransposeOp>(loc, lhs, perm); in matchAndRewrite()
[all …]
|
| D | VectorOps.cpp | 676 auto transposeOp = extractOp.vector().getDefiningOp<vector::TransposeOp>(); in foldExtractOpFromTranspose()
740 auto transposeOp = extractOp.vector().getDefiningOp<vector::TransposeOp>(); in foldExtractOpFromInsertChainAndTranspose()
755 transposeOp = transposed.getDefiningOp<vector::TransposeOp>(); in foldExtractOpFromInsertChainAndTranspose()
775 transposeOp = insertionDest.getDefiningOp<vector::TransposeOp>(); in foldExtractOpFromInsertChainAndTranspose()
809 transposeOp = insertionDest.getDefiningOp<vector::TransposeOp>(); in foldExtractOpFromInsertChainAndTranspose()
2787 void vector::TransposeOp::build(OpBuilder &builder, OperationState &result, in build()
2802 OpFoldResult vector::TransposeOp::fold(ArrayRef<Attribute> operands) { in fold()
2816 static LogicalResult verify(vector::TransposeOp op) { in verify()
2844 class TransposeFolder final : public OpRewritePattern<vector::TransposeOp> {
2846 using OpRewritePattern<vector::TransposeOp>::OpRewritePattern;
[all …]
|
| /external/llvm-project/mlir/examples/toy/Ch2/mlir/ |
| D | Dialect.cpp | 228 void TransposeOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
234 static mlir::LogicalResult verify(TransposeOp op) { in verify()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/EDSC/ |
| D | Intrinsics.h | 34 using vector_transpose = ValueBuilder<vector::TransposeOp>;
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 146 in = rewriter.create<TransposeOp>(op.getLoc(), in, in matchAndRewrite()
150 out = rewriter.create<TransposeOp>(op.getLoc(), out, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_hlo_to_lhlo_op.h | 89 MAP_HLO_TO_LHLO(TransposeOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lower_general_dot.cc | 86 auto transpose_result = rewriter->create<mlir::mhlo::TransposeOp>( in TransposeReshape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2402 static LogicalResult Verify(TransposeOp op) { in Verify()
2453 void TransposeOp::build(OpBuilder &builder, OperationState &result, Value x, in build()
2458 return TransposeOp::build(builder, result, in build()
2478 return TransposeOp::build( in build()
2481 return TransposeOp::build(builder, result, UnrankedTensorType::get(etype), x, in build()
2487 OpFoldResult FoldIdentityTranspose(TransposeOp op) { in FoldIdentityTranspose()
2507 OpFoldResult FoldCancellableTranspose(TransposeOp op) { in FoldCancellableTranspose()
2509 auto transpose = dyn_cast_or_null<TF::TransposeOp>(op.x().getDefiningOp()); in FoldCancellableTranspose()
2527 OpFoldResult TransposeOp::fold(ArrayRef<Attribute> operands) { in fold()
|