Searched defs:reassociationIndices (Results 1 – 4 of 4) sorted by relevance
95 SmallVector<ReassociationIndices> reassociationIndices = in create() local124 const SmallVector<ReassociationIndices> reassociationIndices = in emitLoopNestBody() local162 SmallVector<ReassociationIndices> reassociationIndices = in simplifyCollapseShapeWithRankReducingExtractSlice() local
134 MLIRContext *context, ArrayRef<ReassociationIndices> reassociationIndices) { in convertReassociationIndicesToExprs() argument 172 SmallVector<ReassociationIndices, 2> reassociationIndices; in convertReassociationMapsToIndices() local 287 ArrayRef<ReassociationIndices> reassociationIndices) { in getLinearizedDimensions() argument 374 ArrayRef<ReassociationIndices> reassociationIndices) { in getCollapseShapeTrivialSegments() argument 387 canCollapseShapeBeSimplifiedByRankReducingSlice(RankedTensorType sourceType,ArrayRef<ReassociationIndices> reassociationIndices) canCollapseShapeBeSimplifiedByRankReducingSlice() argument 400 getSimplifyCollapseShapeWithRankReducingSliceInfo(RankedTensorType sourceType,ArrayRef<ReassociationIndices> reassociationIndices) getSimplifyCollapseShapeWithRankReducingSliceInfo() argument [all...]
205 std::optional<SmallVector<ReassociationIndices>> reassociationIndices = in matchAndRewrite() local 462 SliceFromCollapseHelper(ArrayRef<ReassociationIndices> reassociationIndices,ArrayRef<OpFoldResult> collapseShapeInputShape,ArrayRef<OpFoldResult> collapseShapeOutputShape,ArrayRef<Range> extractSliceParams) SliceFromCollapseHelper() argument [all...]
603 SmallVector<SmallVector<int64_t, 2>> reassociationIndices( expandRank() local