Lines Matching defs:genericOp
34 genericOp, \
35 ValueRange{genericOp.getDpsInputs()[(OPERANDS_SWAP) ? 1 : 0], \
36 genericOp.getDpsInputs()[(OPERANDS_SWAP) ? 0 : 1]}, \
37 ValueRange{genericOp.getDpsInits()[0]}))
40 (rewriter.replaceOpWithNewOp<NEWOP>(genericOp, \
41 ValueRange{genericOp.getDpsInputs()[0]}, \
42 ValueRange{genericOp.getDpsInits()[0]}))
58 static bool areBinOpsSwapped(GenericOp genericOp) {
59 Block *body = genericOp.getBody();
135 // Replaces genericOp with `NamedOpTy` op, supplied as a template arg.
149 GenericOp genericOp) {
150 if (genericOp.getNumDpsInputs() != 2 || genericOp.getNumDpsInits() != 1)
154 auto mapRange = genericOp.getIndexingMapsArray();
178 auto res = inferContractionDims(genericOp);
186 *genericOp.getBlock(), [](Operation *first, Operation *second) {
196 auto indexingMaps = genericOp.getIndexingMapsArray();
241 genericOp);
244 genericOp);
245 return replaceWithMatmulVariant<BatchMatmulOp>(rewriter, genericOp);
249 return replaceWithMatmulVariant<MatmulTransposeAOp>(rewriter, genericOp);
251 return replaceWithMatmulVariant<MatmulTransposeBOp>(rewriter, genericOp);
252 return replaceWithMatmulVariant<MatmulOp>(rewriter, genericOp);
261 GenericOp genericOp) {
263 if (isaCopyOpInterface(genericOp)) {
265 genericOp, genericOp.getDpsInputs()[0], genericOp.getDpsInits()[0]);
270 if (isaFillOpInterface(genericOp)) {
272 genericOp, genericOp.getDpsInputs()[0], genericOp.getDpsInits()[0]);
278 isaBroadcastOpInterface(genericOp);
282 genericOp, genericOp.getDpsInputs()[0], genericOp.getDpsInits()[0],
289 isaTransposeOpInterface(genericOp);
293 genericOp, genericOp.getDpsInputs()[0], genericOp.getDpsInits()[0],
299 if (isaElemwiseSingleUnaryOpInterface(genericOp)) {
300 Operation *op = &genericOp.getBody()->front();
308 if (isaElemwiseSingleBinaryOpInterface(genericOp)) {
309 bool swap = areBinOpsSwapped(genericOp);
310 Operation *op = &genericOp.getBody()->front();
330 if (isaContractionOpInterface(genericOp)) {
331 return specializeLinalgContractions(rewriter, genericOp);