From 4e0fa644d06c4b01c98a02cfa059b017016e7f81 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Fri, 18 Jan 2019 14:06:31 +0800 Subject: [PATCH] Fix doc --- topi/python/topi/cuda/vision.py | 2 +- topi/python/topi/generic/vision.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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