Searched refs:aShape (Results 1 – 3 of 3) sorted by relevance
118 ArrayRef<int64_t> aShape = aVector.getShape(); in verify() local152 if (aShape[0] * aShape[1] * kThreads != m * k) in verify()176 if (!((aShape[0] == mTile * kTile) && (aShape[1] == numElementA))) in verify()
167 ArrayRef<int64_t> aShape = aT.getShape(), bShape = bT.getShape(); in getCastCompatibleMemRefType() local173 (aShape[idx] == bShape[idx]) ? aShape[idx] : ShapedType::kDynamicSize; in getCastCompatibleMemRefType()
1234 ArrayRef<int64_t> aShape, bShape, cShape; in verify() local1235 aShape = opTypes[A].getShape(); in verify()1239 if (aShape[1] != bShape[0] || aShape[0] != cShape[0] || in verify()