Searched refs:quantizedDimension (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | TypeDetail.h | 164 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
|
| D | QuantTypes.cpp | 293 int32_t quantizedDimension, int64_t storageTypeMin, in get() argument
296 scales, zeroPoints, quantizedDimension, storageTypeMin, in get()
303 int32_t quantizedDimension, int64_t storageTypeMin, int64_t storageTypeMax, in getChecked() argument
306 zeroPoints, quantizedDimension, storageTypeMin, in getChecked()
313 int32_t quantizedDimension, int64_t storageTypeMin, in verifyConstructionInvariants() argument
355 return getImpl()->quantizedDimension; in getQuantizedDimension()
|
| D | TypeParser.cpp | 201 int32_t quantizedDimension; in parseUniformType() local
233 if (parser.parseInteger(quantizedDimension)) in parseUniformType()
283 quantizedDimension, storageTypeMin, storageTypeMax, loc); in parseUniformType()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantTypes.h | 322 int32_t quantizedDimension, int64_t storageTypeMin,
330 int32_t quantizedDimension, int64_t storageTypeMin,
338 int32_t quantizedDimension,
|
| D | FakeQuantSupport.h | 60 fakeQuantAttrsToType(Location loc, unsigned numBits, int32_t quantizedDimension,
|
| /external/llvm-project/mlir/lib/Dialect/Quant/Utils/ |
| D | FakeQuantSupport.cpp | 140 Location loc, unsigned numBits, int32_t quantizedDimension, in fakeQuantAttrsToType() argument
182 flags, storageType, expressedType, scales, zeroPoints, quantizedDimension, in fakeQuantAttrsToType()
|