Searched refs:reductionShapes (Results 1 – 1 of 1) sorted by relevance
129 SmallVector<int64_t, 4> reductionDims, reductionShapes; in matchAndRewrite() local135 reductionShapes.push_back(srcShape[i]); in matchAndRewrite()150 if (!reductionShapes.empty()) { in matchAndRewrite()152 for (auto d : reductionShapes) in matchAndRewrite()