Home
last modified time | relevance | path

Searched refs:expandedType (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DReshapeOpsUtils.h103 static LogicalResult verifyReshapeLikeTypes(Op op, T expandedType, in verifyReshapeLikeTypes() argument
105 unsigned expandedRank = expandedType.getRank(); in verifyReshapeLikeTypes()
109 << expandedType in verifyReshapeLikeTypes()
119 if (llvm::any_of(expandedType.getShape(), in verifyReshapeLikeTypes()
139 return verifyReshapeLikeShapes(op, collapsedType, expandedType, isExpansion); in verifyReshapeLikeTypes()
157 ShapedType expandedType, in verifyReshapeLikeShapes() argument
161 collapsedType.getShape(), expandedType.getShape(), in verifyReshapeLikeShapes()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp693 RankedTensorType expandedType = isExpanding in fuseWithReshapeByExpansion() local
705 expandedType.getShape(), collapsedType.getShape(), rewriter))) in fuseWithReshapeByExpansion()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp865 RankedTensorType expandedType, in verifyTensorReshapeOp() argument
868 verifyReshapeLikeTypes(op, expandedType, collapsedType, isExpansion))) in verifyTensorReshapeOp()
873 computeTensorReshapeCollapsedType(expandedType, maps); in verifyTensorReshapeOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp673 auto expandedType = in matchAndRewrite() local
675 newLhs = rewriter.create<vector::BroadcastOp>(loc, expandedType, newLhs); in matchAndRewrite()
679 auto expandedType = in matchAndRewrite() local
681 newRhs = rewriter.create<vector::BroadcastOp>(loc, expandedType, newRhs); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DPolynomialApproximation.cpp122 auto expandedType = VectorType::get(expandedShape, eltType); in handleMultidimensionalVectors() local
124 builder.create<vector::ShapeCastOp>(expandedType, operand); in handleMultidimensionalVectors()