| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | TypeUtilities.cpp | 24 Type mlir::getElementTypeOrSelf(Type type) { in getElementTypeOrSelf() function in mlir
30 Type mlir::getElementTypeOrSelf(Value val) { in getElementTypeOrSelf() function in mlir
31 return getElementTypeOrSelf(val.getType()); in getElementTypeOrSelf()
34 Type mlir::getElementTypeOrSelf(Attribute attr) { in getElementTypeOrSelf() function in mlir
35 return getElementTypeOrSelf(attr.getType()); in getElementTypeOrSelf()
|
| H A D | Operation.cpp | 850 auto elementType = getElementTypeOrSelf(op->getOperand(0)); in verifySameOperandsElementType()
853 if (getElementTypeOrSelf(operand) != elementType) in verifySameOperandsElementType()
866 auto elementType = getElementTypeOrSelf(op->getResult(0)); in verifySameOperandsAndResultElementType()
870 if (getElementTypeOrSelf(result) != elementType) in verifySameOperandsAndResultElementType()
877 if (getElementTypeOrSelf(operand) != elementType) in verifySameOperandsAndResultElementType()
891 auto elementType = getElementTypeOrSelf(type); in verifySameOperandsAndResultType()
893 if (getElementTypeOrSelf(resultType) != elementType || in verifySameOperandsAndResultType()
899 if (getElementTypeOrSelf(opType) != elementType || in verifySameOperandsAndResultType()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | TypeUtilities.h | 31 Type getElementTypeOrSelf(Type type);
34 Type getElementTypeOrSelf(Attribute attr);
35 Type getElementTypeOrSelf(Value val);
|
| H A D | OpBase.td | 2259 class ElementType<string name> : StrFunc<"getElementTypeOrSelf($" # name # ")">;
2322 SubstLeaves<"$_self", "getElementTypeOrSelf($_op.getOperand(" # idx # "))",
2336 SubstLeaves<"$_self", "getElementTypeOrSelf($" # name # ")",
2351 CPred<"::mlir::getElementTypeOrSelf($_op.getOperand(" # i # ")) == "
2352 "::mlir::getElementTypeOrSelf($_op.getOperand(" # j # "))">]>;
2373 CPred<"getElementTypeOrSelf($_op.getResult(" # i # ")) == "
2374 "getElementTypeOrSelf($_op.getOperand(" # j # "))">;
2402 "[this](unsigned i) { return getElementTypeOrSelf(this->getOperand(i)); "
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/ |
| H A D | PolynomialApproximation.cpp | 307 if (getElementTypeOrSelf(origType).isF32() || in insertCasts()
368 if (!getElementTypeOrSelf(operand).isF32()) in matchAndRewrite()
425 if (!getElementTypeOrSelf(x).isF32()) in matchAndRewrite()
492 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in matchAndRewrite()
580 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in logMatchAndRewrite()
727 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in matchAndRewrite()
774 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in matchAndRewrite()
902 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in matchAndRewrite()
1028 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in matchAndRewrite()
1098 if (!getElementTypeOrSelf(op.getOperand()).isF32()) in matchAndRewrite()
[all …]
|
| H A D | AlgebraicSimplification.cpp | 95 loc, rewriter.getFloatAttr(getElementTypeOrSelf(op.getType()), 1.0)); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MathToLLVM/ |
| H A D | MathToLLVM.cpp | 105 auto floatType = getElementTypeOrSelf(resultType).cast<FloatType>(); in matchAndRewrite()
158 auto floatType = getElementTypeOrSelf(resultType).cast<FloatType>(); in matchAndRewrite()
212 auto floatType = getElementTypeOrSelf(resultType).cast<FloatType>(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/Dialect/ |
| H A D | Linalg.cpp | 33 argTypes.push_back(getElementTypeOrSelf(opOperand->get().getType())); in mlirLinalgFillBuiltinNamedOpRegion()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/ |
| H A D | Traits.cpp | 137 elementType = getElementTypeOrSelf(type1); in getBroadcastedType()
138 if (elementType != getElementTypeOrSelf(type2)) in getBroadcastedType()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToLLVM/ |
| H A D | ArithmeticToLLVM.cpp | 136 typeConverter->convertType(getElementTypeOrSelf(op.getResult())) in matchAndRewrite()
139 getElementTypeOrSelf(adaptor.getIn()).cast<IntegerType>(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToArith/ |
| H A D | TosaToArith.cpp | 133 if (getElementTypeOrSelf(value.getType()).getIntOrFloatBitWidth() > 32) { in matchAndRewrite()
236 if (!getElementTypeOrSelf(resultTy).isInteger(32)) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 166 VectorType::get(shape, getElementTypeOrSelf(value)); in broadcastIfNeeded()
208 shape, getElementTypeOrSelf(outputOperand->get().getType())); in buildVectorWrite()
478 getElementTypeOrSelf(opOperand->get())); in vectorizeAsLinalgGeneric()
483 getElementTypeOrSelf(opOperand->get())); in vectorizeAsLinalgGeneric()
631 VectorType::get(srcType.getShape(), getElementTypeOrSelf(srcType)); in vectorizeCopy()
633 VectorType::get(dstType.getShape(), getElementTypeOrSelf(dstType)); in vectorizeCopy()
|
| H A D | DropUnitDims.cpp | 284 Type elementType = getElementTypeOrSelf(opOperand->get()); in replaceUnitExtents()
|
| H A D | Transforms.cpp | 248 paddedShape, getElementTypeOrSelf(opOperand->get())); in padOperandToSmallestStaticBoundingBox()
|
| /llvm-project-15.0.7/mlir/test/mlir-tblgen/ |
| H A D | predicate.td | 107 // CHECK-SAME: [this](unsigned i) { return getElementTypeOrSelf(this->getOperand(i)); }))
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/IR/ |
| H A D | ArithmeticOps.cpp | 798 auto underlyingType = getElementTypeOrSelf(type); in getUnderlyingType()
832 Type srcType = getElementTypeOrSelf(op.getIn().getType()); in verifyExtOp()
833 Type dstType = getElementTypeOrSelf(op.getType()); in verifyExtOp()
845 Type srcType = getElementTypeOrSelf(op.getIn().getType()); in verifyTruncateOp()
846 Type dstType = getElementTypeOrSelf(op.getType()); in verifyTruncateOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorMultiDimReductionTransforms.cpp | 243 Type elementType = getElementTypeOrSelf(multiReductionOp.getDestType()); in matchAndRewrite()
|
| H A D | VectorTransforms.cpp | 1217 Type castResTy = getElementTypeOrSelf(op->getResult(0)); in matchAndRewrite()
1422 getElementTypeOrSelf(op.getAcc().getType())), in matchAndRewrite()
1783 getElementTypeOrSelf(op.getAccType()) || in matchAndRewrite()
1784 op.getRhsType().getElementType() != getElementTypeOrSelf(op.getAccType())) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MathToSPIRV/ |
| H A D | MathToSPIRV.cpp | 247 auto ety = getElementTypeOrSelf(ty); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 1689 getElementTypeOrSelf(srcType) == getElementTypeOrSelf(dstVectorType)) in isBroadcastableTo()
4519 getElementTypeOrSelf(getElementTypeOrSelf(memRefType))); in build()
4536 if (getElementTypeOrSelf(getElementTypeOrSelf(sourceType)) != in verify()
4537 getElementTypeOrSelf(getElementTypeOrSelf(resultType))) in verify()
5082 Type t1 = getElementTypeOrSelf(v1.getType()); in makeArithReduction()
5083 Type t2 = getElementTypeOrSelf(v2.getType()); in makeArithReduction()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/ |
| H A D | WmmaOpsToNvvm.cpp | 282 auto floatType = getElementTypeOrSelf(lhs.getType()).cast<FloatType>(); in createMinMaxF()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1100 if (getElementTypeOrSelf(srcType) != getElementTypeOrSelf(dstType)) in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 71 argTypes.push_back(getElementTypeOrSelf(t)); in fillStructuredOpRegion()
684 blockArgTypes.push_back(getElementTypeOrSelf(v)); in build()
1498 Type elementType = getElementTypeOrSelf(outputOperand->get().getType()); in verifyYield()
|
| H A D | LinalgInterfaces.cpp | 743 Type elementType = getElementTypeOrSelf(opOperand->get()); in verifyStructuredOpInterface()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 534 Type elementType = getElementTypeOrSelf(std::get<1>(it)); in applyToOne()
|