Function test_sum_keepdims

tests/python/relax/test_op_gradient_numeric.py:345–347 · view source on GitHub ↗

(target, dev)

Source from the content-addressed store, hash-verified

343
344	@tvm.testing.parametrize_targets("llvm")
345	def test_sum_keepdims(target, dev):
346	data1_numpy = np.random.uniform(0, 16, (3, 3)).astype(np.float32)
347	relax_check_gradients(relax.op.sum, [data1_numpy], target, dev, keepdims=True, axis=1)
348
349
350	@tvm.testing.parametrize_targets("llvm")

nothing calls this directly

relax_check_gradientsFunction · 0.85

uniformMethod · 0.80

astypeMethod · 0.45

no test coverage detected

searching dependent graphs…