Searched refs:vectorShape (Results 1 – 5 of 5) sorted by relevance
174 SmallVector<int64_t, 2> vectorShape; in matchAndRewrite() local177 vectorShape.push_back(flattenedParallelDim); in matchAndRewrite()181 vectorShape.push_back(flattenedReductionDim); in matchAndRewrite()183 if (!useInnerDimsForReduction && vectorShape.size() == 2) { in matchAndRewrite()185 std::swap(vectorShape.front(), vectorShape.back()); in matchAndRewrite()188 vectorShape, multiReductionOp.getSourceVectorType().getElementType()); in matchAndRewrite()
47 SmallVector<int64_t, 4> vectorShape; member in __anon77a0e1ec0111::DecomposeShapeIterator56 : vectorShape(targetShape.begin(), targetShape.end()), in DecomposeShapeIterator()98 computeElementOffsetsFromVectorSliceOffsets(vectorShape, vectorOffsets); in getVectorOffset()
2066 ArrayRef<int64_t> vectorShape = read.getVectorType().getShape(); in matchAndRewrite() local2067 SmallVector<int64_t, 4> unbroadcastedVectorShape(vectorShape.begin(), in matchAndRewrite()2068 vectorShape.end()); in matchAndRewrite()
42 static ArrayRef<int64_t> vectorShape(Type type) { in vectorShape() function47 static ArrayRef<int64_t> vectorShape(Value value) { in vectorShape() function48 return vectorShape(value.getType()); in vectorShape()210 ArrayRef<int64_t> shape = vectorShape(arg); in frexp()250 ArrayRef<int64_t> shape = vectorShape(arg); in exp2I32()274 ArrayRef<int64_t> shape = vectorShape(x); in makePolynomialCalculation()371 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in matchAndRewrite()429 ArrayRef<int64_t> shape = vectorShape(op.getResult()); in matchAndRewrite()495 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in matchAndRewrite()583 ArrayRef<int64_t> shape = vectorShape(op.getOperand()); in logMatchAndRewrite()[all …]
3848 ArrayRef<int64_t> vectorShape = transferOp.getVectorType().getShape(); in matchAndRewrite() local3851 if (transferOp.getMask() || !vectorShape.equals(resultShape)) { in matchAndRewrite()3861 newInBounds.push_back(en.value() == vectorShape[en.index()]); in matchAndRewrite()