Lines Matching defs:aT
132 /// Given two MemRefTypes `aT` and `bT`, return a MemRefType to which both can
135 /// 1. if `aT` and `bT` are cast-compatible, return `aT`.
138 /// a. keeping the ones that are static and equal across `aT` and `bT`.
141 static MemRefType getCastCompatibleMemRefType(MemRefType aT, MemRefType bT) {
142 if (memref::CastOp::areCastCompatible(aT, bT))
143 return aT;
144 if (aT.getRank() != bT.getRank())
148 if (failed(aT.getStridesAndOffset(aStrides, aOffset)) ||
153 ArrayRef<int64_t> aShape = aT.getShape(), bShape = bT.getShape();
155 SmallVector<int64_t, 4> resShape(aT.getRank(), 0),
157 for (int64_t idx = 0, e = aT.getRank(); idx < e; ++idx) {
165 resShape, aT.getElementType(),
166 StridedLayoutAttr::get(aT.getContext(), resOffset, resStrides));