Home
last modified time | relevance | path

Searched refs:expandedDims (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp648 ArrayRef<int64_t> expandedDims = in updateExpandedGenericOpRegion() local
650 assert(!expandedDims.empty() && "expected valid expansion info"); in updateExpandedGenericOpRegion()
653 if (expandedDims.size() == 1 && in updateExpandedGenericOpRegion()
654 expandedDims.front() == (int64_t)indexOp.dim()) in updateExpandedGenericOpRegion()
663 expandedIndices.reserve(expandedDims.size() - 1); in updateExpandedGenericOpRegion()
665 expandedDims.drop_front(), std::back_inserter(expandedIndices), in updateExpandedGenericOpRegion()
667 Value newIndex = rewriter.create<IndexOp>(loc, expandedDims.front()); in updateExpandedGenericOpRegion()