Searched refs:weightType (Results 1 – 5 of 5) sorted by relevance
106 auto weightType = weight.getType().dyn_cast<RankedTensorType>(); in buildConvOpQuantizationAttr() local108 if (!inputType || !weightType) in buildConvOpQuantizationAttr()112 auto weightPerTensorQType = GET_UQTYPE(weightType); in buildConvOpQuantizationAttr()113 auto weightPerAxisQType = weightType.getElementType() in buildConvOpQuantizationAttr()249 auto weightType = weight.getType().dyn_cast<RankedTensorType>(); in buildConvOpResultTypeInfo() local251 assert(inputType && weightType && in buildConvOpResultTypeInfo()255 auto weightQType = GET_QTYPE(weightType); in buildConvOpResultTypeInfo()
121 auto weightType = in matchAndRewrite() local125 if (!weightType) in matchAndRewrite()137 weightType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
118 auto weightType = op.weight().getType().template dyn_cast<RankedTensorType>(); in verifyConvOp() local121 if (!inputType || !weightType) in verifyConvOp()127 weightType.getElementType().template isa<mlir::quant::QuantizedType>(); in verifyConvOp()
545 VectorType weightType = VectorType::get(2, rewriter.getI32Type()); in matchAndRewrite() local547 DenseElementsAttr::get(weightType, weights.getValue().getValue()); in matchAndRewrite()
META-INF/ META-INF/MANIFEST.MF com/ com/google/ com/ ...