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

Function _tvm_to_numpy

tests/python/relax/test_training_optimizer_numeric.py:45–48  ·  view source on GitHub ↗
(data)

Source from the content-addressed store, hash-verified

43
44
45def _tvm_to_numpy(data):
46 if isinstance(data, list | tuple | tvm_ffi.Array):
47 return [_tvm_to_numpy(_data) for _data in data]
48 return data.numpy()
49
50
51def _assert_allclose_nested(data1, data2):

Callers 2

_assert_run_result_sameFunction · 0.70
_test_optimizerFunction · 0.70

Calls 1

numpyMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…