diff --git a/topi/python/topi/cuda/vision.py b/topi/python/topi/cuda/vision.py index 835c346f8321f..abcbdb50074ef 100644 --- a/topi/python/topi/cuda/vision.py +++ b/topi/python/topi/cuda/vision.py @@ -159,7 +159,7 @@ def schedule_proposal(outs): Parameters ---------- outs: Array of Tensor - The computation graph description of roi_align + The computation graph description of proposal in the format of an array of tensors. Returns diff --git a/topi/python/topi/generic/vision.py b/topi/python/topi/generic/vision.py index 8f903c11680d8..9a1e06aa30e83 100644 --- a/topi/python/topi/generic/vision.py +++ b/topi/python/topi/generic/vision.py @@ -165,7 +165,7 @@ def schedule_proposal(outs): Parameters ---------- outs: Array of Tensor - The computation graph description of roi_align + The computation graph description of proposal in the format of an array of tensors. Returns