Home
last modified time | relevance | path

Searched refs:quantizedDimension (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Quant/IR/
H A DTypeDetail.h164 int32_t quantizedDimension, int64_t storageTypeMin, in KeyTy()
168 quantizedDimension(quantizedDimension), in KeyTy()
181 int32_t quantizedDimension; member
197 lhs.quantizedDimension == rhs.quantizedDimension && in genericIsEqual()
225 quantizedDimension(key.quantizedDimension) {} in UniformQuantizedPerAxisTypeStorage()
253 int32_t quantizedDimension; member
H A DQuantTypes.cpp306 int32_t quantizedDimension, int64_t storageTypeMin, in get() argument
309 scales, zeroPoints, quantizedDimension, storageTypeMin, in get()
316 ArrayRef<int64_t> zeroPoints, int32_t quantizedDimension, in getChecked() argument
320 quantizedDimension, storageTypeMin, storageTypeMax); in getChecked()
326 ArrayRef<int64_t> zeroPoints, int32_t quantizedDimension, in verify() argument
367 return getImpl()->quantizedDimension; in getQuantizedDimension()
H A DTypeParser.cpp202 int32_t quantizedDimension; in parseUniformType() local
234 if (parser.parseInteger(quantizedDimension)) in parseUniformType()
284 quantizedDimension, storageTypeMin, storageTypeMax); in parseUniformType()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Quant/
H A DQuantTypes.h326 int32_t quantizedDimension, int64_t storageTypeMin,
334 ArrayRef<int64_t> zeroPoints, int32_t quantizedDimension,
342 int32_t quantizedDimension,
H A DFakeQuantSupport.h60 fakeQuantAttrsToType(Location loc, unsigned numBits, int32_t quantizedDimension,
/llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/
H A DFakeQuantSupport.cpp140 Location loc, unsigned numBits, int32_t quantizedDimension, in fakeQuantAttrsToType() argument
183 quantizedDimension, qmin, qmax); in fakeQuantAttrsToType()
/llvm-project-15.0.7/mlir/lib/CAPI/Dialect/
H A DQuant.cpp166 int32_t quantizedDimension, int64_t storageTypeMin, in mlirUniformQuantizedPerAxisTypeGet() argument
171 quantizedDimension, storageTypeMin, storageTypeMax)); in mlirUniformQuantizedPerAxisTypeGet()
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DDialectQuant.cpp221 std::vector<int64_t> zeroPoints, int32_t quantizedDimension, in populateDialectQuantSubmodule()
229 zeroPoints.data(), quantizedDimension, storageTypeMin, in populateDialectQuantSubmodule()
/llvm-project-15.0.7/mlir/include/mlir-c/Dialect/
H A DQuant.h152 int32_t quantizedDimension, int64_t storageTypeMin, int64_t storageTypeMax);