Searched refs:argmaxOp (Results 1 – 1 of 1) sorted by relevance
1926 LogicalResult matchAndRewrite(tosa::ArgMaxOp argmaxOp, in matchAndRewrite() argument1928 auto loc = argmaxOp.getLoc(); in matchAndRewrite()1929 Value input = argmaxOp.getInput(); in matchAndRewrite()1931 auto resultTy = argmaxOp.getOutput().getType().cast<ShapedType>(); in matchAndRewrite()1934 int axis = argmaxOp.getAxis(); in matchAndRewrite()1939 argmaxOp, in matchAndRewrite()1969 createInitialValueForReduceOp(argmaxOp, inElementTy, rewriter); in matchAndRewrite()1973 argmaxOp, "unsupported tosa.argmax element type"); in matchAndRewrite()2034 argmaxOp, "unsupported tosa.argmax element type"); in matchAndRewrite()2036 rewriter.replaceOp(argmaxOp, linalgOp.getResult(0)); in matchAndRewrite()