From 53d0434e4bce0e6533811d970cfc872718fda068 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Tue, 7 Sep 2021 21:46:31 +0000 Subject: [PATCH] update --- python/tvm/topi/cuda/pooling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/topi/cuda/pooling.py b/python/tvm/topi/cuda/pooling.py index 7ddbea27c174..1b5cc94c3db0 100644 --- a/python/tvm/topi/cuda/pooling.py +++ b/python/tvm/topi/cuda/pooling.py @@ -137,7 +137,7 @@ def _schedule(PaddedInput, Pool): def traverse(OP): """Internal traverse function""" # inline all one-to-one-mapping operators except the last stage (output) - if tag.is_broadcast(OP.tag): + if tag.is_injective(OP.tag): if OP not in s.outputs: s[OP].compute_inline() for tensor in OP.input_tensors: