MCPcopy Create free account
hub / github.com/apache/tvm / swap

Function swap

python/tvm/topi/utils.py:531–533  ·  view source on GitHub ↗

swap arr[axis] and arr[-1]

(arr, axis)

Source from the content-addressed store, hash-verified

529
530
531def swap(arr, axis):
532 """swap arr[axis] and arr[-1]"""
533 return arr[:axis] + [arr[-1]] + arr[axis + 1 : -1] + [arr[axis]]
534
535
536def is_target(names):

Callers 7

exclusive_scanFunction · 0.85
inclusive_scanFunction · 0.85
sortFunction · 0.85
sort_thrustFunction · 0.85
argsortFunction · 0.85
topkFunction · 0.85
topk_thrustFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…