Searched refs:weightType (Results 1 – 6 of 6) sorted by relevance
120 auto weightType = weight.getType().dyn_cast<ShapedType>(); in buildConvOpQuantizationAttr() local122 if (!inputType || !weightType) in buildConvOpQuantizationAttr()126 auto weightPerTensorQType = GET_UQTYPE(weightType); in buildConvOpQuantizationAttr()127 auto weightPerAxisQType = weightType.getElementType() in buildConvOpQuantizationAttr()239 auto weightType = weight.getType().dyn_cast<ShapedType>(); in buildConvOpResultTypeInfo() local241 assert(inputType && weightType && in buildConvOpResultTypeInfo()245 auto weightQType = GET_QTYPE(weightType); in buildConvOpResultTypeInfo()
32 ShapedType weightType = weight.getType().cast<ShapedType>(); in matchAndRewrite() local36 if (!(inputType.hasStaticShape() && weightType.hasStaticShape() && in matchAndRewrite()54 ArrayRef<int64_t> weightShape = weightType.getShape(); in matchAndRewrite()
31 ShapedType weightType = weight.getType().cast<ShapedType>(); in matchAndRewrite() local39 if (!weightType.hasRank()) in matchAndRewrite()49 ArrayRef<int64_t> weightShape = weightType.getShape(); in matchAndRewrite()
121 auto weightType = in matchAndRewrite() local125 if (!weightType) in matchAndRewrite()137 weightType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
106 auto weightType = in verifyConvOp() local114 if (!weightType) { in verifyConvOp()120 auto weightEType = weightType.getElementType(); in verifyConvOp()
541 VectorType weightType = VectorType::get(2, rewriter.getI32Type()); in matchAndRewrite() local542 branchWeights = DenseElementsAttr::get(weightType, weights->getValue()); in matchAndRewrite()