modify doc comments

python/tvm/relay/analysis/analysis.py

@@ -434,10 +434,10 @@ def get_calibration_data(mod, data):
 
 
 def extract_intermdeiate_expr(mod, expr_id):
-    """Extract Relay Expr by the expression ID
+    """Extract Relay Expr by its expression ID
 
     This function is used for extracting Relay Expr
-    by the expression ID of the main function
+    by its expression ID of the main function
     that we can see in `print(mod["main"])`.
 
     Parameters
@@ -454,7 +454,7 @@ def extract_intermdeiate_expr(mod, expr_id):
     --------
     .. code-block:: python
 
-        # Supposed our module is printed like this:
+        # Suppose our module is printed like this:
         # def @main(%x: Tensor[(1, 1, 5, 1), float32], %w1, %w2) {
         #   %0 = nn.conv2d(%x, %w1, padding=[1, 1, 1, 1], channels=1, kernel_size=[3, 3]);
         #   %1 = nn.conv2d(%0, %w2, padding=[1, 1, 1, 1], channels=1, kernel_size=[3, 3]);

src/relay/analysis/extract_intermediate_expr.cc

@@ -39,7 +39,7 @@ class ExtractIntermediateExprWrapper : private MixedModeVisitor {
   IRModule Extract() {
     VisitExpr(this->mod_->Lookup("main"));
 
-    // ensure the target op num we want to dump is valid.
+    // ensure the target expr_id we want to extract is valid.
     ICHECK(target_expr_id_ >= 0 && target_expr_id_ < counter_);
 
     return IRModule::FromExpr(target_op_, {});
@@ -49,7 +49,7 @@ class ExtractIntermediateExprWrapper : private MixedModeVisitor {
   using MixedModeVisitor::VisitExpr_;
 
   const IRModule mod_;
-  /*! \brief the OP Number that we want to dump. */
+  /*! \brief the expr id that we want to extract. */
   const int target_expr_id_;
   int counter_;
   Expr target_op_;