Lines Matching defs:targetShape

282     ArrayRef<int64_t> targetShape, ArrayRef<ReassociationIndices> reassociation,
289                                                 targetShape);
299     auto targetType = MemRefType::get(targetShape, memrefType.getElementType(),
309                                                      targetShape);
319         RankedTensorType::get(targetShape, tensorType.getElementType());
334   SmallVector<int64_t> targetShape;
370     info.targetShape.push_back(operandShape[dim]);
482     targetShapes.push_back(replacementInfo.targetShape);
700     SmallVector<OpFoldResult> targetShape;
702       targetShape.push_back(rewriter.getIndexAttr(size));
703     auto reassociation = getReassociationMapForFoldingUnitDims(targetShape);
734     SmallVector<OpFoldResult> targetShape;
736       targetShape.push_back(rewriter.getIndexAttr(size));
737     auto reassociation = getReassociationMapForFoldingUnitDims(targetShape);