add definition for affine

apache · Jun 8, 2021 · c150e4f · c150e4f
1 parent 734a0d8
commit c150e4f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/tvm/relay/op/op.py b/python/tvm/relay/op/op.py
@@ -440,7 +440,8 @@ def register_fake_quantization_to_integer(op_name, func=None, level=10):
     """Register quantize function for an op
 
     Given an op and Affine Types on it's inputs, this function should return the op
-    in affine space/integer operators and the new type of the output
+    in affine space/integer operators and the new type of the output, where affine
+    denotes the transformation x_real = (x_affine - zero_point) * scale
 
     Parameters
     ----------

diff --git a/src/relay/transforms/fake_quantization_to_integer.cc b/src/relay/transforms/fake_quantization_to_integer.cc
@@ -51,7 +51,8 @@
  *
  * The second pass is an ExprVisitor that recursively searches for subgraphs leading to the
  * quantize for subtraphs bounded by dequantize operations. This pass extracts the affine
- * types of the inputs for later processing
+ * types of the inputs for later processing, where affine denotes the transformation
+ * x_real = (x_affine - zero_point) * scale
  *
  * The third pass is an ExprMutator that recursively rewrites the subgraphs using packed funcs
  * registered with the FTVMFakeQuantizationToInteger attribute. These packed funcs rewrite