Home
last modified time | relevance | path

Searched refs:quantizedType (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DDialectQuant.cpp23 auto quantizedType = in populateDialectQuantSubmodule() local
25 quantizedType.def_staticmethod( in populateDialectQuantSubmodule()
34 quantizedType.def_staticmethod( in populateDialectQuantSubmodule()
73 quantizedType.def( in populateDialectQuantSubmodule()
87 quantizedType.def( in populateDialectQuantSubmodule()
100 quantizedType.def_staticmethod( in populateDialectQuantSubmodule()
112 quantizedType.def( in populateDialectQuantSubmodule()
125 quantizedType.def_staticmethod( in populateDialectQuantSubmodule()
137 quantizedType.def( in populateDialectQuantSubmodule()
181 quantizedType.get_class()); in populateDialectQuantSubmodule()
[all …]
/llvm-project-15.0.7/mlir/unittests/Dialect/Quant/
H A DQuantizationUtilsTest.cpp81 auto quantizedType = getTestQuantizedType(convertedType, &ctx); in TEST() local
82 TestUniformQuantizedValueConverter converter(quantizedType); in TEST()
87 quantizeAttrUniform(realValue, quantizedType, converter, typeResult); in TEST()
99 auto quantizedType = getTestQuantizedType(convertedType, &ctx); in TEST() local
100 TestUniformQuantizedValueConverter converter(quantizedType); in TEST()
106 quantizeAttrUniform(realValue, quantizedType, converter, returnedType); in TEST()
125 auto quantizedType = getTestQuantizedType(convertedType, &ctx); in TEST() local
126 TestUniformQuantizedValueConverter converter(quantizedType); in TEST()
132 quantizeAttrUniform(realValue, quantizedType, converter, returnedType); in TEST()
151 auto quantizedType = getTestQuantizedType(convertedType, &ctx); in TEST() local
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Quant/IR/
H A DQuantTypes.cpp127 if (quantizedType.isa<QuantizedType>()) { in castToStorageType()
131 if (quantizedType.isa<ShapedType>()) { in castToStorageType()
133 ShapedType sType = quantizedType.cast<ShapedType>(); in castToStorageType()
139 if (quantizedType.isa<RankedTensorType>()) { in castToStorageType()
142 if (quantizedType.isa<UnrankedTensorType>()) { in castToStorageType()
145 if (quantizedType.isa<VectorType>()) { in castToStorageType()
182 if (quantizedType.isa<QuantizedType>()) { in castToExpressedType()
186 if (quantizedType.isa<ShapedType>()) { in castToExpressedType()
194 if (quantizedType.isa<RankedTensorType>()) { in castToExpressedType()
197 if (quantizedType.isa<UnrankedTensorType>()) { in castToExpressedType()
[all …]
H A DQuantOps.cpp56 if (auto quantizedType = spec.dyn_cast<QuantizedType>()) in isValidQuantizationSpec() local
57 return quantizedType.isCompatibleExpressedType(expressed); in isValidQuantizationSpec()
/llvm-project-15.0.7/mlir/lib/Dialect/Quant/Transforms/
H A DConvertSimQuant.cpp65 Type quantizedType = converter.convert(elementType); in failableRewrite() local
66 assert(quantizedType && in failableRewrite()
71 auto qbarrier = rewriter.create<QuantizeCastOp>(op.getLoc(), quantizedType, in failableRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Quant/
H A DQuantTypes.h152 static Type castToStorageType(Type quantizedType);
165 static Type castToExpressedType(Type quantizedType);