Lines Matching full:indices

49 /// Given the 'indices' of a load/store operation where the memref is a result
50 /// of a expand_shape op, returns the indices w.r.t to the source memref of the
65                                 ValueRange indices,
93   // Traverse all reassociation groups to determine the appropriate indices
96     assert(!groups.empty() && "association indices groups cannot be empty");
127       dynamicIndices[i] = indices[groups[i]];
129     // Supply suffix product results followed by load op indices as operands
152 /// Given the 'indices' of a load/store operation where the memref is a result
153 /// of a collapse_shape op, returns the indices w.r.t to the source memref of
168                                   ValueRange indices,
171   SmallVector<Value> tmp(indices.size());
174     assert(!groups.empty() && "association indices groups cannot be empty");
175     dynamicIndices.push_back(indices[cnt++]);
377                                const SmallVector<Value> &indices, Location loc,
380       llvm::map_range(indices, [](Value v) -> OpFoldResult { return v; })));
439   SmallVector<Value> indices(loadOp.getIndices().begin(),
442   // "actual" indices.
447         affineMap, indices, loadOp.getLoc(), rewriter);
448     indices.assign(expandedIndices.begin(), expandedIndices.end());
453       subViewOp.getMixedStrides(), subViewOp.getDroppedDims(), indices,
505   SmallVector<Value> indices(loadOp.getIndices().begin(),
508   // "actual" indices.
513         affineMap, indices, loadOp.getLoc(), rewriter);
514     indices.assign(expandedIndices.begin(), expandedIndices.end());
518           loadOp.getLoc(), rewriter, expandShapeOp, indices, sourceIndices)))
553   SmallVector<Value> indices(loadOp.getIndices().begin(),
556   // "actual" indices.
561         affineMap, indices, loadOp.getLoc(), rewriter);
562     indices.assign(expandedIndices.begin(), expandedIndices.end());
566           loadOp.getLoc(), rewriter, collapseShapeOp, indices, sourceIndices)))
606   SmallVector<Value> indices(storeOp.getIndices().begin(),
609   // "actual" indices.
614         affineMap, indices, storeOp.getLoc(), rewriter);
615     indices.assign(expandedIndices.begin(), expandedIndices.end());
620       subViewOp.getMixedStrides(), subViewOp.getDroppedDims(), indices,
668   SmallVector<Value> indices(storeOp.getIndices().begin(),
671   // "actual" indices.
676         affineMap, indices, storeOp.getLoc(), rewriter);
677     indices.assign(expandedIndices.begin(), expandedIndices.end());
681           storeOp.getLoc(), rewriter, expandShapeOp, indices, sourceIndices)))
717   SmallVector<Value> indices(storeOp.getIndices().begin(),
720   // "actual" indices.
725         affineMap, indices, storeOp.getLoc(), rewriter);
726     indices.assign(expandedIndices.begin(), expandedIndices.end());
730           storeOp.getLoc(), rewriter, collapseShapeOp, indices, sourceIndices)))
771   // If the source is a subview, we need to resolve the indices.
784   // If the destination is a subview, we need to resolve the indices.