Searched refs:axisValue (Results 1 – 1 of 1) sorted by relevance
1652 Value axisValue = rewriter.createOrFold<arith::ConstantOp>( in matchAndRewrite() local1672 auto size = rewriter.createOrFold<tensor::DimOp>(loc, arg, axisValue); in matchAndRewrite()1695 sizes[axis] = rewriter.createOrFold<tensor::DimOp>(loc, arg, axisValue); in matchAndRewrite()