Home
last modified time | relevance | path

Searched refs:dropDim (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp256 int64_t dropDim = oldType.getRank() - newType.getRank(); in matchAndRewrite() local
271 write.getLoc(), write.getVector(), splatZero(dropDim)); in matchAndRewrite()
302 int64_t dropDim = 1; in matchAndRewrite() local
389 splatZero(dropDim)) in matchAndRewrite()
417 int64_t dropDim = vecType.getRank() - newVecType.getRank(); in matchAndRewrite() local
422 op->getLoc(), operand, splatZero(dropDim))); in matchAndRewrite()
H A DVectorTransforms.cpp91 Type lowType = VectorType::Builder(type).dropDim(0); in reshapeLoad()
99 Type resType = VectorType::Builder(type).dropDim(index); in reshapeLoad()
127 Type lowType = VectorType::Builder(type).dropDim(0); in reshapeStore()
129 Type insType = VectorType::Builder(vType).dropDim(0); in reshapeStore()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp880 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()
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()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypes.h243 Builder &dropDim(unsigned pos) { in dropDim() function
306 Builder &dropDim(unsigned pos) { in dropDim() function