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

Function test_sum

tests/python/relax/test_op_gradient_numeric.py:333–335  ·  view source on GitHub ↗
(target, dev)

Source from the content-addressed store, hash-verified

331
332@tvm.testing.parametrize_targets("llvm")
333def test_sum(target, dev):
334 data1_numpy = np.random.uniform(0, 16, (3, 3)).astype(np.float32)
335 relax_check_gradients(relax.op.sum, [data1_numpy], target, dev)
336
337
338@tvm.testing.parametrize_targets("llvm")

Callers

nothing calls this directly

Calls 3

relax_check_gradientsFunction · 0.85
uniformMethod · 0.80
astypeMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…