Lines Matching refs:indexingMaps
538 ArrayAttr indexingMaps, in build() argument
540 build(builder, result, lhs, rhs, acc, indexingMaps, iteratorTypes, in build()
546 ArrayAttr indexingMaps, in build() argument
550 result.addAttribute(::mlir::getIndexingMapsAttrName(), indexingMaps); in build()
815 getDimMap(ArrayRef<AffineMap> indexingMaps, ArrayAttr iteratorTypes, in getDimMap() argument
824 int64_t lhsDim = getResultIndex(indexingMaps[0], targetExpr); in getDimMap()
825 int64_t rhsDim = getResultIndex(indexingMaps[1], targetExpr); in getDimMap()
836 SmallVector<AffineMap, 4> indexingMaps(getIndexingMapsArray()); in getIterationBounds() local
844 int64_t lhsDimIndex = getResultIndex(indexingMaps[0], targetExpr); in getIterationBounds()
850 int64_t resDimIndex = getResultIndex(indexingMaps[2], targetExpr); in getIterationBounds()
872 SmallVector<AffineMap, 4> indexingMaps(getIndexingMapsArray()); in getContractingDimMap() local
873 return getDimMap(indexingMaps, getIteratorTypes(), in getContractingDimMap()
878 SmallVector<AffineMap, 4> indexingMaps(getIndexingMapsArray()); in getBatchDimMap() local
879 return getDimMap(indexingMaps, getIteratorTypes(), in getBatchDimMap()