Lines Matching defs:indexingMaps
73 ArrayRef<AffineMap> indexingMaps) {
77 loc, resultTy, ValueRange({bias, conv}), result, indexingMaps,
129 SmallVector<AffineMap, 2> indexingMaps;
130 indexingMaps.push_back(getBroadcastingMap(rewriter, source, result));
131 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank));
136 loc, resultTy, ValueRange({source}), result, indexingMaps,
510 SmallVector<AffineMap, 4> indexingMaps;
511 indexingMaps.push_back(getBroadcastingMap(rewriter, bias, biasEmptyTensor));
512 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank));
513 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank));
531 biasEmptyTensor, indexingMaps,
555 rewriter, loc, bias, convReshape, biasEmptyTensor, indexingMaps);