Searched refs:bT (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferSplitRewritePatterns.cpp | 155 static MemRefType getCastCompatibleMemRefType(MemRefType aT, MemRefType bT) { in getCastCompatibleMemRefType() argument
156 if (memref::CastOp::areCastCompatible(aT, bT)) in getCastCompatibleMemRefType()
158 if (aT.getRank() != bT.getRank()) in getCastCompatibleMemRefType()
163 failed(getStridesAndOffset(bT, bStrides, bOffset)) || in getCastCompatibleMemRefType()
167 ArrayRef<int64_t> aShape = aT.getShape(), bShape = bT.getShape(); in getCastCompatibleMemRefType()
170 resStrides(bT.getRank(), 0); in getCastCompatibleMemRefType()
|
| /llvm-project-15.0.7/libclc/generic/lib/math/ |
| H A D | cbrt.cl | 77 float bT = mad(cbrtH, remT, mad(cbrtT, remH, cbrtT*remT));
79 float z = mad(poly, bH, mad(poly, bT, bT)) + bH;
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 570 auto bT = b.dyn_cast<MemRefType>(); in areCastCompatible() local
575 if (aT && bT) { in areCastCompatible()
576 if (aT.getElementType() != bT.getElementType()) in areCastCompatible()
578 if (aT.getLayout() != bT.getLayout()) { in areCastCompatible()
582 failed(getStridesAndOffset(bT, bStrides, bOffset)) || in areCastCompatible()
600 if (aT.getMemorySpace() != bT.getMemorySpace()) in areCastCompatible()
604 if (aT.getRank() != bT.getRank()) in areCastCompatible()
608 int64_t aDim = aT.getDimSize(i), bDim = bT.getDimSize(i); in areCastCompatible()
616 if (!bT && !ubT) in areCastCompatible()
623 auto bEltType = (bT) ? bT.getElementType() : ubT.getElementType(); in areCastCompatible()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 149 auto bT = b.dyn_cast<TensorType>(); in areCastCompatible() local
150 if (!aT || !bT) in areCastCompatible()
153 if (aT.getElementType() != bT.getElementType()) in areCastCompatible()
156 return succeeded(verifyCompatibleShape(aT, bT)); in areCastCompatible()
|