Searched refs:expandedShapeMap (Results 1 – 1 of 1) sorted by relevance
494 return expandedShapeMap[i]; in getExpandedShapeOfDim()504 SmallVector<SmallVector<int64_t>> expandedShapeMap; member in __anon750e3d170611::ExpansionInfo525 expandedShapeMap.clear(); in compute()529 expandedShapeMap.resize(fusedIndexMap.getNumDims()); in compute()536 expandedShapeMap[pos].assign(shape.begin(), shape.end()); in compute()540 if (expandedShapeMap[i].empty()) in compute()541 expandedShapeMap[i] = {originalLoopExtent[i]}; in compute()