Searched refs:rhsReductionDims (Results 1 – 1 of 1) sorted by relevance
619 SmallVector<int64_t> rhsReductionDims = in matchAndRewrite() local626 for (int64_t dim : rhsReductionDims) { in matchAndRewrite()635 numParallelDims - (rhsMap.getNumResults() - rhsReductionDims.size()); in matchAndRewrite()642 for (int64_t dim : rhsReductionDims) in matchAndRewrite()687 SmallVector<int64_t, 4> rhsOffsets(rhsReductionDims.size(), 0); in matchAndRewrite()