Searched refs:quantizedType (Results 1 – 5 of 5) sorted by relevance
| /external/llvm-project/mlir/unittests/Dialect/Quant/ |
| D | QuantizationUtilsTest.cpp | 81 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()
124 auto quantizedType = getTestQuantizedType(convertedType, &ctx); in TEST() local
125 TestUniformQuantizedValueConverter converter(quantizedType); in TEST()
131 quantizeAttrUniform(realValue, quantizedType, converter, returnedType); in TEST()
149 auto quantizedType = getTestQuantizedType(convertedType, &ctx); in TEST() local
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | QuantTypes.cpp | 122 Type QuantizedType::castToStorageType(Type quantizedType) { in castToStorageType() argument
123 if (quantizedType.isa<QuantizedType>()) { in castToStorageType()
125 return quantizedType.cast<QuantizedType>().getStorageType(); in castToStorageType()
126 } else if (quantizedType.isa<ShapedType>()) { in castToStorageType()
128 ShapedType sType = quantizedType.cast<ShapedType>(); in castToStorageType()
134 if (quantizedType.isa<RankedTensorType>()) { in castToStorageType()
136 } else if (quantizedType.isa<UnrankedTensorType>()) { in castToStorageType()
138 } else if (quantizedType.isa<VectorType>()) { in castToStorageType()
171 Type QuantizedType::castToExpressedType(Type quantizedType) { in castToExpressedType() argument
172 if (quantizedType.isa<QuantizedType>()) { in castToExpressedType()
[all …]
|
| D | QuantOps.cpp | 54 if (auto quantizedType = spec.dyn_cast<QuantizedType>()) in isValidQuantizationSpec() local
55 return quantizedType.isCompatibleExpressedType(expressed); in isValidQuantizationSpec()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/Transforms/ |
| D | ConvertSimQuant.cpp | 65 Type quantizedType = converter.convert(elementType); in failableRewrite() local
66 assert(quantizedType && in failableRewrite()
71 auto qbarrier = rewriter.create<QuantizeCastOp>(op.getLoc(), quantizedType, in failableRewrite()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantTypes.h | 152 static Type castToStorageType(Type quantizedType);
165 static Type castToExpressedType(Type quantizedType);
|