Searched refs:parallelShapes (Results 1 – 1 of 1) sorted by relevance
128 SmallVector<int64_t, 4> parallelDims, parallelShapes; in matchAndRewrite() local138 parallelShapes.push_back(srcShape[i]); in matchAndRewrite()145 if (!parallelShapes.empty()) { in matchAndRewrite()147 for (auto d : parallelShapes) in matchAndRewrite()205 if (parallelShapes.empty()) { in matchAndRewrite()212 parallelShapes, in matchAndRewrite()