Lines Matching defs:parallelDims

74     SmallVector<int64_t, 4> parallelDims;
77         parallelDims.push_back(i);
81     if (parallelDims.empty())
84         (parallelDims ==
85          llvm::to_vector<4>(llvm::seq<int64_t>(0, parallelDims.size()))))
89         (parallelDims == llvm::to_vector<4>(llvm::seq<int64_t>(
91                              parallelDims.size() + reductionDims.size()))))
96       indices.append(parallelDims.begin(), parallelDims.end());
100       indices.append(parallelDims.begin(), parallelDims.end());
183     SmallVector<int64_t, 4> parallelDims, parallelShapes;
195         parallelDims.push_back(i);
219     // Check parallelDims are exactly [0 .. size).
222         llvm::any_of(parallelDims, [&](int64_t i) { return i != counter++; }))
224     // Check parallelDims are exactly {reductionDims.size()} + [0 .. size).
227         llvm::any_of(parallelDims, [&](int64_t i) { return i != counter++; }))