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

Function _numpy_to_sinfo

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

Source from the content-addressed store, hash-verified

81
82 # Helper functions
83 def _numpy_to_sinfo(data):
84 if isinstance(data, list):
85 return relax.TupleStructInfo([_numpy_to_sinfo(d) for d in data])
86 return relax.TensorStructInfo(data.shape, str(data.dtype))
87
88 def _numpy_to_tvm(data):
89 if isinstance(data, list):

Callers 1

relax_check_gradientsFunction · 0.85

Calls 1

strFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…