Searched refs:getSrcType (Results 1 – 5 of 5) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | FusionOnTensors.cpp | 318 RankedTensorType operandType = reshapeOp.getSrcType(); in isTensorReshapeOpFoldableByLinearization()
325 if ((asProducer && reshapeOp.getSrcType().hasStaticShape() && in isTensorReshapeOpFoldableByLinearization()
444 reshapeOp.getSrcType().getRank() < reshapeOp.getResultType().getRank(); in fuseWithReshapeByExpansion()
446 isExpanding ? reshapeOp.getResultType() : reshapeOp.getSrcType(); in fuseWithReshapeByExpansion()
763 if (reshapeOp.getSrcType().getRank() < in matchAndRewrite()
806 linearizeCollapsedDims(invMap, reshapeOp.getSrcType().getShape(), in matchAndRewrite()
850 if (reshapeOp.getSrcType().getRank() > reshapeOp.getResultType().getRank()) in matchAndRewrite()
|
| D | DropUnitDims.cpp | 425 RankedTensorType srcType = reshapeOp.getSrcType(), in matchAndRewrite()
427 parentSrcType = parentReshapeOp.getSrcType(); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 622 if (areReshapeOpsFoldable(reshapeOp.getResultType(), reshapeOp.getSrcType(), in matchAndRewrite()
623 srcReshapeOp.getSrcType())) { in matchAndRewrite()
632 if (areReshapeOpsFoldable(srcReshapeOp.getSrcType(), reshapeOp.getSrcType(), in matchAndRewrite()
654 if (reshapeSrcOp && reshapeSrcOp.getSrcType().hasStaticShape() && in foldReshapeOp()
656 reshapeSrcOp.getSrcType() == reshapeOp.getResultType()) in foldReshapeOp()
865 expandedType = op.getSrcType(); in verifyReshapeLikeTypes()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgOps.td | 158 MemRefType getSrcType() { return src().getType().cast<MemRefType>(); }
207 RankedTensorType getSrcType() {
|
| /external/mesa3d/src/gallium/drivers/nouveau/codegen/ |
| D | nv50_ir_lowering_nvc0.cpp | 2193 getSrcType(const TexInstruction::ImgFormatDesc *t, int c) in getSrcType() function
2293 getSrcType(format, i), untypedDst[i / 2]) in convertSurfaceFormat()
2297 getSrcType(format, i), untypedDst[0])->subOp = i; in convertSurfaceFormat()
2302 bld.mkCvt(OP_CVT, TYPE_F32, typedDst[i], getSrcType(format, i), typedDst[i]); in convertSurfaceFormat()
|