| /llvm-project-15.0.7/mlir/lib/Dialect/Quant/IR/ |
| H A D | TypeDetail.h | 163 ArrayRef<double> scales, ArrayRef<int64_t> zeroPoints, in KeyTy()
167 scales(scales), zeroPoints(zeroPoints), in KeyTy()
180 ArrayRef<int64_t> zeroPoints; member
187 ArrayRef<int64_t> getZeroPoints() const { return zeroPoints; } in getZeroPoints()
212 llvm::hash_combine_range(zeroPoints.begin(), zeroPoints.end()), in getHashValue()
220 ArrayRef<int64_t> zeroPoints) in UniformQuantizedPerAxisTypeStorage()
223 scaleElements(scales.data()), zeroPointElements(zeroPoints.data()), in UniformQuantizedPerAxisTypeStorage()
235 ArrayRef<int64_t> zeroPoints = allocator.copyInto(key.zeroPoints); in construct() local
237 UniformQuantizedPerAxisTypeStorage(key, scales, zeroPoints); in construct()
|
| H A D | TypeParser.cpp | 204 SmallVector<int64_t, 1> zeroPoints; in parseUniformType() local
256 zeroPoints.resize(zeroPoints.size() + 1); in parseUniformType()
257 if (parseQuantParams(parser, scales.back(), zeroPoints.back())) { in parseUniformType()
281 ArrayRef<int64_t> zeroPointsRef(zeroPoints.begin(), zeroPoints.end()); in parseUniformType()
288 typeFlags, storageType, expressedType, scales.front(), zeroPoints.front(), in parseUniformType()
403 ArrayRef<int64_t> zeroPoints = type.getZeroPoints(); in printUniformQuantizedPerAxisType() local
408 printQuantParams(scales[index], zeroPoints[index], out); in printUniformQuantizedPerAxisType()
|
| H A D | QuantTypes.cpp | 305 ArrayRef<double> scales, ArrayRef<int64_t> zeroPoints, in get() argument
309 scales, zeroPoints, quantizedDimension, storageTypeMin, in get()
316 ArrayRef<int64_t> zeroPoints, int32_t quantizedDimension, in getChecked() argument
319 storageType, expressedType, scales, zeroPoints, in getChecked()
326 ArrayRef<int64_t> zeroPoints, int32_t quantizedDimension, in verify() argument
345 if (scales.size() != zeroPoints.size()) in verify()
347 << scales.size() << ", " << zeroPoints.size(); in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 161 SmallVector<int64_t, 4> zeroPoints; in fakeQuantAttrsToType() local
163 zeroPoints.reserve(axisSize); in fakeQuantAttrsToType()
169 zeroPoints.push_back(qmin); in fakeQuantAttrsToType()
177 zeroPoints.push_back(nudgedZeroPoint); in fakeQuantAttrsToType()
182 loc, flags, storageType, expressedType, scales, zeroPoints, in fakeQuantAttrsToType()
|
| /llvm-project-15.0.7/mlir/lib/Bindings/Python/ |
| H A D | DialectQuant.cpp | 221 std::vector<int64_t> zeroPoints, int32_t quantizedDimension, in populateDialectQuantSubmodule()
223 if (scales.size() != zeroPoints.size()) in populateDialectQuantSubmodule()
229 zeroPoints.data(), quantizedDimension, storageTypeMin, in populateDialectQuantSubmodule()
256 std::vector<int64_t> zeroPoints; in populateDialectQuantSubmodule() local
257 zeroPoints.reserve(nDim); in populateDialectQuantSubmodule()
261 zeroPoints.push_back(zeroPoint); in populateDialectQuantSubmodule()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Quant/ |
| H A D | UniformSupport.h | 181 zeroPoints(uniformType.getZeroPoints()), in UniformQuantizedPerAxisValueConverter()
189 assert(scales.size() == zeroPoints.size()); in UniformQuantizedPerAxisValueConverter()
203 UniformQuantizedValueConverter converter(scales[index], zeroPoints[index], in getPerChunkConverter()
210 const ArrayRef<int64_t> zeroPoints; variable
|
| H A D | QuantTypes.h | 325 ArrayRef<double> scales, ArrayRef<int64_t> zeroPoints,
334 ArrayRef<int64_t> zeroPoints, int32_t quantizedDimension,
341 ArrayRef<int64_t> zeroPoints,
|
| /llvm-project-15.0.7/mlir/test/CAPI/ |
| H A D | quant.c | 167 int64_t zeroPoints[] = {0, 120}; in testUniformPerAxisType() local
170 /*nDims=*/2, scales, zeroPoints, in testUniformPerAxisType()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/Dialect/ |
| H A D | Quant.cpp | 165 intptr_t nDims, double *scales, int64_t *zeroPoints, in mlirUniformQuantizedPerAxisTypeGet() argument
170 llvm::makeArrayRef(scales, nDims), llvm::makeArrayRef(zeroPoints, nDims), in mlirUniformQuantizedPerAxisTypeGet()
|
| /llvm-project-15.0.7/mlir/include/mlir-c/Dialect/ |
| H A D | Quant.h | 151 intptr_t nDims, double *scales, int64_t *zeroPoints,
|