Searched refs:bShape (Results 1 – 3 of 3) sorted by relevance
119 ArrayRef<int64_t> bShape = bVector.getShape(); in verify() local157 if (bShape[0] * bShape[1] * kThreads != k * n) in verify()181 if (!((bShape[0] == kTile * nTile) && (bShape[1] == numElementB))) in verify()
1234 ArrayRef<int64_t> aShape, bShape, cShape; in verify() local1236 bShape = opTypes[B].getShape(); in verify()1239 if (aShape[1] != bShape[0] || aShape[0] != cShape[0] || in verify()1240 bShape[1] != cShape[1]) 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()