Searched defs:firstDimToCollapse (Results 1 – 1 of 1) sorted by relevance
494 collapseInnerDims(PatternRewriter & rewriter,mlir::Location loc,Value input,int64_t firstDimToCollapse) collapseInnerDims() argument 514 getCollapsedIndices(RewriterBase & rewriter,Location loc,ArrayRef<int64_t> shape,ValueRange indices,int64_t firstDimToCollapse) getCollapsedIndices() argument 615 int64_t firstDimToCollapse = sourceType.getRank() - vectorType.getRank(); matchAndRewrite() local 707 int64_t firstDimToCollapse = sourceType.getRank() - vectorType.getRank(); matchAndRewrite() local [all...]