Skip to content

Commit

Permalink
more appeasing the formatting gods
Browse files Browse the repository at this point in the history
  • Loading branch information
AndrewZhaoLuo committed Apr 27, 2021
1 parent 1ab0e24 commit 800449e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions python/tvm/topi/testing/poolnd_python.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,8 @@
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.

# pylint: disable=invalid-name, unused-argument, unused-variable
"""Ground truth max and average pooling operators in python."""
import itertools
import math
from typing import List, Tuple
Expand Down Expand Up @@ -110,7 +111,7 @@ def poolnd_python(
out_shape.append(int(math.floor(val) + 1))
out_shape = tuple(out_shape)

# Create a padded array, and a boolean mask showing which values are padded values and which are not
# Create a padded array, and a boolean mask showing which values are padded values
pad_value = 0
if pool_type == "max" and not count_include_pad:
pad_value = tvm.te.min_value(dtype).value
Expand Down

0 comments on commit 800449e

Please sign in to comment.