Lines Matching defs:expandedShape
598 ArrayRef<int64_t> expandedShape,
627 ArrayRef<int64_t> expandedShape,
648 expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]);
682 ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i);
683 if (expandedShape.size() == 1)
685 for (int64_t shape : expandedShape.drop_front()) {
719 SmallVector<int64_t> expandedShape;
723 expandedShape.append(dimExpansion.begin(), dimExpansion.end());
725 return RankedTensorType::get(expandedShape, originalType.getElementType());
800 ArrayRef<int64_t> expandedShape = expansionInfo.getExpandedShapeOfDim(i);
801 if (expandedShape.size() == 1)
804 for (int64_t shape : expandedShape) {