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

Function _numpy_to_tvm

tests/python/relax/test_op_gradient_numeric.py:88–91  ·  view source on GitHub ↗
(data)

Source from the content-addressed store, hash-verified

86 return relax.TensorStructInfo(data.shape, str(data.dtype))
87
88 def _numpy_to_tvm(data):
89 if isinstance(data, list):
90 return [_numpy_to_tvm(d) for d in data]
91 return tvm.runtime.tensor(data)
92
93 def _tvm_to_numpy(data, ignore_idx=[]):
94 if isinstance(data, tvm_ffi.Array):

Callers 2

forwardFunction · 0.70
relax_check_gradientsFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…