| /llvm-project-15.0.7/mlir/test/python/dialects/ |
| H A D | quant.py | 30 assert quant.UniformQuantizedType.isinstance(uniform)
35 assert not quant.UniformQuantizedType.isinstance(per_axis)
74 uniform = quant.UniformQuantizedType.get(
75 quant.UniformQuantizedType.FLAG_SIGNED, i8, f32, 0.99872, 127, -8, 7)
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Quant/ |
| H A D | QuantOpsBase.td | 66 // An implementation of UniformQuantizedType.
69 CPred<"$_self.isa<UniformQuantizedType>()">,
70 "UniformQuantizedType">;
72 // Predicate for detecting a container or primitive of UniformQuantizedType.
|
| H A D | QuantizeUtils.h | 18 class UniformQuantizedType; variable
55 UniformQuantizedType quantizedElementType,
|
| H A D | QuantTypes.h | 256 class UniformQuantizedType
257 : public Type::TypeBase<UniformQuantizedType, QuantizedType,
265 static UniformQuantizedType get(unsigned flags, Type storageType,
272 static UniformQuantizedType
|
| H A D | FakeQuantSupport.h | 51 UniformQuantizedType fakeQuantAttrsToType(Location loc, unsigned numBits,
|
| H A D | UniformSupport.h | 63 explicit UniformQuantizedValueConverter(UniformQuantizedType uniformType) in UniformQuantizedValueConverter()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Quant/IR/ |
| H A D | QuantTypes.cpp | 252 UniformQuantizedType UniformQuantizedType::get(unsigned flags, Type storageType, in get()
261 UniformQuantizedType UniformQuantizedType::getChecked( in getChecked()
270 LogicalResult UniformQuantizedType::verify( in verify()
297 double UniformQuantizedType::getScale() const { return getImpl()->scale; } in getScale()
299 int64_t UniformQuantizedType::getZeroPoint() const { in getZeroPoint()
|
| H A D | TypeParser.cpp | 287 return parser.getChecked<UniformQuantizedType>( in parseUniformType()
381 static void printUniformQuantizedType(UniformQuantizedType type, in printUniformQuantizedType()
426 else if (auto uniformType = type.dyn_cast<UniformQuantizedType>()) in printType()
|
| H A D | QuantOps.cpp | 29 addTypes<AnyQuantizedType, CalibratedQuantizedType, UniformQuantizedType, in initialize()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/Dialect/ |
| H A D | Quant.cpp | 131 return unwrap(type).isa<quant::UniformQuantizedType>(); in mlirTypeIsAUniformQuantizedType()
138 return wrap(quant::UniformQuantizedType::get( in mlirUniformQuantizedTypeGet()
144 return unwrap(type).cast<quant::UniformQuantizedType>().getScale(); in mlirUniformQuantizedTypeGetScale()
148 return unwrap(type).cast<quant::UniformQuantizedType>().getZeroPoint(); in mlirUniformQuantizedTypeGetZeroPoint()
152 return unwrap(type).cast<quant::UniformQuantizedType>().isFixedPoint(); in mlirUniformQuantizedTypeIsFixedPoint()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Tosa/ |
| H A D | TosaTestPasses.cpp | 58 outputType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
135 inputType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
137 weightType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
139 outputType.getElementType().dyn_cast<mlir::quant::UniformQuantizedType>(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/unittests/Dialect/Quant/ |
| H A D | QuantizationUtilsTest.cpp | 26 TestUniformQuantizedValueConverter(UniformQuantizedType type) in TestUniformQuantizedValueConverter()
33 UniformQuantizedType qtype;
70 UniformQuantizedType getTestQuantizedType(Type storageType, MLIRContext *ctx) { in getTestQuantizedType()
71 return UniformQuantizedType::get(/*flags=*/false, storageType, in getTestQuantizedType()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 107 UniformQuantizedType
126 return UniformQuantizedType::getChecked( in fakeQuantAttrsToType()
134 return UniformQuantizedType::getChecked(loc, flags, storageType, in fakeQuantAttrsToType()
|
| H A D | QuantizeUtils.cpp | 100 Attribute realValue, UniformQuantizedType quantizedElementType, in quantizeAttrUniform()
131 quantizedElementType.dyn_cast<UniformQuantizedType>()) { in quantizeAttr()
|
| /llvm-project-15.0.7/mlir/python/mlir/_mlir_libs/_mlir/dialects/ |
| H A D | quant.pyi | 12 "UniformQuantizedType",
76 class UniformQuantizedType(QuantizedType):
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Utils/ |
| H A D | QuantUtils.cpp | 107 ((input_type).getElementType().dyn_cast<quant::UniformQuantizedType>())
|
| /llvm-project-15.0.7/mlir/docs/ |
| H A D | Quantization.md | 206 * UniformQuantizedType
241 appropriate *UniformQuantizedType* as the target of the qcast operation.
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 149 if (auto quantType = inputETy.dyn_cast<mlir::quant::UniformQuantizedType>()) in verify()
152 if (auto quantType = resultETy.dyn_cast<mlir::quant::UniformQuantizedType>()) in verify()
248 .dyn_cast<mlir::quant::UniformQuantizedType>(); in buildMatMulOpWithQuantInfo()
|