Lines Matching defs:reassociation

527 ///  reassociation map helps compute the indexing maps of the modified op.
528 ///  For the above example, based on the reassociation map it
601   unsigned getOrigOpNumDims() const { return reassociation.size(); }
604     return reassociation[i];
614   SmallVector<ReassociationIndices> reassociation;
637   reassociation.clear();
656   // Compute reassociation map from the original op to the expanded op.
658   reassociation.reserve(fusedIndexMap.getNumDims());
661     reassociation.emplace_back(seq.begin(), seq.end());
728 /// Returns the reassociation maps to use in the `tensor.expand_shape`
737   SmallVector<ReassociationIndices> reassociation;
744     reassociation.emplace_back(std::move(indices));
747   return reassociation;
810                       "dims in the same reassociation group");
880         SmallVector<ReassociationIndices> reassociation =
887                 reassociation,
891             loc, expandedOperandType, opOperand->get(), reassociation));
905       SmallVector<ReassociationIndices> reassociation =
912               reassociation,
916           loc, expandedOutputType, opOperand.get(), reassociation));
947       SmallVector<ReassociationIndices> reassociation =
954           fusedOp->getResult(resultNumber), reassociation));
1145 /// reassociation are distinct.
1262                                  ArrayRef<ReassociationIndices> reassociation) {
1281   for (ReassociationIndicesRef foldedRangeDims : reassociation) {
1282     assert(!foldedRangeDims.empty() && "unexpected empty reassociation");
1459            "reassociation indices expected to have non-empty sets");
1493 /// Return the `reassociation` indices to use to collapse the operand when the
1531   // If the number of entries in the reassociation for the operand is same as
1748       SmallVector<ReassociationIndices> reassociation =
1755             loc, expandShapeResultType, collapsedOpResult, reassociation);
1758             loc, originalResultType, collapsedOpResult, reassociation);