Searched refs:axisDimSize (Results 1 – 1 of 1) sorted by relevance
1728 Value axisDimSize = rewriter.create<tensor::DimOp>(loc, input, axis); in matchAndRewrite() local1750 rewriter.create<arith::SubIOp>(nestedLoc, axisDimSize, one); in matchAndRewrite()