Searched refs:inElementTy (Results 1 – 2 of 2) sorted by relevance
763 Type inElementTy = inputTy.getElementType(); in matchAndRewrite() local769 inElementTy.isa<IntegerType>() ? rewriter.getI32Type() : inElementTy; in matchAndRewrite()783 Attribute padAttr = rewriter.getZeroAttr(inElementTy); in matchAndRewrite()
1932 auto inElementTy = inputTy.getElementType(); in matchAndRewrite() local1935 auto resultMaxTy = RankedTensorType::get(resultTy.getShape(), inElementTy); in matchAndRewrite()1966 loc, dynDims, resultTy.getShape(), inElementTy) in matchAndRewrite()1969 createInitialValueForReduceOp(argmaxOp, inElementTy, rewriter); in matchAndRewrite()2013 if (inElementTy.isa<FloatType>()) { in matchAndRewrite()2016 } else if (inElementTy.isa<IntegerType>()) { in matchAndRewrite()