Lines Matching full:dimensions
230 /// Input dimensions of the schedule space, which represent free
241 /// Sizes of tensor dimensions for corresponding input dimensions of
277 /// Permute the two dimensions of the isl map.
279 /// Permute @p DstPos and @p SrcPos dimensions of the isl map @p Map that
283 /// @param DimType The type of the dimensions.
421 "functions have at least three dimensions.");
531 /// Permute two dimensions of the band node.
533 /// Permute FirstDim and SecondDim dimensions of the Node.
568 /// of dimensions of the band node and interchanging of two innermost
569 /// modified dimensions. The values of MacroKernelParams's fields are used
779 /// @param FirstDim, SecondDim The input dimensions that are used to define
864 // dimensions (to match the 3 band nodes surrounding the extension node) and
1024 /// Restore the initial ordering of dimensions of the band node
1026 /// In case the band node represents all the dimensions of the iteration
1028 /// dimensions.
1059 "functions have at least three dimensions.");
1090 /// 2. there are exactly three input dimensions.
1142 /// indexes I0, …, In are specified by @p Dimensions.
1146 /// @param Dimensions The permutation of the subset of the input dimensions.
1150 ArrayRef<int> Dimensions) {
1152 if (unsignedFromIslSize(Space.dim(isl::dim::out)) != Dimensions.size())
1157 // Il, …, In are specified by @p Dimensions.
1160 for (unsigned i = 0; i < Dimensions.size(); i++) {
1161 const int InPos = Dimensions[i];
1181 /// into @p IndexSet, @p DimensionSizes, and @p Dimensions, respectively.
1185 /// @param IndexSet The subset of the input dimensions.
1186 /// @param DimensionSizes Sizes of the input dimensions of @p Dimensions.
1187 /// @param Dimensions The permutation of the subset of the input dimensions.
1193 SmallVectorImpl<int> &Dimensions) {
1198 // Fix values of output dimensions with respect to their positions.
1199 // In the case of the tensor contraction, values of output dimensions are
1200 // fixed and form a permutation of a subset of values of input dimensions.
1204 // the output dimensions Stmt[1][j][0] -> A[0][1].
1206 // We store the permutation of the subset of the input dimensions {2, 0} into
1207 // @p Dimensions.
1220 // Try to obtain the permutation and sizes of corresponding input dimensions.
1221 Dimensions.assign(OutDimNum, -1);
1234 Dimensions[OutPos] = i;
1239 return isCorrectAccessMap(Domain, AccMap, Dimensions);
1317 /// @param Dimensions The permutation of the subset of the input dimensions.
1324 ArrayRef<int> Dimensions, TCInfoTy &TCI) {
1341 TCI.ADimensions.end(), Dimensions.begin(), Dimensions.end());
1353 TCI.BDimensions.end(), Dimensions.begin(), Dimensions.end());
1403 SmallVector<int> Dimensions;
1406 Dimensions))
1409 if (!setReadAccess(MemAccessPtr, IndexSet, IandJIndexSet, Dimensions, TCI))
1722 /// 4. Analyses the dependencies to determine contraction dimensions.