Lines Matching defs:segments

111 static std::optional<unsigned> findSegment(ArrayAttr segments,
114   for (auto attr : segments) {
126                       std::optional<llvm::ArrayRef<int32_t>> segments,
133       nbOperandsBefore += (*segments)[i];
134     return range.drop_front(nbOperandsBefore).take_front((*segments)[*pos]);
142                    std::optional<llvm::ArrayRef<int32_t>> segments,
149       return getValuesFromSegments(deviceTypeAttr, operands, segments,
158                            std::optional<llvm::ArrayRef<int32_t>> segments,
162       getValuesFromSegments(deviceTypeAttr, operands, segments, deviceType);
959     Op op, OperandRange operands, DenseI32ArrayAttr segments,
964   if (segments) {
965     for (auto segCount : segments.asArrayRef()) {
977            << keyword << " operand count does not match count in segments";
1145     mlir::DenseI32ArrayAttr &segments) {
1180   segments = DenseI32ArrayAttr::get(parser.getContext(), seg);
1194                           std::optional<mlir::DenseI32ArrayAttr> segments) {
1199         llvm::seq<int32_t>(0, (*segments)[it.index()]), p, [&](auto it) {
1212     mlir::DenseI32ArrayAttr &segments) {
1249   segments = DenseI32ArrayAttr::get(parser.getContext(), seg);
1257     std::optional<mlir::DenseI32ArrayAttr> segments) {
1262         llvm::seq<int32_t>(0, (*segments)[it.index()]), p, [&](auto it) {
1275     mlir::DenseI32ArrayAttr &segments, mlir::ArrayAttr &hasDevNum,
1349   segments = DenseI32ArrayAttr::get(parser.getContext(), seg);
1369                             std::optional<mlir::DenseI32ArrayAttr> segments,
1389         llvm::seq<int32_t>(0, (*segments)[it.index()]), p, [&](auto it) {
1829     mlir::ArrayAttr &deviceType, mlir::DenseI32ArrayAttr &segments,
1948   segments = DenseI32ArrayAttr::get(parser.getContext(), seg);
1956                      std::optional<mlir::DenseI32ArrayAttr> segments,
1977           llvm::seq<int32_t>(0, (*segments)[it.index()]), p, [&](auto it) {
1998     std::optional<mlir::ArrayAttr> segments,
2000   if (!segments)
2002   for (auto attr : *segments) {