Lines Matching defs:fields

73                             MutableArrayRef<Value> fields,
79   scf::ForOp forOp = builder.create<scf::ForOp>(loc, lower, upper, one, fields);
80   for (unsigned i = 0, e = fields.size(); i < e; i++)
81     fields[i] = forOp.getRegionIterArg(i);
177                               /*out*/ SmallVectorImpl<Value> &fields) {
206   // Initializes all fields. An initial storage specifier and allocated
210       [&builder, &fields, stt, loc, posHeuristic, crdHeuristic, valHeuristic,
213         assert(fields.size() == fIdx);
233         fields.push_back(field);
238   // and gives all position fields an initial zero entry, so that it is
240   MutSparseTensorDescriptor desc(stt, fields);
332   // If present (fields unaffected, update pnext to plast).
341   // If !present (changes fields, update pnext).
355   // Update fields and return next pos.
473     // Extract fields and coordinates from args.
474     SmallVector<Value> fields = llvm::to_vector(args.drop_back(lvlRank + 1));
475     MutSparseTensorDescriptor desc(stt, fields);
520     return fields;
762       SmallVector<Value> fields;
763       fields.reserve(desc.getNumFields());
764       // Memcpy on memref fields.
771         fields.push_back(copied);
774       fields.push_back(desc.getSpecifier());
775       assert(fields.size() == desc.getNumFields());
776       rewriter.replaceOpWithMultiple(op, {fields});
792     SmallVector<Value> fields;
794                       sizeHint, lvlSizesValues, fields);
797     rewriter.replaceOpWithMultiple(op, {fields});
833     SmallVector<Value> fields;
835                       sizeHint, lvlSizesValues, fields);
838     rewriter.replaceOpWithMultiple(op, {fields});
962     SmallVector<Value> fields;
963     auto desc = getMutDescriptorFromTensorTuple(adaptor.getTensor(), fields,
1186     SmallVector<Value> fields;
1189         [&rewriter, &fields, srcDesc,
1194             fields.push_back(srcDesc.getSpecifier());
1222             fields.push_back(dstMem);
1227     rewriter.replaceOpWithMultiple(op, {fields});
1248     SmallVector<Value> fields;
1249     auto desc = getMutDescriptorFromTensorTuple(adaptor.getSource(), fields,
1314     SmallVector<Value> fields;
1318         [&rewriter, &fields, &op, &stt,
1321           assert(fields.size() == fIdx);
1323             fields.push_back(
1342             fields.push_back(mem);
1347     MutSparseTensorDescriptor desc(stt, fields);
1533     SmallVector<Value> fields;
1535                       lvlSizesValues, fields);
1538     MutSparseTensorDescriptor desc(dstTp, fields);
1590     rewriter.replaceOpWithMultiple(op, {fields});