Lines Matching refs:quant
16 MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(quant, quant, quant::QuantizationDialect) in MLIR_DEFINE_CAPI_DIALECT_REGISTRATION() argument
23 return unwrap(type).isa<quant::QuantizedType>(); in MLIR_DEFINE_CAPI_DIALECT_REGISTRATION()
27 return quant::QuantizationFlags::Signed; in mlirQuantizedTypeGetSignedFlag()
32 return quant::QuantizedType::getDefaultMinimumForInteger(isSigned, in mlirQuantizedTypeGetDefaultMinimumForInteger()
38 return quant::QuantizedType::getDefaultMaximumForInteger(isSigned, in mlirQuantizedTypeGetDefaultMaximumForInteger()
43 return wrap(unwrap(type).cast<quant::QuantizedType>().getExpressedType()); in mlirQuantizedTypeGetExpressedType()
47 return unwrap(type).cast<quant::QuantizedType>().getFlags(); in mlirQuantizedTypeGetFlags()
51 return unwrap(type).cast<quant::QuantizedType>().isSigned(); in mlirQuantizedTypeIsSigned()
55 return wrap(unwrap(type).cast<quant::QuantizedType>().getStorageType()); in mlirQuantizedTypeGetStorageType()
59 return unwrap(type).cast<quant::QuantizedType>().getStorageTypeMin(); in mlirQuantizedTypeGetStorageTypeMin()
63 return unwrap(type).cast<quant::QuantizedType>().getStorageTypeMax(); in mlirQuantizedTypeGetStorageTypeMax()
68 .cast<quant::QuantizedType>() in mlirQuantizedTypeGetStorageTypeIntegralWidth()
74 return unwrap(type).cast<quant::QuantizedType>().isCompatibleExpressedType( in mlirQuantizedTypeIsCompatibleExpressedType()
79 return wrap(quant::QuantizedType::getQuantizedElementType(unwrap(type))); in mlirQuantizedTypeGetQuantizedElementType()
84 return wrap(unwrap(type).cast<quant::QuantizedType>().castFromStorageType( in mlirQuantizedTypeCastFromStorageType()
89 return wrap(quant::QuantizedType::castToStorageType( in mlirQuantizedTypeCastToStorageType()
90 unwrap(type).cast<quant::QuantizedType>())); in mlirQuantizedTypeCastToStorageType()
95 return wrap(unwrap(type).cast<quant::QuantizedType>().castFromExpressedType( in mlirQuantizedTypeCastFromExpressedType()
100 return wrap(quant::QuantizedType::castToExpressedType(unwrap(type))); in mlirQuantizedTypeCastToExpressedType()
106 unwrap(type).cast<quant::QuantizedType>().castExpressedToStorageType( in mlirQuantizedTypeCastExpressedToStorageType()
115 return unwrap(type).isa<quant::AnyQuantizedType>(); in mlirTypeIsAAnyQuantizedType()
121 return wrap(quant::AnyQuantizedType::get(flags, unwrap(storageType), in mlirAnyQuantizedTypeGet()
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()
160 return unwrap(type).isa<quant::UniformQuantizedPerAxisType>(); in mlirTypeIsAUniformQuantizedPerAxisType()
168 return wrap(quant::UniformQuantizedPerAxisType::get( in mlirUniformQuantizedPerAxisTypeGet()
176 .cast<quant::UniformQuantizedPerAxisType>() in mlirUniformQuantizedPerAxisTypeGetNumDims()
183 .cast<quant::UniformQuantizedPerAxisType>() in mlirUniformQuantizedPerAxisTypeGetScale()
190 .cast<quant::UniformQuantizedPerAxisType>() in mlirUniformQuantizedPerAxisTypeGetZeroPoint()
196 .cast<quant::UniformQuantizedPerAxisType>() in mlirUniformQuantizedPerAxisTypeGetQuantizedDimension()
201 return unwrap(type).cast<quant::UniformQuantizedPerAxisType>().isFixedPoint(); in mlirUniformQuantizedPerAxisTypeIsFixedPoint()
209 return unwrap(type).isa<quant::CalibratedQuantizedType>(); in mlirTypeIsACalibratedQuantizedType()
215 quant::CalibratedQuantizedType::get(unwrap(expressedType), min, max)); in mlirCalibratedQuantizedTypeGet()
219 return unwrap(type).cast<quant::CalibratedQuantizedType>().getMin(); in mlirCalibratedQuantizedTypeGetMin()
223 return unwrap(type).cast<quant::CalibratedQuantizedType>().getMax(); in mlirCalibratedQuantizedTypeGetMax()