Return True if the name of the current target is one of provided names
(names)
| 534 | |
| 535 | |
| 536 | def is_target(names): |
| 537 | """Return True if the name of the current target is one of provided names""" |
| 538 | names = [names] if isinstance(names, str) else names |
| 539 | target = tvm.target.Target.current(allow_none=False) |
| 540 | return any(name in target.keys for name in names) |
| 541 | |
| 542 | |
| 543 | def is_dynamic_shape(shape): |