Home
last modified time | relevance | path

Searched refs:RTTBuilder (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp878 using RTTBuilder = RankedTensorType::Builder; in returningMatchAndRewrite() typedef
880 RTTBuilder(inputType).dropDim((removeH ? 1 : 2)); in returningMatchAndRewrite()
882 RTTBuilder(kernelType).dropDim((removeH ? 0 : 1)); in returningMatchAndRewrite()
884 RTTBuilder(outputType).dropDim(removeH ? 1 : 2); in returningMatchAndRewrite()
948 using RTTBuilder = RankedTensorType::Builder; in returningMatchAndRewrite() typedef
950 RTTBuilder(inputType).dropDim((removeH ? 1 : 2)); in returningMatchAndRewrite()
952 RTTBuilder(kernelType).dropDim((removeH ? 0 : 1)); in returningMatchAndRewrite()
954 RTTBuilder(outputType).dropDim(removeH ? 1 : 2); in returningMatchAndRewrite()
H A DHoistPadding.cpp162 using RTTBuilder = RankedTensorType::Builder; in computeTransposedType() typedef
164 RTTBuilder(rankedTensorType).setShape(transposedShape); in computeTransposedType()