Searched refs:sourceShape (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 1449 auto sourceShape = sourceVectorType.getShape(); in verify() local
1452 destShape.size() - sourceShape.size(), 0); in verify()
1453 sourceShapeAsDestShape.append(sourceShape.begin(), sourceShape.end()); in verify()
2575 auto sourceShape = sourceVectorType.getShape(); in verifyVectorShapeCast() local
2580 sourceShape.begin(), sourceShape.end(), 1LL, std::multiplies<int64_t>{}); in verifyVectorShapeCast()
2590 if (!isValidShapeCast(sourceShape, resultShape)) in verifyVectorShapeCast()
2593 if (!isValidShapeCast(resultShape, sourceShape)) in verifyVectorShapeCast()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | FusionOnTensors.cpp | 287 ArrayRef<int64_t> sourceShape, in linearizeCollapsedDims() argument
303 sourceShape.slice(startDim, collapsedDims.size()), in linearizeCollapsedDims()
|