Lines Matching defs:indexingMaps

192                           ArrayRef<AffineMap> indexingMaps) {
193   // Initialize indexingMaps attribute, for MatmulOp.
1000     ValueRange inputs, ValueRange outputs, ArrayAttr indexingMaps,
1004   build(builder, result, resultTensorTypes, inputs, outputs, indexingMaps,
1014     ValueRange inputs, ValueRange outputs, ArrayRef<AffineMap> indexingMaps,
1020         builder.getAffineMapArrayAttr(indexingMaps),
1033     ValueRange outputs, ArrayRef<AffineMap> indexingMaps,
1038   build(builder, result, TypeRange{}, inputs, outputs, indexingMaps,
1044     ValueRange outputs, ArrayRef<AffineMap> indexingMaps,
1048   build(builder, result, inputs, outputs, indexingMaps, iteratorTypes,
1055     ValueRange inputs, ValueRange outputs, ArrayRef<AffineMap> indexingMaps,
1059   build(builder, result, resultTensorTypes, inputs, outputs, indexingMaps,
2780   SmallVector<AffineMap> indexingMaps{identityMap, reductionMap};
2781   return std::make_tuple(iteratorTypes, indexingMaps);
2792   auto [iteratorTypes, indexingMaps] =
2794   assert(indexingMaps.size() == 2 &&
2796   assert(indexingMaps[0].isIdentity() && "input map should be identity");
2799       loc, output.getType(), input, output, indexingMaps, iteratorTypes,
2815   auto [iteratorTypes, indexingMaps] = computeIteratorTypesAndIndexingMaps(
2817   assert(indexingMaps.size() == 2 && "We should have one map for each input");
2818   assert(indexingMaps[0].isIdentity() && "input map should be identity");
2820   indexingMaps.push_back(indexingMaps[0]);
2822       loc, input.getType(), ValueRange{input, max}, output, indexingMaps,
2841   auto [iteratorTypes, indexingMaps] = computeIteratorTypesAndIndexingMaps(
2843   assert(indexingMaps.size() == 2 &&
2845   assert(indexingMaps[0].isIdentity() && "Numerator map should be identity");
2847   indexingMaps.push_back(indexingMaps[0]);
2850       indexingMaps, iteratorTypes,
3466   SmallVector<AffineMap> indexingMaps;
3468   indexingMaps.push_back(AffineMap::get(3, 0, {d0, d2}, context));
3469   indexingMaps.push_back(AffineMap::get(3, 0, {d2, d1}, context));
3470   indexingMaps.push_back(AffineMap::get(3, 0, {d0, d1}, context));
3471   return indexingMaps;
3573   SmallVector<Attribute, 3> indexingMaps = llvm::map_to_vector(
3576   if (!llvm::equal(getIndexingMaps(), indexingMaps)) {