Tvm.select error when i try tutorial

#1

C = tvm.compute(A.shape, lambda *i: B(*i) * tvm.select(A(*i) > 0, 1, 0))
AttributeError: module ‘tvm’ has no attribute ‘select’

#2

Use tvm.expr.Select or tvm.if_then_else

1 Like