Home
last modified time | relevance | path

Searched defs:quantizedDimension (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/Quant/IR/
H A DQuantTypes.cpp307 get(unsigned flags,Type storageType,Type expressedType,ArrayRef<double> scales,ArrayRef<int64_t> zeroPoints,int32_t quantizedDimension,int64_t storageTypeMin,int64_t storageTypeMax) get() argument
317 getChecked(function_ref<InFlightDiagnostic ()> emitError,unsigned flags,Type storageType,Type expressedType,ArrayRef<double> scales,ArrayRef<int64_t> zeroPoints,int32_t quantizedDimension,int64_t storageTypeMin,int64_t storageTypeMax) getChecked() argument
327 verify(function_ref<InFlightDiagnostic ()> emitError,unsigned flags,Type storageType,Type expressedType,ArrayRef<double> scales,ArrayRef<int64_t> zeroPoints,int32_t quantizedDimension,int64_t storageTypeMin,int64_t storageTypeMax) verify() argument
H A DTypeDetail.h181 int32_t quantizedDimension; member
253 int32_t quantizedDimension; member
H A DTypeParser.cpp201 int32_t quantizedDimension; in parseUniformType() local
/llvm-project/mlir/lib/Dialect/Quant/Utils/
H A DFakeQuantSupport.cpp140 Location loc, unsigned numBits, int32_t quantizedDimension, in fakeQuantAttrsToType() argument
/llvm-project/mlir/lib/CAPI/Dialect/
H A DQuant.cpp163 int32_t quantizedDimension, int64_t storageTypeMin, in mlirUniformQuantizedPerAxisTypeGet() argument
/llvm-project/mlir/lib/Bindings/Python/
H A DDialectQuant.cpp227 __anond4067f691602(py::object cls, unsigned flags, MlirType storageType, MlirType expressedType, std::vector<double> scales, std::vector<int64_t> zeroPoints, int32_t quantizedDimension, int64_t storageTypeMin, int64_t storageTypeMax) populateDialectQuantSubmodule() argument